From 5ee5331527a19cb477cfebb8643c606fa2e42797 Mon Sep 17 00:00:00 2001
From: qixiaoning <jony.kee@outlook.com>
Date: 星期一, 01 十二月 2025 09:47:17 +0800
Subject: [PATCH] milvus查询record记录接口修改,查询milvus暂存数据

---
 camera-service/controllers/camera.go |  130 ++++++++++++++++++++++++++++++++++++-------
 1 files changed, 108 insertions(+), 22 deletions(-)

diff --git a/camera-service/controllers/camera.go b/camera-service/controllers/camera.go
index 2029bfc..7e54e1f 100644
--- a/camera-service/controllers/camera.go
+++ b/camera-service/controllers/camera.go
@@ -1068,8 +1068,12 @@
 		return &bhomeclient.Reply{Success: false, Msg: "鍙傛暟鏈夎"}
 	}
 	startTime := info.StartTime
-	endTime := info.EndTime
-	if len(startTime) > 0 && len(endTime) > 0 {
+	if len(startTime) > 0 && len(info.EndTime) > 0 {
+		endTime, err1 := addDaySimple(info.EndTime)
+		if err1 != nil {
+			endTime = info.EndTime
+		}
+
 		filter += fmt.Sprintf(" and detect_time > '%s'", startTime)
 		filter += fmt.Sprintf(" and detect_time < '%s'", endTime)
 		deleteRecord, err := models.DeleteRecord("smartobject", filter)
@@ -1081,6 +1085,16 @@
 	} else {
 		return &bhomeclient.Reply{Success: false, Msg: "鍒犻櫎澶辫触", Data: nil}
 	}
+}
+
+func addDaySimple(dateStr string) (string, error) {
+	// 鍋囪鏍煎紡涓� "2006-01-02"
+	t, err := time.Parse("2006-01-02", dateStr)
+	if err != nil {
+		return "", err
+	}
+	t = t.AddDate(0, 0, 1)
+	return t.Format("2006-01-02"), nil
 }
 
 /**
@@ -1199,7 +1213,7 @@
 			IsWarning:       v.IsWarning,
 			DetectNum:       v.DetectNum,
 			TaskName:        v.TaskName,
-			WarnTaskName:    v.WarnTaskName,
+			WarnTaskId:      v.WarnTaskId,
 			ZhDescClass:     v.ZhDescClass,
 			EventLevelName:  v.EventLevelName,
 			KnowledgeId:     v.KnowledgeId,
@@ -1217,14 +1231,15 @@
 			VideoName:       v.CameraName,
 			CameraAddr:      v.CameraName,
 			CameraName:      v.CameraName,
+			DynamicLevel:    v.DynamicLevel,
 		}
 		var taskId []int64
 
 		//浠诲姟id閲嶆柊璧嬪��
-		if len(v.WarnTaskName) > 0 {
+		if len(v.WarnTaskId) > 0 {
 			v.TaskId = []int64{}
 			taskId = []int64{}
-			arr := strings.Split(v.WarnTaskName, ";")
+			arr := strings.Split(v.WarnTaskId, ";")
 			for _, temp := range arr {
 				if len(temp) > 0 {
 
@@ -1235,21 +1250,50 @@
 			}
 			v.TaskId = taskId
 		}
-		items, _ := models.GetTaskByIds(v.TaskId)
+		// items, _ := models.GetTaskByIds(v.TaskId)
+		//浠诲姟鍚嶇О
+		// items, _ := db.GetTaskByIds(v.TaskId)
+		items := []models.TaskOption{}
+		//if strings.Contains(v.TaskName, ",") {
+		taskNames := strings.Split(v.TaskName, ",")
+		for _, name := range taskNames {
+			task := models.TaskOption{}
+			if len(name) > 0 {
+				if strings.Contains(name, "_") {
+					tempNames := strings.Split(name, "_")
+					if len(tempNames[0]) > 0 {
+						id, _ := strconv.ParseInt(tempNames[0], 10, 64)
+						task.TaskId = id
+					}
+					task.TaskName = tempNames[1]
+				}
+			}
+			items = append(items, task)
+		}
+		//}
+
 		temp.TaskNames = items
+
 		//浠诲姟鍚嶇О閲嶆柊璧嬪��
 		var result string
-		if len(taskId) > 0 && len(items) > 0 {
-			v.EventLevelId = []int64{}
-			for index, item := range items {
-				v.EventLevelId = append(v.EventLevelId, item.EventLevel)
-				if index > 0 {
-					result += ","
+		var result1 string
+		if len(items) > 0 {
+			for _, item := range items {
+				// v.EventLevelId = append(v.EventLevelId, item.EventLevel)
+				if containsID(taskId, item.TaskId) {
+					result += item.TaskName + ","
 				}
-				result += item.TaskName
+				result1 += item.TaskName + ","
 			}
-			temp.TaskName = result
+			if len(taskId) > 0 {
+				temp.TaskName = result
+			} else {
+				temp.TaskName = result1
+			}
 		}
+
+		//浠诲姟
+		temp.TaskName = strings.TrimSuffix(temp.TaskName, ",")
 
 		//瑙嗛鍐呭
 		// items4, _ := models.GetVideoById(v.VideoPointId)
@@ -1265,19 +1309,51 @@
 		temp.RuleNames = items3
 
 		//浜嬩欢
-		items5, _ := models.GetEventByIds(v.EventLevelId)
-		temp.EventLevels = items5
+		items5 := []models.TaskOption{}
+		//if strings.Contains(v.EventLevelName, ",") {
+		eventNames := strings.Split(v.EventLevelName, ",")
+		for _, name := range eventNames {
+			task := models.TaskOption{}
+			if len(name) > 0 {
+				if strings.Contains(name, "_") {
+					tempNames := strings.Split(name, "_")
+					if len(tempNames[0]) > 0 {
+						id, _ := strconv.ParseInt(tempNames[0], 10, 64)
+						task.TaskId = id
+					}
+					task.EventLevelName = tempNames[1]
+				}
+			}
+			items5 = append(items5, task)
+		}
+		//}
+
+		tempEventName := temp.DynamicLevel
 
 		//浜嬩欢绛夌骇閲嶆柊璧嬪��
 		var result2 string
-		if len(taskId) > 0 && len(items5) > 0 {
-			for i, item5 := range items5 {
-				if i > 0 {
-					result2 += ","
+		var result22 string
+		if len(taskId) > 0 {
+			for _, item5 := range items5 {
+				if containsID(taskId, item5.TaskId) {
+					result2 += item5.EventLevelName + ","
 				}
-				result2 += item5.DictValue
+				result22 += item5.EventLevelName + ","
 			}
-			temp.EventLevelName = result2
+			if len(taskId) > 0 {
+				temp.EventLevelName = result2
+			} else {
+				temp.EventLevelName = result22
+			}
+			temp.EventLevelName = strings.TrimSuffix(temp.EventLevelName, ",")
+
+			if len(tempEventName) > 0 {
+				if len(temp.EventLevelName) > 0 {
+					temp.EventLevelName = tempEventName + "," + temp.EventLevelName
+				} else {
+					temp.EventLevelName = tempEventName
+				}
+			}
 		}
 
 		//鐭ヨ瘑搴�
@@ -1309,6 +1385,16 @@
 	return &bhomeclient.Reply{Success: true, Msg: "鑾峰彇鎴愬姛", Data: pageRes}
 }
 
+// 鍖呭惈id
+func containsID(taskId []int64, id int64) bool {
+	for _, v := range taskId {
+		if v == id {
+			return true
+		}
+	}
+	return false
+}
+
 /**
  *鑾峰彇鐫f煡浠诲姟
  */

--
Gitblit v1.8.0