package AreaDao

import (
	"dsBigData/Business/Area/AreaModel"
	"dsBigData/Utils/CommonUtil"
	"dsBigData/Utils/EsUtil"
	"encoding/json"
	"github.com/olivere/elastic/v7"
	"github.com/tidwall/gjson"
)

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

/**
功能：根据市编码获取该市下的区(县)列表
作者：吴缤
日期：2020-06-30
typeId -1：全部  1：县市区  2：开发区
*/
func PageDistrictByCityCode(cityCode string, typeId string) ([]AreaModel.Area, error) {
	enableFlagTerm := elastic.NewTermQuery("enable_flag", 1)
	delFlagTerm := elastic.NewTermQuery("del_flag", 0)
	masterCodeTerm := elastic.NewTermQuery("data_content.master_code", cityCode)
	var boolQuery *elastic.BoolQuery
	if typeId == "-1" {
		boolQuery = elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, masterCodeTerm)
	} else {
		typeIdTerm := elastic.NewTermQuery("data_content.area_type_id", typeId)
		boolQuery = elastic.NewBoolQuery().Must(enableFlagTerm, delFlagTerm, masterCodeTerm, typeIdTerm)
	}
	result, err := esClient.Search().
		Index("org_area").
		Query(boolQuery).
		Size(50).
		Do(CTX)
	if err != nil {
		return nil, err
	}
	resByte, err := json.Marshal(result.Hits)
	if err != nil {
		return nil, err
	}
	var areaArr []AreaModel.Area
	resStr := string(resByte)
	resCount := gjson.Get(resStr, "total.value")
	for i := 0; i < int(resCount.Num); i++ {
		var area AreaModel.Area
		area.AreaCode = gjson.Get(resStr, "hits."+CommonUtil.ConvertIntToString(i)+"._source.data_content.area_code").Str
		area.AreaName = gjson.Get(resStr, "hits."+CommonUtil.ConvertIntToString(i)+"._source.data_content.area_name").Str
		areaArr = append(areaArr, area)
	}
	return areaArr, nil
}