milvus查询record记录接口修改，查询milvus暂存数据

qixiaoning

2025-12-01 5ee5331527a19cb477cfebb8643c606fa2e42797

 camera-service/controllers/camera.go

@@ -7,6 +7,7 @@
   "fmt"
   "image"
   "path/filepath"
   "time"

   // "fmt"
   // "io/ioutil"
@@ -85,7 +86,39 @@
   }
   logger.Debug("update sensors:", camVo.Sensors)
   sv := service.CameraService{}
   ok, cam := sv.GetById(camVo.Id)
   if b, err := sv.CameraUpdate(camVo); b {
      //更新分辨率
      if ok {
         //
         logger.Debug("DoPostRequest:", camVo)
         if cam.IsRunning && (cam.ResolutionWidth != camVo.ResolutionWidth && cam.ResolutionHeight != camVo.ResolutionHeight) {
            header := map[string]string{}
            paramBody := map[string]interface{}{
               "isOpen":    1,
               "videoId":   camVo.Id,
               "videoType": 0,
            }
            res, err := util.DoPostRequest("http://127.0.0.1:8088/v1/task/openAnalysis", util.CONTENT_TYPE_JSON, paramBody, nil, header, time.Second*10)
            if err != nil {
               logger.Debug("DoPostRequest err:", err)
            }
            logger.Debug("DoPostRequest:", res)
         }
         //更新状态
         if cam.IsRunning {
            header := map[string]string{}
            paramBody := map[string]interface{}{
               "state": 1,
            }
            res, err := util.DoPostRequest("http://127.0.0.1:8088/v1/task/updateAllStatus?state=1", util.CONTENT_TYPE_JSON, paramBody, nil, header, time.Second*10)
            if err != nil {
               logger.Debug("DoPostRequest err:", err)
            }
            logger.Debug("DoPostRequest:", res)
         }
      }

      cc.addDbChangeMsg(h.Bk, protomsg.TableChanged_T_Camera, camVo.Id, protomsg.DbAction_Update, "")
      _, _ = service2.DoBusReq("/data/api-v/saas/syncCamera", config.Server.AnalyServerId, aiot.RequestMethod_Post, aiot.RequestContentType_ApplicationJson, map[string]interface{}{})
      return &bhomeclient.Reply{Success: true, Data: camVo.Copy2Camera()}
@@ -109,10 +142,26 @@
   if cid == "" {
      return &bhomeclient.Reply{Msg: "参数有误，摄像机id不能为空"}
   }

   //判断是否有配置任务
   // total, _ := models.GetTaskTotal(cid)
   // if total > 0 {
   //    return &bhomeclient.Reply{Success: false, Msg: "请先删除配置的任务"}
   // }

   rows, _ := cam.Delete(cid)
   if rows == 0 {
      return &bhomeclient.Reply{Success: true, Msg: "记录不存在"}
   }

   //删除任务
   items, _ := models.GetTaskTotal(cid)
   if len(items) > 0 {
      for _, item := range items {
         cam.DelLink(item.ID)
      }
   }

   pMsg := protomsg.DbChangeMessage{ //需要重新计算算力占用情况
      Table:  protomsg.TableChanged_T_Camera,
      Action: protomsg.DbAction_Delete,
@@ -166,10 +215,14 @@
   }

   m["resolutions"] = []models.Resolution{dRe0, dRe1, dRe2}

   //获取督查任务
   tasks := models.GetTasks()
   //处理摄像机的状态
   var crApi bhomedbapi.CameraRuleApi
   //var crApi bhomedbapi.CameraRuleApi
   if cam.RunType == models.TYPE_RUNTYPE_POLL || cam.RunType == models.TYPE_RUNTYPE_REALTIME {
      if crApi.ExistRunningTask(cam.Id) {
      //if crApi.ExistRunningTask(cam.Id) {
      if taskInfo, ok := tasks[cam.Id]; ok && taskInfo != nil && len(taskInfo) > 0 {
         if cam.IsRunning {
            m["status"] = models.Camera_Status_Doing
         } else {
@@ -1006,6 +1059,45 @@
}

/**
 *  删除记录
 */
func (cc CameraController) DelRecord(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
   filter := " id > 0 "
   info := models.TimeRange{}
   if err := c.BindJSON(&info); err != nil {
      return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
   }
   startTime := info.StartTime
   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)
      if err != nil {
         return &bhomeclient.Reply{Success: false, Msg: "删除失败" + err.Error(), Data: nil}
      } else {
         return &bhomeclient.Reply{Success: true, Msg: "删除成功", Data: deleteRecord}
      }
   } 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
}

/**
 *  获取抓拍记录
 */
func (cc CameraController) GetRecord(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
@@ -1019,6 +1111,7 @@
   pageSize := info.PageSize
   idsStr := info.Ids
   TaskIds := info.TaskIds
   TaskName := info.TaskName
   videoIdStr := info.VideoIds
   Warning := info.Warning

@@ -1045,7 +1138,7 @@
         }
      }
      //根据id查询视频id
      var vids []int
      var vids []int64
      cameras, _ := models.GetCameraIds(videoIds)
      for _, camera := range cameras {
         if len(camera.Id) > 0 {
@@ -1054,7 +1147,7 @@
      }
      idStr := make([]string, len(vids))
      for i, id := range vids {
         idStr[i] = strconv.Itoa(id)
         idStr[i] = strconv.FormatInt(id, 10)
      }
      filter += fmt.Sprintf(" and %s in [%s]", "video_point_id", strings.Join(idStr, ","))
   }
@@ -1062,6 +1155,19 @@
   //根据任务查询
   if len(TaskIds) > 0 {
      filter += fmt.Sprintf(" and array_contains(task_id, %d)", TaskIds[0])
   }

   //根据任务名称
   if len(TaskName) > 0 {
      quotedNames := make([]string, len(TaskName))
      for i, name := range TaskName {
         quotedNames[i] = " task_name like '%%" + name + "%%' " // 每个元素加单引号
      }
      filter += " and (" + strings.Join(quotedNames, "or") + ")"
   } else {
      if Warning > 0 {
         return &bhomeclient.Reply{Success: true, Msg: "获取成功", Data: nil}
      }
   }

   //是否预警
@@ -1107,6 +1213,7 @@
         IsWarning:       v.IsWarning,
         DetectNum:       v.DetectNum,
         TaskName:        v.TaskName,
         WarnTaskId:      v.WarnTaskId,
         ZhDescClass:     v.ZhDescClass,
         EventLevelName:  v.EventLevelName,
         KnowledgeId:     v.KnowledgeId,
@@ -1121,16 +1228,77 @@
         Content:         v.CameraName,
         PicDate:         v.PicDate,
         LikeDate:        v.LikeDate,
         VideoName:       v.CameraName,
         CameraAddr:      v.CameraName,
         CameraName:      v.CameraName,
         DynamicLevel:    v.DynamicLevel,
      }
      var taskId []int64

      //任务id重新赋值
      if len(v.WarnTaskId) > 0 {
         v.TaskId = []int64{}
         taskId = []int64{}
         arr := strings.Split(v.WarnTaskId, ";")
         for _, temp := range arr {
            if len(temp) > 0 {

               tempId, _ := strconv.ParseInt(strings.TrimSpace(temp), 10, 64)

               taskId = append(taskId, tempId)
            }
         }
         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
      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 + ","
            }
            result1 += item.TaskName + ","
         }
         if len(taskId) > 0 {
            temp.TaskName = result
         } else {
            temp.TaskName = result1
         }
      }

      //任务
      temp.TaskName = strings.TrimSuffix(temp.TaskName, ",")

      //视频内容
      items4, _ := models.GetVideoById(v.VideoPointId)
      temp.VideoName = items4.Name
      temp.CameraAddr = items4.Addr
      // items4, _ := models.GetVideoById(v.VideoPointId)
      // temp.VideoName = items4.Name
      // temp.CameraAddr = items4.Addr

      //检查内容
      items2, _ := models.GetCheckByIds(v.DetectId)
@@ -1141,8 +1309,52 @@
      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
      var result22 string
      if len(taskId) > 0 {
         for _, item5 := range items5 {
            if containsID(taskId, item5.TaskId) {
               result2 += item5.EventLevelName + ","
            }
            result22 += item5.EventLevelName + ","
         }
         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
            }
         }
      }

      //知识库
      items6, _ := models.GetKnowledgeDocumentByIds(v.KnowledgeId)
@@ -1173,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
}

/**
 *获取督查任务
 */
@@ -1193,3 +1415,14 @@
   return &bhomeclient.Reply{Success: true, Msg: "获取成功", Data: tasks}

}

// 获取所有
func (cc CameraController) GetAllTasks(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
   //绑定json和结构体

   var tasks []models.TaskAggregate
   tasks, _ = models.GetAllTask()

   return &bhomeclient.Reply{Success: true, Msg: "获取成功", Data: tasks}

}