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

qixiaoning

2025-12-01 5ee5331527a19cb477cfebb8643c606fa2e42797

 camera-common/models/Record.go

@@ -50,7 +50,7 @@
   IsWarning          int64                     `json:"is_warning"`

   ZhDescClass        string                    `json:"zh_desc_class"`

   TaskName           string                    `json:"task_name"`

   WarnTaskName       string                    `json:"warn_task_name"`

   WarnTaskId         string                    `json:"warn_task_id"`

   EventLevelName     string                    `json:"event_level_name"`

   DetectNum          int64                     `json:"detect_num"`

   Suggestion         string                    `json:"suggestion"`

@@ -84,6 +84,7 @@
   TargetInfo         []TargetInfo              `json:"targetInfo"`

   FileId             string                    `json:"fileId"`     // 数据栈文件id

   DataSource         string                    `json:"dataSource"` // 数据来源:摄像机, 数据栈

   DynamicLevel       string                    `json:"dynamic_level"`

}



type Point struct {

@@ -131,9 +132,10 @@
}



type TaskOption struct {

   TaskId     int64  `db:"task_id" json:"taskId"`

   TaskName   string `db:"task_name" json:"taskName"`

   EventLevel int64  `db:"event_level" json:"eventLevel"`

   TaskId         int64  `db:"task_id" json:"taskId"`

   TaskName       string `db:"task_name" json:"taskName"`

   EventLevel     int64  `db:"event_level" json:"eventLevel"`

   EventLevelName string `db:"event_level_name" json:"eventLevelName"`

}



type CheckOption struct {

@@ -220,6 +222,10 @@
      // client.WithOffset(offset),

   )



   if err != nil {



   }



   // 2. 提取ID和文件路径

   var ids []int64

   var imgPaths []string

@@ -256,38 +262,29 @@
               VideoPath := VideoPathColumn[i]

               videoPaths = append(videoPaths, VideoPath)

            }



         }

      }

   }



   // 图片删除文件

   deletedFiles := 0

   if len(ids) > 0 {

      idField := entity.NewColumnInt64("id", ids)

      err = do.client.client.DeleteByPks(ctx, collectionName, "", idField)

      if err != nil {

         return 0, fmt.Errorf("failed to delete from Milvus: %v", err)

      }

      err = do.client.client.DeleteByPks(ctx, "smartrag", "", idField)

      if err != nil {

         return 0, fmt.Errorf("failed to delete from Milvus: %v", err)

      }

      deletedFiles = len(ids)

   }



   // 图片删除文件

   deletedFiles := 0

   for _, path := range imgPaths {

      if err := do.deleteFile(path); err != nil {

         logger.Debug("failed to delete file " + path + " " + err.Error())

      } else {

         deletedFiles++

      }

   }

   //删除描述图片

   for _, path := range imgDesPaths {

      if err := do.deleteFile(path); err != nil {

         logger.Debug("failed to delete file " + path + " " + err.Error())

      } else {

         deletedFiles++

      }

   }



@@ -295,9 +292,13 @@
   for _, path := range videoPaths {

      if err := do.deleteFile(path); err != nil {

         logger.Debug("failed to delete file " + path + " " + err.Error())

      } else {

         deletedFiles++

      }

   }



   //删除rag

   err = do.client.client.Delete(ctx, "smartrag", "", filterExpr)

   if err != nil {

      logger.Debug("delete rag failed!")

   }



   return deletedFiles, err

@@ -418,7 +419,7 @@
         case "warn_task_id":

            WarnColumn := field.(*entity.ColumnVarChar).Data()

            if len(WarnColumn) > 0 {

               record.WarnTaskName = WarnColumn[i]

               record.WarnTaskId = WarnColumn[i]

            }

         case "event_level_name":

            EVColumn := field.(*entity.ColumnVarChar).Data()

@@ -503,15 +504,27 @@
   lists := convertResultToMap(result)

   layout := "2006-01-02 15:04:05.999999"



   var temp1 time.Time

   var temp2 time.Time

   // var temp1 time.Time

   // var temp2 time.Time



   //排序

   // //排序

   // sort.Slice(lists, func(i, j int) bool {

   //    //return lists[i].DetectTime > lists[j].:DetectTime

   //    temp1, _ = time.Parse(layout, lists[i].DetectTime)

   //    temp2, _ = time.Parse(layout, lists[j].DetectTime)

   //    return temp1.After(temp2)

   // })



   sort.Slice(lists, func(i, j int) bool {

      //return lists[i].DetectTime > lists[j].:DetectTime

      temp1, _ = time.Parse(layout, lists[i].DetectTime)

      temp2, _ = time.Parse(layout, lists[j].DetectTime)

      return temp1.After(temp2)

      t1, err1 := time.Parse(layout, lists[i].DetectTime)

      t2, err2 := time.Parse(layout, lists[j].DetectTime)



      // 错误处理：解析错误的时间排在后面

      if err1 != nil || err2 != nil {

         return err1 == nil && err2 != nil

      }



      return t1.After(t2)

   })



   items := Paginate(lists, int(pageNum), int(pageSize))

@@ -611,14 +624,25 @@


// 获取事件

func GetEventByIds(ids []int64) (items []DictOption, err error) {

   placeholders := make([]string, len(ids))

   args := make([]interface{}, len(ids))

   // 动态生成 UNION ALL 子查询

   unionAllParts := make([]string, len(ids))

   for i, id := range ids {

      placeholders[i] = "?"

      args[i] = id

      unionAllParts[i] = fmt.Sprintf("SELECT %d AS dict_id", id)

   }

   sqlStr := `select dict_id,dict_name,dict_value from mal_dict_type where dict_id in (` + strings.Join(placeholders, ",") + `)`

   if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil {



   // 拼接 UNION ALL 子查询

   unionAllQuery := strings.Join(unionAllParts, " UNION ALL ")



   // 拼接最终 SQL 查询

   sqlStr := fmt.Sprintf(`

      SELECT t.dict_id, t.dict_name, t.dict_value

      FROM mal_dict_type t

      JOIN (

      %s

      ) AS input_list

      ON t.dict_id = input_list.dict_id;`, unionAllQuery)



   if err := db.Raw(sqlStr).Scan(&items).Error; err != nil {

      return nil, err

   }