From f99b8a6d2ec0fc2414b939c3fef4d9338d13fce4 Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@smartai.com>
Date: 星期三, 12 二月 2025 14:07:07 +0800
Subject: [PATCH] 调整场所分析身份条件
---
models/locationAnalysis.go | 476 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++--
1 files changed, 455 insertions(+), 21 deletions(-)
diff --git a/models/locationAnalysis.go b/models/locationAnalysis.go
index e9fbc23..a1c352e 100644
--- a/models/locationAnalysis.go
+++ b/models/locationAnalysis.go
@@ -1,7 +1,17 @@
package models
import (
+ "bytes"
+ "context"
+ "encoding/json"
+ "errors"
"fmt"
+ "github.com/elastic/go-elasticsearch/v6"
+ //"log"
+ "model-engine/config"
+ "model-engine/pkg/set"
+ "strconv"
+ "strings"
"time"
"model-engine/db"
@@ -10,36 +20,64 @@
)
type LocationModel struct {
- AreaIds map[string]struct{}
- Building string // 妤兼爧
- AlarmType db.AlarmType // 棰勮鏂瑰紡
- KeyPersonType string // 浜哄憳绫诲瀷
- PersonLabel string
- PersonCount int // 浜烘暟, 鍑虹幇鐨勫悓绫诲瀷浜哄憳鏁伴噺
- Appearances int // 鍑虹幇娆℃暟,
- Duration int // 鏃堕棿鑼冨洿, 鍗曚綅澶�
- Task *db.ModelTask
+ AreaIds []interface{} `json:"-"`
+ OrgIds []interface{} `json:"-"`
+ Building string // 妤兼爧
+ Floor string
+ AlarmType db.AlarmType // 棰勮鏂瑰紡
+ KeyPersonType string // 浜哄憳绫诲瀷
+ PersonLabel string // 浜哄憳鏍囩
+ PersonIdentity []string // 浜哄憳韬唤
+ KeyTypes []string // 浜哄憳韬唤
+ Duration int // 鏃堕棿鑼冨洿
+ Appearances int // 鍑虹幇娆℃暟,
+ StartTime int // 鏃堕棿鑼冨洿, 寮�濮嬫椂闂�
+ EndTime int // 鏃堕棿鑼冨洿, 缁撴潫鏃堕棿
+ Task *db.ModelTask
}
func (m *LocationModel) Init(task *db.ModelTask) error {
- m.AreaIds = make(map[string]struct{})
- for _, a := range task.DomainUnitIds {
- m.AreaIds[a] = struct{}{}
+ //m.AreaIds = make(map[string]struct{})
+ //for _, a := range task.DomainUnitIds {
+ // m.AreaIds[a] = struct{}{}
+ //}
+ if len(task.DomainUnitIds) == 0 {
+ return errors.New("empty domain set")
+ }
+ orgIds, areaIds, err := service.GetOrgIdsAndAreaIdsByDomainUnitIds(task.DomainUnitIds)
+ if err != nil {
+ return err
}
m.Task = task
+ m.OrgIds = orgIds
+ m.AreaIds = areaIds
m.Building = task.Building
m.AlarmType = task.AlarmType
+ m.KeyTypes = []string{"1", "2", "3", "4", "5"} //task.IdentityType
m.KeyPersonType = task.PersonType
m.PersonLabel = task.PersonLabel
-
- for _, v := range task.Rules {
- if v.Alias == "personCount" {
- if val, ok := v.Value.(float64); ok {
- m.PersonCount = int(val)
+ if task.IdentityType != "" {
+ for _, t := range strings.Split(task.IdentityType, ",") {
+ if t == "all" {
+ m.PersonIdentity = []string{"stranger", "visitor", "resident"}
+ break
+ } else {
+ m.PersonIdentity = append(m.PersonIdentity, t)
}
}
+ } else {
+ m.PersonIdentity = []string{"stranger", "visitor", "resident"}
+ }
+ if task.PersonType != "" {
+ m.KeyTypes = strings.Split(task.PersonType, ",")
+ }
+
+ //if len(m.AreaIds) == 0 {
+ // m.AreaIds = m.OrgIds
+ //}
+ for _, v := range task.Rules {
if v.Alias == "appearances" {
if val, ok := v.Value.(float64); ok {
m.Appearances = int(val)
@@ -49,6 +87,13 @@
if v.Alias == "duration" {
if val, ok := v.Value.(float64); ok {
m.Duration = int(val)
+ }
+ }
+ if v.Alias == "timeRange" {
+ if val, ok := v.Value.(string); ok {
+ ages := strings.Split(val, ",")
+ m.StartTime, _ = strconv.Atoi(ages[0])
+ m.EndTime, _ = strconv.Atoi(ages[1])
}
}
}
@@ -63,13 +108,165 @@
return nil
}
+type LocationRecord struct {
+ IDCard string `json:"idCard"`
+ PicDate string `json:"picDate"`
+ DocumentNumbers []string
+ CommunityId string `json:"communityId"`
+ OrgId string `json:"orgId"`
+ Building string `json:"building"`
+ Floor string `json:"floor"`
+ AppearCount int `gorm:"type:int;" json:"appearCount"` // 鍑虹幇娆℃暟
+ //AppearInterval int `gorm:"type:int;" json:"appearInterval"` // 鍑虹幇闂撮殧锛屽崟浣嶄负绉�
+}
+
+type LocationPersonInfo struct {
+ Id string `json:"id"`
+ DocumentNumber string `json:"document_number"`
+ PersonType string `json:"person_type"`
+ LastAppearanceTime int64 `json:"last_appearance_time"`
+}
+
func (m *LocationModel) Run() error {
- // 鏍规嵁閰嶇疆鐨勬椂闂存澶╂暟, 杩囨护瑙勫垯閰嶇疆鐨勭浉鍚岄噸鐐逛汉鍛樼被鍨嬪拰鐩稿悓鏍囩浜哄憳棰戠箒鍑虹幇鐨勬ゼ灞傚湴鍧�
+ // 鏍规嵁閰嶇疆鐨勬椂闂存澶╂暟, 姣忓ぉ鐨勬椂闂磋寖鍥村唴锛� 閲嶇偣浜哄憳绫诲瀷鎴栬�呯壒瀹氭爣绛句汉鍛樺嚭鐜扮殑妤煎眰娆℃暟瓒呰繃闃堝��
results := make([]*db.ModelTaskResults, 0)
+ var baseFilter, labelFilter, keyFilter []LocationPersonInfo
+ var documentNumberMap = make(map[string]LocationPersonInfo)
+ var documentNumberList []string
+ communityIds := m.AreaIds
+ if len(communityIds) == 0 {
+ communityIds = m.OrgIds
+ }
+ err := db.GetDB().Raw(`
+ SELECT
+ p.id,
+ s.document_number,
+ s.last_appearance_time
+ FROM
+ snapshot_count_summary AS s
+ JOIN person AS p ON p.id = s.document_number
+ WHERE
+ (s.community_id IN ? OR s.org_id IN ?)
+ AND s.status IN ?
+ `, communityIds, m.OrgIds, m.PersonIdentity).Scan(&baseFilter).Error
+ if err != nil {
+ logger.Warnf(err.Error())
+ }
+ if len(baseFilter) == 0 {
+ return fmt.Errorf("no results found that match the identity condition %+v", m.Task.Rules)
+ }
+
+ logger.Debugf("task %s match identity result %d", m.Task.Name, len(baseFilter))
+ for _, i := range baseFilter {
+ if _, ok := documentNumberMap[i.DocumentNumber]; !ok {
+ documentNumberList = append(documentNumberList, i.DocumentNumber)
+ tmp := i
+ documentNumberMap[i.DocumentNumber] = tmp
+ }
+ }
+ if m.PersonLabel != "" {
+ labels := strings.Split(m.PersonLabel, ",")
+ err := db.GetDB().Raw(`
+ SELECT
+ p.id
+ FROM
+ person AS p
+ JOIN person_label AS l ON p.id = l.person_id
+ WHERE
+ p.id IN ?
+ AND l.label_id IN ?
+ `, documentNumberList, labels).Scan(&labelFilter).Error
+ if err != nil {
+ logger.Warnf(err.Error())
+ }
+
+ if len(labelFilter) == 0 {
+ return fmt.Errorf("no results found that match the label condition %s", m.PersonLabel)
+ }
+
+ logger.Debugf("task %s match label result %d", m.Task.Name, len(labelFilter))
+ documentNumberList = []string{}
+ for _, i := range labelFilter {
+ documentNumberList = append(documentNumberList, i.Id)
+ }
+ }
+
+ if len(m.KeyTypes) > 0 {
+ //keyTypes := strings.Split(m.KeyPersonType, ",")
+ err := db.GetDB().Raw(`
+ SELECT
+ p.id,
+ k.person_type
+ FROM
+ person AS p
+ JOIN key_person AS k ON k.id_card = p.id_card
+ WHERE
+ p.id IN ?
+ AND k.person_type IN ?
+ `, documentNumberList, m.KeyTypes).Scan(&keyFilter).Error
+ if err != nil {
+ logger.Warnf(err.Error())
+ }
+ if len(keyFilter) == 0 {
+ return fmt.Errorf("no results found that match the key condition %v------ %s", documentNumberList, m.KeyTypes)
+ }
+ logger.Debugf("task %s match key person result %d", m.Task.Name, len(keyFilter))
+ }
+ //logger.Debugf("task %s last result %d", m.Task.Name, len(lastFilter))
+ documentNumberList = []string{}
+ for _, i := range keyFilter {
+ //logger.Debugf("keyFilter %s ------------- %s", i.Id, i.PersonType)
+ documentNumberList = append(documentNumberList, i.Id)
+ person := documentNumberMap[i.Id]
+ person.PersonType = i.PersonType
+ documentNumberMap[i.Id] = person
+ }
+ //logger.Debugf("document_number_map %s last result %s", m.Task.Name, document_number_map)
+ records, err := queryEsLocation(db.GetEsClient(), m, documentNumberList)
+ if err != nil {
+ return err
+ }
+ logger.Debugf("records %s last result %d", m.Task.Name, len(records))
+ domains, err := domainToLocation(records)
+ if err != nil {
+ return err
+ }
+ var tagTypes []string
+ //var lastAppearanceTime int64
+ for _, record := range records {
+ tagTypes = []string{}
+ for _, personId := range record.DocumentNumbers {
+ logger.Debugf("document_number_map[personId] %+v", documentNumberMap[personId])
+ tagTypes = append(tagTypes, documentNumberMap[personId].PersonType)
+ //lastAppearanceTime = document_number_map[personId].LastAppearanceTime
+ }
+ logger.Debugf("tagTypes %s ---------------%v--------", record.PicDate, record.DocumentNumbers)
+ _, typeNames, err := service.GetPersonTypeNameByTypes(tagTypes)
+ if err != nil {
+ return err
+ }
+ event := strings.Join(typeNames, ",")
+ result := &db.ModelTaskResults{
+ Title: m.Task.Name,
+ Event: m.eventFormat(event, record.AppearCount),
+ ModelID: m.Task.ModelID,
+ ModelTaskID: m.Task.ID,
+ CommunityId: record.CommunityId,
+ 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(int64(record.PicDate/1000), 0).Format("2006-01-02 15:04:05"),
+ PicDate: record.PicDate,
+ FirstPersonID: record.DocumentNumbers[0],
+ }
+ results = append(results, result)
+ }
logger.Debugf("task %s last filter result %d", m.Task.Name, len(results))
- return service.SaveTaskResults(results)
+ return service.SaveLocationTaskResults(results)
}
func (m *LocationModel) KeepAlive() error {
@@ -83,6 +280,243 @@
return nil
}
-func (m *LocationModel) eventFormat() string {
- return ""
+func (m *LocationModel) eventFormat(event string, AppearCount int) string {
+ return fmt.Sprintf("%s浜哄憳杩涘嚭%d娆�", event, AppearCount)
+}
+
+func queryEsLocation(esClient *elasticsearch.Client, locationModel *LocationModel, documentNumbers []string) ([]*LocationRecord, error) {
+ var buf bytes.Buffer
+ nowTime := time.Now()
+ startTime := nowTime.Add(-time.Duration(locationModel.Duration) * 24 * time.Hour)
+
+ // 鏋勫缓杩囨护鏉′欢
+ var filters []map[string]interface{}
+ documentNumberFilter := map[string]interface{}{
+ "terms": map[string]interface{}{
+ "documentNumber": documentNumbers,
+ },
+ }
+ filters = append(filters, documentNumberFilter)
+
+ if len(locationModel.OrgIds) > 0 || len(locationModel.AreaIds) > 0 {
+ // 鑾峰彇鏁版嵁鏉冮檺杩囨护鏉′欢
+ authFilters := GetDomainFilters(locationModel.OrgIds, locationModel.AreaIds)
+ filters = append(filters, authFilters...)
+ }
+
+ // 鍦板潃杩囨护
+ if locationModel.Building != "" || locationModel.Floor != "" {
+ var addrParams map[string]interface{}
+ if locationModel.Floor != "" {
+ addrParams = map[string]interface{}{"bool": map[string]interface{}{
+ "must": []interface{}{
+ map[string]interface{}{
+ "term": map[string]interface{}{
+ "cameraLocation.building": locationModel.Building,
+ }},
+ map[string]interface{}{
+ "term": map[string]interface{}{
+ "cameraLocation.floor": locationModel.Floor,
+ }},
+ },
+ }}
+ } else if locationModel.Building != "" {
+ addrParams = map[string]interface{}{
+ "term": map[string]interface{}{
+ "cameraLocation.building": locationModel.Building,
+ }}
+ }
+ filters = append(filters, addrParams)
+ }
+ 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())
+
+ queryTimes = append(queryTimes, map[string]interface{}{
+ "range": map[string]interface{}{
+ "picDate": map[string]interface{}{
+ "gte": start.Format(time.DateTime), // "2006-01-02 15:04:05d"
+ "lte": end.Format(time.DateTime),
+ },
+ },
+ })
+ }
+ 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{}{
+ "bool": map[string]interface{}{
+ "filter": filters,
+ },
+ },
+ "aggs": map[string]interface{}{
+ "orgs": map[string]interface{}{ // 鍏堣仛鍚坥rgId
+ "terms": map[string]interface{}{
+ "field": "orgId", // 鑱氬悎orgId
+ "size": 10000,
+ },
+ "aggs": map[string]interface{}{
+ "community": map[string]interface{}{ // 鍦╫rgId鑱氬悎涓嬭仛鍚坈ommunityId
+ "terms": map[string]interface{}{
+ "field": "communityId", // 鑱氬悎communityId
+ "size": 10000,
+ },
+ "aggs": map[string]interface{}{
+ "location": map[string]interface{}{ // 鍦╟ommunityId涓嬭仛鍚坆uilding
+ "terms": map[string]interface{}{
+ "field": "cameraLocation.building", // 鑱氬悎妤兼爧
+ "size": 10000,
+ },
+ "aggs": map[string]interface{}{
+ "floor": map[string]interface{}{ // 鍦╞uilding涓嬭仛鍚坒loor
+ "terms": map[string]interface{}{
+ "field": "cameraLocation.floor", // 鑱氬悎妤煎眰
+ "size": 10000,
+ },
+ "aggs": map[string]interface{}{
+ "filter_floor": map[string]interface{}{
+ "bucket_selector": map[string]interface{}{
+ "buckets_path": map[string]interface{}{
+ "eventCount": "_count",
+ },
+ "script": map[string]interface{}{
+ "source": "params.eventCount >= params.threshold",
+ "params": map[string]interface{}{
+ "threshold": locationModel.Appearances,
+ },
+ },
+ },
+ },
+ "document_numbers": map[string]interface{}{ // 鏂板鎸� documentNumber 鑱氬悎
+ "terms": map[string]interface{}{
+ "field": "documentNumber",
+ "size": 10000,
+ },
+ },
+ "picDate": map[string]interface{}{ // 鏂板鎸� picDate 鑱氬悎
+ "date_histogram": map[string]interface{}{
+ "field": "picDate",
+ "interval": "1s", // 鎸夊ぉ鑱氬悎
+ //"time_zone": "+08:00", // 璁剧疆鏃跺尯
+ },
+ },
+ },
+ },
+ },
+ },
+ },
+ },
+ },
+ },
+ },
+ "size": 0,
+ }
+
+ if err := json.NewEncoder(&buf).Encode(query); err != nil {
+ return nil, fmt.Errorf("error encoding query: %s", err)
+ }
+
+ res, err := esClient.Search(
+ esClient.Search.WithContext(context.Background()),
+ esClient.Search.WithIndex(config.EsInfo.EsIndex.AiOcean.IndexName),
+ esClient.Search.WithDocumentType(config.EsInfo.EsIndex.AiOcean.IndexType),
+ esClient.Search.WithBody(&buf),
+ esClient.Search.WithTrackTotalHits(true),
+ esClient.Search.WithPretty(),
+ )
+ if err != nil {
+ return nil, fmt.Errorf("error getting response: %s", err)
+ }
+ defer res.Body.Close()
+
+ // Check for a successful status code (2xx range)
+ if res.IsError() {
+ return nil, fmt.Errorf("error getting response: %s", res.String())
+ }
+
+ var result map[string]interface{}
+ if err := json.NewDecoder(res.Body).Decode(&result); err != nil {
+ return nil, fmt.Errorf("error parsing response body: %s", err)
+ }
+ //logger.Debugf("result--------------------------------------- %s", result)
+ // 瑙ｆ瀽鑱氬悎缁撴灉
+ var records []*LocationRecord
+ 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)
+ // 瑙ｆ瀽鎸塩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)
+ // 瑙ｆ瀽鎸塨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)
+ // 瑙ｆ瀽鎸塮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))
+ }
+ }
+ var PicDate string
+ if docNumBuckets, ok := floorBucket.(map[string]interface{})["picDate"].(map[string]interface{})["buckets"].([]interface{}); ok {
+ for _, docNumBucket := range docNumBuckets {
+ PicDate = docNumBucket.(map[string]interface{})["key_as_string"].(string)
+ }
+ }
+ logger.Debugf("floor, persons--,appearCount------------------------------------- %s-- %s --", floor, persons, appearCount)
+ record := &LocationRecord{
+ PicDate: PicDate,
+ DocumentNumbers: persons,
+ CommunityId: communityId,
+ Building: building,
+ Floor: floor,
+ OrgId: orgId,
+ AppearCount: int(appearCount),
+ }
+
+ records = append(records, record)
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+ }
+
+ return records, nil
+}
+
+func domainToLocation(records []*LocationRecord) (map[string]*db.DomainUnit, error) {
+ if len(records) == 0 {
+ return nil, nil
+ }
+ domainIds := set.NewStringSet()
+ for _, record := range records {
+ domainIds.Add(record.CommunityId)
+ }
+ domains, err := service.GetUnitsMapByIds(domainIds.Elements())
+ if err != nil {
+ return nil, err
+ }
+ return domains, nil
}
--
Gitblit v1.8.0