From 5900cb2a31e6f8066fcb40aec448eec04bb5f64a Mon Sep 17 00:00:00 2001 From: zhaoqingang <zhaoqg0118@163.com> Date: 星期一, 10 二月 2025 10:13:21 +0800 Subject: [PATCH] 测试 场景分析 bug --- models/locationAnalysis.go | 145 +++++++++++++++++++++--------------------------- 1 files changed, 63 insertions(+), 82 deletions(-) diff --git a/models/locationAnalysis.go b/models/locationAnalysis.go index 347927f..7c2f942 100644 --- a/models/locationAnalysis.go +++ b/models/locationAnalysis.go @@ -242,6 +242,7 @@ tagTypes = append(tagTypes, document_number_map[personId].PersonType) lastAppearanceTime = document_number_map[personId].LastAppearanceTime } + logger.Debugf("tagTypes %s ---------------%s--------", tagTypes, record.DocumentNumbers) _, typeNames, err := service.GetPersonTypeNameByTypes(tagTypes) if err != nil { return err @@ -256,6 +257,8 @@ OrgID: record.OrgId, ObjectIds: strings.Join(record.DocumentNumbers, ","), Location: fmt.Sprintf("%s%s%s", domains[record.CommunityId].Name, record.Building, record.Floor), + Building: record.Building, + Floor: record.Floor, PicDate: time.Unix(lastAppearanceTime, 0).Format("2006-01-02 15:04:05"), FirstPersonID: record.DocumentNumbers[0], } @@ -343,11 +346,12 @@ // }, // }, //}) + var queryTimes []map[string]interface{} for date := startTime; date.Before(nowTime); date = date.Add(24 * time.Hour) { start := time.Date(date.Year(), date.Month(), date.Day(), locationModel.StartTime, 0, 0, 0, date.Location()) end := time.Date(date.Year(), date.Month(), date.Day(), locationModel.EndTime, 0, 0, 0, date.Location()) - filters = append(filters, map[string]interface{}{ + queryTimes = append(queryTimes, map[string]interface{}{ "range": map[string]interface{}{ "picDate": map[string]interface{}{ "gte": start.Format("2006-01-02 15:04:05"), @@ -356,6 +360,16 @@ }, }) } + //var authParams = map[string]interface{}{ + // "bool": map[string]interface{}{ + // "should": queryTimes, + // }, + //} + filters = append(filters, map[string]interface{}{ + "bool": map[string]interface{}{ + "should": queryTimes, + }, + }) //logger.Debugf("filters--------------------------------------- %s", filters) query := map[string]interface{}{ "query": map[string]interface{}{ @@ -419,14 +433,6 @@ "size": 0, } - query = map[string]interface{}{ - "query": map[string]interface{}{ - "bool": map[string]interface{}{ - "filter": filters, - }, - }, - "size": 10000, - } if err := json.NewEncoder(&buf).Encode(query); err != nil { return nil, fmt.Errorf("error encoding query: %s", err) } @@ -453,84 +459,59 @@ if err := json.NewDecoder(res.Body).Decode(&result); err != nil { return nil, fmt.Errorf("error parsing response body: %s", err) } - logger.Debugf("ressss--------------------------------------- %s", result) // 瑙f瀽鑱氬悎缁撴灉 var records []*LocationRecord - //dat, ok := result["hits"].(map[string]interface{}) + if aggs, ok := result["aggregations"].(map[string]interface{}); ok { + if orgBuckets, ok := aggs["orgs"].(map[string]interface{})["buckets"].([]interface{}); ok { + for _, orgBucket := range orgBuckets { + orgId := orgBucket.(map[string]interface{})["key"].(string) + //logger.Debugf("orgId--------------------------------------- %s", orgId) + // 瑙f瀽鎸塩ommunityId鐨勮仛鍚堢粨鏋� + if communityBuckets, ok := orgBucket.(map[string]interface{})["community"].(map[string]interface{})["buckets"].([]interface{}); ok { + for _, communityBucket := range communityBuckets { + communityId := communityBucket.(map[string]interface{})["key"].(string) + //logger.Debugf("communityId--------------------------------------- %s", communityId) + // 瑙f瀽鎸塨uilding鐨勮仛鍚堢粨鏋� + if locationBuckets, ok := communityBucket.(map[string]interface{})["location"].(map[string]interface{})["buckets"].([]interface{}); ok { + for _, locationBucket := range locationBuckets { + building := locationBucket.(map[string]interface{})["key"].(string) + //logger.Debugf("building--------------------------------------- %s", building) + // 瑙f瀽鎸塮loor鐨勮仛鍚堢粨鏋� + if floorBuckets, ok := locationBucket.(map[string]interface{})["floor"].(map[string]interface{})["buckets"].([]interface{}); ok { + for _, floorBucket := range floorBuckets { + floor := floorBucket.(map[string]interface{})["key"].(string) + //logger.Debugf("floor--------------------------------------- %s -----%s ", floor, floorBucket) + appearCount := floorBucket.(map[string]interface{})["doc_count"].(float64) + //logger.Debugf("appearCount--------------------------------------- %s- --", appearCount) + // 鏋勫缓 LocationRecord 缁撴瀯浣� + var persons []string + if docNumBuckets, ok := floorBucket.(map[string]interface{})["document_numbers"].(map[string]interface{})["buckets"].([]interface{}); ok { + for _, docNumBucket := range docNumBuckets { + persons = append(persons, docNumBucket.(map[string]interface{})["key"].(string)) + } + } - dat := result["hits"].(map[string]interface{}) + logger.Debugf("floor, persons--,appearCount------------------------------------- %s-- %s --", floor, persons, appearCount) + record := &LocationRecord{ + //PicDate: timestamp, + DocumentNumbers: persons, + CommunityId: communityId, + Building: building, + Floor: floor, + OrgId: orgId, + AppearCount: int(appearCount), + } - var data = make(map[string]interface{}, 2) - - tmp, b := dat["total"].(map[string]interface{}) - if b != true { - data["total"] = dat["total"].(float64) - } else { - data["total"] = tmp["value"].(float64) + records = append(records, record) + } + } + } + } + } + } + } + } } - sources := []interface{}{} - for _, value := range dat["hits"].([]interface{}) { - sourceData := value.(map[string]interface{}) - source := sourceData["_source"].(map[string]interface{}) - source["_id"] = sourceData["_id"] - /*sdkType := source["sdkType"] - if sdkType != nil { - sdk, _ := strconv.Atoi(sdkType.(string)) - source["sdkType"] = sdkTypeToValue(sdk) - }*/ - sources = append(sources, source) - } - data["datalist"] = sources - logger.Debugf("ressss--------------data------------------------- %s", data) - //if aggs, ok := result["aggregations"].(map[string]interface{}); ok { - // if orgBuckets, ok := aggs["orgs"].(map[string]interface{})["buckets"].([]interface{}); ok { - // for _, orgBucket := range orgBuckets { - // orgId := orgBucket.(map[string]interface{})["key"].(string) - // logger.Debugf("orgId--------------------------------------- %s", orgId) - // // 瑙f瀽鎸塩ommunityId鐨勮仛鍚堢粨鏋� - // if communityBuckets, ok := orgBucket.(map[string]interface{})["community"].(map[string]interface{})["buckets"].([]interface{}); ok { - // for _, communityBucket := range communityBuckets { - // communityId := communityBucket.(map[string]interface{})["key"].(string) - // logger.Debugf("communityId--------------------------------------- %s", communityId) - // // 瑙f瀽鎸塨uilding鐨勮仛鍚堢粨鏋� - // if locationBuckets, ok := communityBucket.(map[string]interface{})["location"].(map[string]interface{})["buckets"].([]interface{}); ok { - // for _, locationBucket := range locationBuckets { - // building := locationBucket.(map[string]interface{})["key"].(string) - // logger.Debugf("building--------------------------------------- %s", building) - // // 瑙f瀽鎸塮loor鐨勮仛鍚堢粨鏋� - // if floorBuckets, ok := locationBucket.(map[string]interface{})["floor"].(map[string]interface{})["buckets"].([]interface{}); ok { - // for _, floorBucket := range floorBuckets { - // floor := floorBucket.(map[string]interface{})["key"].(string) - // appearCount := floorBucket.(map[string]interface{})["filter_floor"].(int) - // // 鏋勫缓 LocationRecord 缁撴瀯浣� - // var persons []string - // if docNumBuckets, ok := floorBucket.(map[string]interface{})["document_numbers"].(map[string]interface{})["buckets"].([]interface{}); ok { - // for _, docNumBucket := range docNumBuckets { - // persons = append(persons, docNumBucket.(map[string]interface{})["key"].(string)) - // } - // } - // - // logger.Debugf("floor--------------------------------------- %s-- %s --", floor, appearCount) - // record := &LocationRecord{ - // //PicDate: timestamp, - // DocumentNumbers: persons, - // CommunityId: communityId, - // Building: building, - // Floor: floor, - // OrgId: orgId, - // AppearCount: appearCount, - // } - // - // records = append(records, record) - // } - // } - // } - // } - // } - // } - // } - // } - //} return records, nil } -- Gitblit v1.8.0