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 | 88 +++++++++++++++++++++----------------------
1 files changed, 43 insertions(+), 45 deletions(-)
diff --git a/models/locationAnalysis.go b/models/locationAnalysis.go
index 62ff299..a1c352e 100644
--- a/models/locationAnalysis.go
+++ b/models/locationAnalysis.go
@@ -54,13 +54,22 @@
m.AreaIds = areaIds
m.Building = task.Building
m.AlarmType = task.AlarmType
- m.PersonIdentity = []string{"stranger", "visitor", "resident"} //task.IdentityType
- m.KeyTypes = []string{"1", "2", "3", "4", "5"} //task.IdentityType
+ m.KeyTypes = []string{"1", "2", "3", "4", "5"} //task.IdentityType
m.KeyPersonType = task.PersonType
m.PersonLabel = task.PersonLabel
if task.IdentityType != "" {
- m.PersonIdentity = strings.Split(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, ",")
}
@@ -112,33 +121,22 @@
}
type LocationPersonInfo struct {
- Id string `json:"id"`
- DocumentNumber string `json:"document_number"`
- PersonType string `json:"person_type"`
- //CommunityId string `json:"community_id"`
- //OrgId string `json:"org_id"`
- //PersonName string `json:"person_name"`
- //IdCard string `json:"id_card"`
- LastAppearanceTime int64 `json:"last_appearance_time"`
- //LastDirection string `json:"last_direction"`
- //LastLocation string `json:"last_location"`
+ Id string `json:"id"`
+ DocumentNumber string `json:"document_number"`
+ PersonType string `json:"person_type"`
+ LastAppearanceTime int64 `json:"last_appearance_time"`
}
-
-//var (
-// processed sync.Map // 瀛樺偍宸插鐞嗚褰�
-// cleanupThreshold = time.Now().Add(-100 * time.Hour) // 瀹氫箟涓�涓椂闂寸獥鍙ｏ紝鍋囪鍙繚瀛樻渶杩�100灏忔椂鐨勮褰�
-//)
func (m *LocationModel) Run() error {
// 鏍规嵁閰嶇疆鐨勬椂闂存澶╂暟, 姣忓ぉ鐨勬椂闂磋寖鍥村唴锛� 閲嶇偣浜哄憳绫诲瀷鎴栬�呯壒瀹氭爣绛句汉鍛樺嚭鐜扮殑妤煎眰娆℃暟瓒呰繃闃堝��
results := make([]*db.ModelTaskResults, 0)
var baseFilter, labelFilter, keyFilter []LocationPersonInfo
- var document_number_map = make(map[string]LocationPersonInfo)
- var document_number_list []string
- community_ids := m.AreaIds
- if len(community_ids) == 0 {
- community_ids = m.OrgIds
+ var documentNumberMap = make(map[string]LocationPersonInfo)
+ var documentNumberList []string
+ communityIds := m.AreaIds
+ if len(communityIds) == 0 {
+ communityIds = m.OrgIds
}
err := db.GetDB().Raw(`
SELECT
@@ -149,24 +147,23 @@
snapshot_count_summary AS s
JOIN person AS p ON p.id = s.document_number
WHERE
- p.id_card != ""
- AND (s.community_id IN ? OR s.org_id IN ?)
+ (s.community_id IN ? OR s.org_id IN ?)
AND s.status IN ?
- `, community_ids, m.OrgIds, m.PersonIdentity).Scan(&baseFilter).Error
+ `, 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 age condition %s - %s - %s ", m.AreaIds, m.OrgIds, m.PersonIdentity)
+ return fmt.Errorf("no results found that match the identity condition %+v", m.Task.Rules)
}
- logger.Debugf("task %s match age result %d", m.Task.Name, len(baseFilter))
+ logger.Debugf("task %s match identity result %d", m.Task.Name, len(baseFilter))
for _, i := range baseFilter {
- if _, ok := document_number_map[i.DocumentNumber]; !ok {
- document_number_list = append(document_number_list, i.DocumentNumber)
+ if _, ok := documentNumberMap[i.DocumentNumber]; !ok {
+ documentNumberList = append(documentNumberList, i.DocumentNumber)
tmp := i
- document_number_map[i.DocumentNumber] = tmp
+ documentNumberMap[i.DocumentNumber] = tmp
}
}
if m.PersonLabel != "" {
@@ -180,7 +177,7 @@
WHERE
p.id IN ?
AND l.label_id IN ?
- `, document_number_list, labels).Scan(&labelFilter).Error
+ `, documentNumberList, labels).Scan(&labelFilter).Error
if err != nil {
logger.Warnf(err.Error())
}
@@ -190,11 +187,12 @@
}
logger.Debugf("task %s match label result %d", m.Task.Name, len(labelFilter))
- document_number_list = []string{}
+ documentNumberList = []string{}
for _, i := range labelFilter {
- document_number_list = append(document_number_list, i.Id)
+ documentNumberList = append(documentNumberList, i.Id)
}
}
+
if len(m.KeyTypes) > 0 {
//keyTypes := strings.Split(m.KeyPersonType, ",")
err := db.GetDB().Raw(`
@@ -207,30 +205,30 @@
WHERE
p.id IN ?
AND k.person_type IN ?
- `, document_number_list, m.KeyTypes).Scan(&keyFilter).Error
+ `, 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%s------ %s", document_number_list, m.KeyTypes)
+ 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))
- document_number_list = []string{}
+ documentNumberList = []string{}
for _, i := range keyFilter {
//logger.Debugf("keyFilter %s ------------- %s", i.Id, i.PersonType)
- document_number_list = append(document_number_list, i.Id)
- person := document_number_map[i.Id]
+ documentNumberList = append(documentNumberList, i.Id)
+ person := documentNumberMap[i.Id]
person.PersonType = i.PersonType
- document_number_map[i.Id] = person
+ 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, document_number_list)
+ records, err := queryEsLocation(db.GetEsClient(), m, documentNumberList)
if err != nil {
return err
}
- logger.Debugf("records %s last result %+v", m.Task.Name, records)
+ logger.Debugf("records %s last result %d", m.Task.Name, len(records))
domains, err := domainToLocation(records)
if err != nil {
return err
@@ -240,11 +238,11 @@
for _, record := range records {
tagTypes = []string{}
for _, personId := range record.DocumentNumbers {
- logger.Debugf("document_number_map[personId] %s", document_number_map[personId])
- tagTypes = append(tagTypes, document_number_map[personId].PersonType)
+ 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 ---------------%s--------", record.PicDate, record.DocumentNumbers)
+ logger.Debugf("tagTypes %s ---------------%v--------", record.PicDate, record.DocumentNumbers)
_, typeNames, err := service.GetPersonTypeNameByTypes(tagTypes)
if err != nil {
return err
--
Gitblit v1.8.0