package SchoolDao

import (
	"dsBigData/Business/School/SchoolModel"
	"dsBigData/Utils/CommonUtil"
	"dsBigData/Utils/EsUtil"
	"encoding/json"
	"github.com/olivere/elastic/v7"
	"github.com/tidwall/gjson"
	"strings"
	"time"
)

var esClient = EsUtil.EsClient
var CTX = EsUtil.CTX

func PageEduAssistInfoByDistrictCode(districtCode string) ([]SchoolModel.EduAssist, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	districtCodeTerm := elastic.NewTermQuery("district_code", districtCode)
	orgTypeTerm := elastic.NewTermQuery("org_type", "3")
	bUseTerm := elastic.NewTermQuery("data_content.b_use", 1)

	boolQuery := elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, districtCodeTerm, orgTypeTerm, bUseTerm)

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		Size(1000).
		Do(CTX)

	if err != nil {
		return nil, err
	}

	resByte, err := json.Marshal(result.Hits)
	if err != nil {
		return nil, err
	}
	var eduAssistArr []SchoolModel.EduAssist
	resStr := string(resByte)
	resCount := gjson.Get(resStr, "total.value")
	for i := 0; i < int(resCount.Num); i++ {
		var eduAssist SchoolModel.EduAssist
		eduAssist.EduAssistId = gjson.Get(resStr, "hits."+CommonUtil.ConvertIntToString(i)+"._source.data_content.org_id").Str
		eduAssist.EduAssistName = gjson.Get(resStr, "hits."+CommonUtil.ConvertIntToString(i)+"._source.data_content.org_name").Str
		eduAssistArr = append(eduAssistArr, eduAssist)
	}
	return eduAssistArr, nil
}

/**
功能：根据区编码获取该区下的学校列表
作者：吴缤
日期：2020-06-30
xxbxlxm -1：全部
*/
func PageSchoolInfoByDistrictCode(districtCode string, xxbxlxm string) ([]SchoolModel.School, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	districtCodeTerm := elastic.NewTermQuery("district_code", districtCode)
	orgTypeTerm := elastic.NewTermQuery("org_type", "2")
	var boolQuery *elastic.BoolQuery
	if xxbxlxm == "-1" {
		boolQuery = elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, districtCodeTerm, orgTypeTerm)
	} else {
		xxbxlxmTerm := elastic.NewTermQuery("school_type", xxbxlxm)
		boolQuery = elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, districtCodeTerm, orgTypeTerm, xxbxlxmTerm)
	}
	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		Size(1000).
		Do(CTX)
	if err != nil {
		return nil, err
	}
	resByte, err := json.Marshal(result.Hits)
	if err != nil {
		return nil, err
	}
	var schoolArr []SchoolModel.School
	resStr := string(resByte)
	resCount := gjson.Get(resStr, "total.value")
	for i := 0; i < int(resCount.Num); i++ {
		var school SchoolModel.School
		school.SchoolId = gjson.Get(resStr, "hits."+CommonUtil.ConvertIntToString(i)+"._source.data_content.org_id").Str
		school.SchoolName = gjson.Get(resStr, "hits."+CommonUtil.ConvertIntToString(i)+"._source.data_content.org_name").Str
		schoolArr = append(schoolArr, school)
	}
	return schoolArr, nil
}

func AggsDistrictStatEduAssist(districtCodeArr []string) ([]SchoolModel.EduAssistCountAggsDistrict, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	orgTypeTerm := elastic.NewTermQuery("org_type", "3")
	interfaceArr := CommonUtil.ConvertStringArrToInterfaceArr(districtCodeArr)
	cityCodeTerms := elastic.NewTermsQuery("district_code", interfaceArr...)

	boolQuery := elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, cityCodeTerms, orgTypeTerm)

	districtCodeAggs := elastic.
		NewTermsAggregation().
		Field("district_code").
		Size(50)

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		Size(0).
		Aggregation("district", districtCodeAggs).
		Do(CTX)

	if err != nil {
		return nil, err
	}

	resByte, err := json.Marshal(result.Aggregations)
	resStr := string(resByte)

	var eduAssistCountAggsDistrictArr []SchoolModel.EduAssistCountAggsDistrict
	resCount := gjson.Get(resStr, "district.buckets.#")
	for i := 0; i < int(resCount.Num); i++ {
		var eduAssistCountAggsDistrict SchoolModel.EduAssistCountAggsDistrict
		eduAssistCountAggsDistrict.AreaCode = gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".key").Str
		eduAssistCountAggsDistrict.Count = gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".doc_count").Raw

		eduAssistCountAggsDistrictArr = append(eduAssistCountAggsDistrictArr, eduAssistCountAggsDistrict)
	}
	return eduAssistCountAggsDistrictArr, nil
}

func AggsDistrictStatSchoolCount(districtCodeArr []string) ([]SchoolModel.SchoolCountAggsDistrict, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	orgTypeTerm := elastic.NewTermQuery("org_type", "2")
	interfaceArr := CommonUtil.ConvertStringArrToInterfaceArr(districtCodeArr)
	cityCodeTerm := elastic.NewTermsQuery("district_code", interfaceArr...)

	boolQuery := elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, cityCodeTerm, orgTypeTerm)

	districtCodeAggs := elastic.
		NewTermsAggregation().
		Field("district_code").
		Size(50)

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		Size(0).
		Aggregation("district", districtCodeAggs).
		Do(CTX)

	if err != nil {
		return nil, err
	}

	resByte, err := json.Marshal(result.Aggregations)
	resStr := string(resByte)

	var schoolCountAggsDistrictArr []SchoolModel.SchoolCountAggsDistrict
	resCount := gjson.Get(resStr, "district.buckets.#")
	for i := 0; i < int(resCount.Num); i++ {
		var schoolCountAggsDistrict SchoolModel.SchoolCountAggsDistrict
		schoolCountAggsDistrict.AreaCode = gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".key").Str
		schoolCountAggsDistrict.Count = gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".doc_count").Raw

		schoolCountAggsDistrictArr = append(schoolCountAggsDistrictArr, schoolCountAggsDistrict)
	}
	return schoolCountAggsDistrictArr, nil
}

func AggsXxbxlxStatSchoolCountByDistrictCode(districtCode string) ([]SchoolModel.SchoolCountAggsXxbxlx, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	orgTypeTerm := elastic.NewTermQuery("org_type", "2")
	districtCodeTerm := elastic.NewTermQuery("district_code", districtCode)

	boolQuery := elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, orgTypeTerm, districtCodeTerm)

	xxbxlxAggs := elastic.
		NewTermsAggregation().
		Field("school_type").
		Size(50)

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		Size(0).
		Aggregation("xxbxlx", xxbxlxAggs).
		Do(CTX)

	if err != nil {
		return nil, err
	}

	resByte, err := json.Marshal(result.Aggregations)
	resStr := string(resByte)

	var schoolCountAggsXxbxlxArr []SchoolModel.SchoolCountAggsXxbxlx
	resCount := gjson.Get(resStr, "xxbxlx.buckets.#")
	for i := 0; i < int(resCount.Num); i++ {
		var schoolCountAggsXxbxlx SchoolModel.SchoolCountAggsXxbxlx
		schoolCountAggsXxbxlx.Xxbxlxm = gjson.Get(resStr, "xxbxlx.buckets."+CommonUtil.ConvertIntToString(i)+".key").Str
		schoolCountAggsXxbxlx.Count = gjson.Get(resStr, "xxbxlx.buckets."+CommonUtil.ConvertIntToString(i)+".doc_count").Raw
		schoolCountAggsXxbxlxArr = append(schoolCountAggsXxbxlxArr, schoolCountAggsXxbxlx)
	}

	return schoolCountAggsXxbxlxArr, nil
}

/**
功能：聚合区统计学校数量
作者：吴缤
日期：2020-07-02
districtCodes：一个或多个区编码
aggsXxbxlx：是否需要办学类型的子聚合 true：需要  false：不需要
*/
func GetSchoolCountAggsDistrict(districtCodeArr []string, aggsXxbxlx bool) ([]SchoolModel.SchoolCountAggsDistrict, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	orgTypeTerm := elastic.NewTermQuery("org_type", "2")
	interfaceArr := CommonUtil.ConvertStringArrToInterfaceArr(districtCodeArr)
	cityCodeTerm := elastic.NewTermsQuery("district_code", interfaceArr...)

	boolQuery := elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, cityCodeTerm, orgTypeTerm)

	districtCodeAggs := elastic.
		NewTermsAggregation().
		Field("district_code").
		Size(50).
		CollectionMode("breadth_first")

	if aggsXxbxlx {
		xxbxlxAggs := elastic.
			NewTermsAggregation().
			Field("school_type").
			Size(20)

		districtCodeAggs.SubAggregation("xxbxlx", xxbxlxAggs)
	}

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		Size(0).
		Aggregation("district", districtCodeAggs).
		Do(CTX)
	if err != nil {
		return nil, err
	}

	resByte, err := json.Marshal(result.Aggregations)
	resStr := string(resByte)

	var schoolCountAggsDistrictArr []SchoolModel.SchoolCountAggsDistrict
	resCount := gjson.Get(resStr, "district.buckets.#")
	for i := 0; i < int(resCount.Num); i++ {
		var schoolCountAggsDistrict SchoolModel.SchoolCountAggsDistrict
		schoolCountAggsDistrict.AreaCode = gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".key").Str
		schoolCountAggsDistrict.Count = gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".doc_count").Raw
		XxbxlxArr := make([]SchoolModel.SchoolCountAggsXxbxlx, 0)
		if aggsXxbxlx {
			xxbxlxCount := gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".xxbxlx.buckets.#")

			for j := 0; j < int(xxbxlxCount.Num); j++ {
				var xxbxlx SchoolModel.SchoolCountAggsXxbxlx
				xxbxlx.Xxbxlxm = gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".xxbxlx.buckets."+CommonUtil.ConvertIntToString(j)+".key").Str
				xxbxlx.Count = gjson.Get(resStr, "district.buckets."+CommonUtil.ConvertIntToString(i)+".xxbxlx.buckets."+CommonUtil.ConvertIntToString(j)+".doc_count").Raw
				XxbxlxArr = append(XxbxlxArr, xxbxlx)
			}
		}
		//schoolCountAggsDistrict.Xxbxlx = XxbxlxArr

		schoolCountAggsDistrictArr = append(schoolCountAggsDistrictArr, schoolCountAggsDistrict)
	}
	return schoolCountAggsDistrictArr, nil
}

func PageStageBySchoolId(schoolId string) ([]SchoolModel.Stage, error) {
	orgIdTerm := elastic.NewTermQuery("org_id", schoolId)
	bUseTerm := elastic.NewTermQuery("data_content.b_use", 1)
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)

	boolQuery := elastic.NewBoolQuery().Must(orgIdTerm, bUseTerm, enableFlagTerm, delFlagTerm)

	stagAggs := elastic.
		NewTermsAggregation().
		Field("data_content.stage_id").
		Size(10)

	result, err := esClient.Search().
		Index("user_student").
		Query(boolQuery).
		Size(0).
		Aggregation("stag", stagAggs).
		Do(CTX)

	if err != nil {
		return nil, err
	}

	resByte, err := json.Marshal(result.Aggregations)
	resStr := string(resByte)

	var stageArr []SchoolModel.Stage
	resCount := gjson.Get(resStr, "stag.buckets.#")
	for i := 0; i < int(resCount.Num); i++ {
		var stage SchoolModel.Stage
		stageId := gjson.Get(resStr, "stag.buckets."+CommonUtil.ConvertIntToString(i)+".key").Str
		stageName := ""
		stage.StageId = stageId
		switch stageId {
		case "2":
			stageName = "小学"
			break
		case "3":
			stageName = "初中"
			break
		case "4":
			stageName = "高中"
			break
		}

		stage.StageName = stageName

		stageArr = append(stageArr, stage)
	}
	return stageArr, nil
}

/**
功能：根据市或区的编码(也可传单位ID)和机构类型获取总数
*/
func GetOrgTotalCount(areaCode string, orgType string) (int, error) {
	var areaCodeTerm *elastic.TermQuery
	if len(areaCode) == 6 {
		if strings.HasSuffix(areaCode, "00") { //市
			areaCodeTerm = elastic.NewTermQuery("city_code", areaCode)
		} else { //区
			areaCodeTerm = elastic.NewTermQuery("district_code", areaCode)
		}
	} else {
		areaCodeTerm = elastic.NewTermQuery("org_id", areaCode)
	}
	orgTypeTerm := elastic.NewTermQuery("data_content.org_type", orgType)
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	bUseTerm := elastic.NewTermQuery("data_content.b_use", 1)

	boolQuery := elastic.NewBoolQuery().Must(areaCodeTerm, orgTypeTerm, enableFlagTerm, delFlagTerm, bUseTerm)

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		TrackTotalHits(true).
		Do(CTX)

	if err != nil {
		return 0, err
	}

	return int(result.TotalHits()), nil
}

/**
功能：根据市或区的编码和机构类型获取当前学期新增数
*/
func GetOrgCurrentTermAddCount(areaCode string, orgType string) (int, error) {
	endYearStr := CommonUtil.ConvertIntToString(time.Now().Year())
	month := CommonUtil.ConvertStringToInt32(time.Now().Format("1"))
	monthStr := "08"
	if month >= 3 && month <= 7 {
		monthStr = "03"
	}

	createTime := endYearStr + "/" + monthStr + "/01 00:00:01"

	var areaCodeTerm *elastic.TermQuery
	if len(areaCode) == 6 {
		if strings.HasSuffix(areaCode, "00") { //市
			areaCodeTerm = elastic.NewTermQuery("city_code", areaCode)
		} else { //区
			areaCodeTerm = elastic.NewTermQuery("district_code", areaCode)
		}
	} else {
		areaCodeTerm = elastic.NewTermQuery("org_id", areaCode)
	}
	orgTypeTerm := elastic.NewTermQuery("data_content.org_type", orgType)
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	bUseTerm := elastic.NewTermQuery("data_content.b_use", 1)
	createTimeRang := elastic.NewRangeQuery("data_content.create_time").Gte(createTime)

	boolQuery := elastic.NewBoolQuery().Must(areaCodeTerm, orgTypeTerm, enableFlagTerm, delFlagTerm, bUseTerm, createTimeRang)

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		TrackTotalHits(true).
		Do(CTX)

	if err != nil {
		return 0, err
	}

	return int(result.TotalHits()), nil
}

/**
功能：根据学校ID获取班级总数
*/
func GetClassTotalCount(schoolId string) (int, error) {
	orgIdTerm := elastic.NewTermQuery("org_id", schoolId)
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	bUseTerm := elastic.NewTermQuery("data_content.b_use", 1)

	boolQuery := elastic.NewBoolQuery().Must(orgIdTerm, enableFlagTerm, delFlagTerm, bUseTerm)

	result, err := esClient.Search().
		Index("org_class").
		Query(boolQuery).
		TrackTotalHits(true).
		Do(CTX)

	if err != nil {
		return 0, err
	}

	return int(result.TotalHits()), nil
}

/**
功能：根据学校ID获取班级当前学期新增数
*/
func GetClassCurrentTermAddCount(schoolId string) (int, error) {
	endYearStr := CommonUtil.ConvertIntToString(time.Now().Year())
	month := CommonUtil.ConvertStringToInt32(time.Now().Format("1"))
	monthStr := "08"
	if month >= 3 && month <= 7 {
		monthStr = "03"
	}

	createTime := endYearStr + "/" + monthStr + "/01 00:00:01"

	orgIdTerm := elastic.NewTermQuery("org_id", schoolId)
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	bUseTerm := elastic.NewTermQuery("data_content.b_use", 1)
	createTimeRang := elastic.NewRangeQuery("data_content.create_time").Gte(createTime)

	boolQuery := elastic.NewBoolQuery().Must(orgIdTerm, enableFlagTerm, delFlagTerm, bUseTerm, createTimeRang)

	result, err := esClient.Search().
		Index("org_class").
		Query(boolQuery).
		TrackTotalHits(true).
		Do(CTX)

	if err != nil {
		return 0, err
	}

	return int(result.TotalHits()), nil
}

func GetBaseAccessTop(areaCode string, topNum string) ([]SchoolModel.BaseAccessTopAggsOrgId, error) {
	var areaCodeTerm *elastic.TermQuery
	if strings.HasSuffix(areaCode, "00") { //市
		areaCodeTerm = elastic.NewTermQuery("city_code", areaCode)
	} else { //区
		areaCodeTerm = elastic.NewTermQuery("district_code", areaCode)
	}
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)

	boolQuery := elastic.NewBoolQuery().Must(areaCodeTerm, enableFlagTerm, delFlagTerm)

	orgIdAggs := elastic.
		NewTermsAggregation().
		Field("org_id").
		Size(CommonUtil.ConvertStringToInt(topNum))

	result, err := esClient.Search().
		Index("log_login").
		Query(boolQuery).
		Size(0).
		Aggregation("orgId", orgIdAggs).
		Do(CTX)

	if err != nil {
		return nil, err
	}

	resByte, err := json.Marshal(result.Aggregations)
	resStr := string(resByte)

	var baseAccessTopAggsOrgIdArr []SchoolModel.BaseAccessTopAggsOrgId
	resCount := gjson.Get(resStr, "orgId.buckets.#")
	for i := 0; i < int(resCount.Num); i++ {
		var baseAccessTopAggsOrgId SchoolModel.BaseAccessTopAggsOrgId
		baseAccessTopAggsOrgId.OrgId = gjson.Get(resStr, "orgId.buckets."+CommonUtil.ConvertIntToString(i)+".key").Str
		baseAccessTopAggsOrgId.Count = int(gjson.Get(resStr, "orgId.buckets."+CommonUtil.ConvertIntToString(i)+".doc_count").Num)

		baseAccessTopAggsOrgIdArr = append(baseAccessTopAggsOrgIdArr, baseAccessTopAggsOrgId)
	}
	return baseAccessTopAggsOrgIdArr, nil
}

func GetOrgInfoByOrgIds(orgIds []string) ([]SchoolModel.Org, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	bUseTerm := elastic.NewTermQuery("data_content.b_use", 1)
	interfaceArr := CommonUtil.ConvertStringArrToInterfaceArr(orgIds)
	orgIdTerms := elastic.NewTermsQuery("org_id", interfaceArr...)

	boolQuery := elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, bUseTerm, orgIdTerms)

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		Size(100).
		Do(CTX)

	if err != nil {
		return nil, err
	}

	resByte, err := json.Marshal(result.Hits)
	resStr := string(resByte)

	var orgArr []SchoolModel.Org
	resCount := gjson.Get(resStr, "total.value")
	for i := 0; i < int(resCount.Num); i++ {
		var org SchoolModel.Org
		org.OrgId = gjson.Get(resStr, "hits."+CommonUtil.ConvertIntToString(i)+"._source.data_content.org_id").Str
		org.OrgName = gjson.Get(resStr, "hits."+CommonUtil.ConvertIntToString(i)+"._source.data_content.org_name").Str

		orgArr = append(orgArr, org)
	}
	return orgArr, nil
}

/**
功能：根据市编码获取学校数量
*/
func GetCityDirectlySchoolCount(cityCode string) (int, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	bUseTerm := elastic.NewTermQuery("data_content.b_use", 1)
	districtCodeTerm := elastic.NewTermQuery("district_code", cityCode)
	orgTypeTerm := elastic.NewTermQuery("org_type", "2")

	boolQuery := elastic.NewBoolQuery().Must(districtCodeTerm, orgTypeTerm, enableFlagTerm, delFlagTerm, bUseTerm)

	result, err := esClient.Search().
		Index("org_school").
		Query(boolQuery).
		Size(0).
		Do(CTX)

	if err != nil {
		return 0, err
	}

	return int(result.TotalHits()), nil

}