| | |
|---|
| | | package models |
|---|
| | | |
|---|
| | | import ( |
|---|
| | | "context" |
|---|
| | | "fmt" |
|---|
| | | "math" |
|---|
| | | "sort" |
|---|
| | | "strconv" |
|---|
| | | "strings" |
|---|
| | | "time" |
|---|
| | | |
|---|
| | | "github.com/milvus-io/milvus-sdk-go/v2/entity" |
|---|
| | | ) |
|---|
| | | |
|---|
| | | // DataOperator 数据操作相关方法 |
|---|
| | | type DataOperator struct { |
|---|
| | | client *MilvusClient |
|---|
| | | } |
|---|
| | | |
|---|
| | | // NewDataOperator 创建数据操作器 |
|---|
| | | func NewDataOperator(client *MilvusClient) *DataOperator { |
|---|
| | | return &DataOperator{client: client} |
|---|
| | | } |
|---|
| | | |
|---|
| | | type PicWidHei struct { |
|---|
| | | PicW int `json:"picW"` |
|---|
| | | PicH int `json:"picH"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 统计查询 |
|---|
| | | type StaticRecord struct { |
|---|
| | | Id string `json:"id"` |
|---|
| | | TaskId []int64 `json:"task_id"` |
|---|
| | | RuleId []int64 `json:"rule_id"` |
|---|
| | | EventLevelId []int64 `json:"event_level_id"` |
|---|
| | | VideoPointId int64 `json:"video_point_id"` |
|---|
| | | DetectId []int64 `json:"detect_id"` |
|---|
| | | VideoName string `json:"video_name"` |
|---|
| | | TaskNames []TaskOption `json:"task_names"` |
|---|
| | | CheckNames []CheckOption `json:"check_names"` |
|---|
| | | RuleNames []RuleOption `json:"rule_names"` |
|---|
| | | EventLevels []DictOption `json:"event_levels"` |
|---|
| | | KnowledgeDocuments []KnowledgeDocumentOption `json:"knowledge_documents"` |
|---|
| | | ImagePath string `json:"image_path"` |
|---|
| | | VideoPath string `json:"video_path"` |
|---|
| | | DetectTime string `json:"detect_time"` |
|---|
| | | IsWarning int64 `json:"is_warning"` |
|---|
| | | ZhDescClass string `json:"zh_desc_class"` |
|---|
| | | TaskName string `json:"task_name"` |
|---|
| | | EventLevelName string `json:"event_level_name"` |
|---|
| | | DetectNum int64 `json:"detect_num"` |
|---|
| | | Suggestion string `json:"suggestion"` |
|---|
| | | RiskDescription string `json:"risk_description"` |
|---|
| | | KnowledgeId []int64 `json:"knowledge_id"` |
|---|
| | | IsDesc int64 `json:"is_desc"` |
|---|
| | | CameraId string `json:"cameraId"` |
|---|
| | | CameraName string `json:"cameraName"` |
|---|
| | | CameraAddr string `json:"cameraAddr"` |
|---|
| | | PicDate string `json:"picDate"` |
|---|
| | | PicId int64 `json:"picId"` |
|---|
| | | PicMaxUrl []string `json:"picMaxUrl"` |
|---|
| | | PicSrcUrl []string `json:"picSrcUrl"` |
|---|
| | | PicWH PicWidHei `json:"picWH"` |
|---|
| | | SdkName string `json:"sdkName"` |
|---|
| | | Content string `json:"content"` |
|---|
| | | AlarmRules []AlarmRule `json:"alarmRules"` |
|---|
| | | LikeDate string `json:"likeDate"` |
|---|
| | | ShowLabels string `json:"showLabels"` |
|---|
| | | OtherLabels string `json:"otherLabels"` |
|---|
| | | VideoUrl string `json:"videoUrl"` |
|---|
| | | AnalyServerId string `json:"analyServerId"` |
|---|
| | | AnalyServerName string `json:"analyServerName"` |
|---|
| | | AnalyServerIp string `json:"analyServerIp"` |
|---|
| | | ClusterId string `json:"clusterId"` |
|---|
| | | IsAlarm bool `json:"isAlarm"` |
|---|
| | | IsAckAlarm bool `json:"isAckAlarm"` |
|---|
| | | IsCollect bool `json:"isCollect"` |
|---|
| | | IsDelete bool `json:"isDelete"` |
|---|
| | | BaseInfo []*BaseCompareInfo `json:"baseInfo"` |
|---|
| | | TargetInfo []TargetInfo `json:"targetInfo"` |
|---|
| | | FileId string `json:"fileId"` // 数据栈文件id |
|---|
| | | DataSource string `json:"dataSource"` // 数据来源:摄像机, 数据栈 |
|---|
| | | } |
|---|
| | | |
|---|
| | | type Point struct { |
|---|
| | | X float64 `json:"x"` |
|---|
| | | Y float64 `json:"y"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | type Points struct { |
|---|
| | | TopLeft Point `json:"topLeft"` |
|---|
| | | BottomRight Point `json:"bottomRight"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | type BaseCompareInfo struct { |
|---|
| | | TableId string `json:"tableId"` |
|---|
| | | TableName string `json:"tableName"` |
|---|
| | | BwType string `json:"bwType"` |
|---|
| | | CompareScore float64 `json:"compareScore"` |
|---|
| | | TargetId string `json:"targetId"` |
|---|
| | | TargetName string `json:"targetName"` |
|---|
| | | TargetPicUrl string `json:"targetPicUrl"` |
|---|
| | | MonitorLevel string `json:"monitorLevel"` |
|---|
| | | Content string `json:"content"` |
|---|
| | | DbLabel string `json:"labels"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | type TargetInfo struct { |
|---|
| | | TargetId string `json:"targetId"` |
|---|
| | | AreaId string `json:"areaId"` |
|---|
| | | AreaName string `json:"areaName"` |
|---|
| | | TargetScore float64 `json:"targetScore"` |
|---|
| | | TargetType string `json:"targetType"` |
|---|
| | | Feature string `json:"feature"` |
|---|
| | | PicSmUrl string `json:"picSmUrl"` |
|---|
| | | TargetLocation Points `json:"targetLocation"` |
|---|
| | | BelongsTargetID string `json:"belongsTargetId"` |
|---|
| | | Attribute string `json:"attribute"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | type AlarmRule struct { |
|---|
| | | AlarmLevel string `db:"alarmLevel" json:"alarmLevel"` |
|---|
| | | DefenceState bool `db:"defenceState" json:"defenceState"` |
|---|
| | | GroupId string `db:"groupId" json:"groupId"` |
|---|
| | | LinkInfo string `db:"linkInfo" json:"linkInfo"` |
|---|
| | | RuleText string `db:"ruleText" json:"ruleText"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | type TaskOption struct { |
|---|
| | | TaskId int64 `db:"task_id" json:"taskId"` |
|---|
| | | TaskName string `db:"task_name" json:"taskName"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | type CheckOption struct { |
|---|
| | | TaskId int64 `db:"task_id" json:"taskId,omitempty"` |
|---|
| | | CheckId int64 `db:"check_id" json:"checkId"` |
|---|
| | | FileName string `db:"file_name" json:"fileName"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | type RuleOption struct { |
|---|
| | | TaskId int64 `db:"task_id" json:"taskId,omitempty"` |
|---|
| | | RuleId int64 `db:"rule_id" json:"ruleId"` |
|---|
| | | FileName string `db:"file_name" json:"fileName"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 字典 |
|---|
| | | type DictOption struct { |
|---|
| | | DictId int64 `db:"dict_id" json:"dictId"` |
|---|
| | | DictName string `db:"dict_name" json:"dictName"` |
|---|
| | | DictValue string `db:"dict_value" json:"dictValue"` |
|---|
| | | DictType string `db:"dict_type" json:"dictType"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 知识库 |
|---|
| | | type KnowledgeDocumentOption struct { |
|---|
| | | Id string `db:"id" json:"id"` |
|---|
| | | Title string `db:"title" json:"title"` |
|---|
| | | KnowId int64 `db:"know_id" json:"know_id"` |
|---|
| | | FileUrl string `db:"title" json:"file_url"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | // Pagination 分页信息 |
|---|
| | | type Pagination struct { |
|---|
| | | Page int `json:"page"` // 当前页码 |
|---|
| | | PageSize int `json:"pageSize"` // 每页大小 |
|---|
| | | Total int `json:"total"` // 总记录数 |
|---|
| | | TotalPage int `json:"totalPage"` // 总页数 |
|---|
| | | } |
|---|
| | | |
|---|
| | | // PaginatedResult 分页结果 |
|---|
| | | type PaginatedResult struct { |
|---|
| | | Items []StaticRecord `json:"list"` |
|---|
| | | Pagination Pagination `json:"pagination"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | type PaginatedResult2 struct { |
|---|
| | | Items map[string]interface{} `json:"list"` |
|---|
| | | Pagination Pagination `json:"pagination"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 记录 |
|---|
| | | type RecordReq struct { |
|---|
| | | Ids []string `json:"ids"` |
|---|
| | | VideoIds []string `json:"treeNodes"` |
|---|
| | | TaskIds []int64 `json:"taskIds` |
|---|
| | | TaskName []string `json:"taskName"` |
|---|
| | | Warning int64 `json:"warning` |
|---|
| | | Page int64 `json:"page"` |
|---|
| | | PageSize int64 `json:"pageSize"` |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 根据分页取数据 |
|---|
| | | func GetWithPage(collectionName string, pageNum int64, pageSize int64, filter string) (*PaginatedResult, error) { |
|---|
| | | |
|---|
| | | if pageNum < 1 { |
|---|
| | | pageNum = 1 |
|---|
| | | } |
|---|
| | | if pageSize < 1 { |
|---|
| | | pageSize = 10 |
|---|
| | | } |
|---|
| | | |
|---|
| | | if dbClient == nil { |
|---|
| | | return nil, nil |
|---|
| | | } |
|---|
| | | |
|---|
| | | dataOp := &DataOperator{client: dbClient} |
|---|
| | | |
|---|
| | | fmt.Println(filter) |
|---|
| | | records, err := dataOp.queryWithPagination(collectionName, filter, pageNum, pageSize) |
|---|
| | | if err != nil { |
|---|
| | | return nil, err |
|---|
| | | } |
|---|
| | | |
|---|
| | | return records, nil |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 转换数据 |
|---|
| | | func convertResultToMap(result []entity.Column) []StaticRecord { |
|---|
| | | if len(result) == 0 { |
|---|
| | | return nil |
|---|
| | | } |
|---|
| | | count := result[0].Len() |
|---|
| | | var records []StaticRecord |
|---|
| | | for i := 0; i < count; i++ { |
|---|
| | | record := StaticRecord{} |
|---|
| | | for _, field := range result { |
|---|
| | | switch field.Name() { |
|---|
| | | case "id": |
|---|
| | | IdColumn := field.(*entity.ColumnInt64).Data() |
|---|
| | | if len(IdColumn) > 0 { |
|---|
| | | record.Id = strconv.FormatInt(IdColumn[i], 10) |
|---|
| | | } |
|---|
| | | case "task_id": |
|---|
| | | TaskIdColumn := field.(*entity.ColumnInt64Array).Data() |
|---|
| | | if len(TaskIdColumn) > 0 { |
|---|
| | | record.TaskId = TaskIdColumn[i] |
|---|
| | | } |
|---|
| | | case "rule_id": |
|---|
| | | RuleIdColumn := field.(*entity.ColumnInt64Array).Data() |
|---|
| | | if len(RuleIdColumn) > 0 { |
|---|
| | | record.RuleId = RuleIdColumn[i] |
|---|
| | | } |
|---|
| | | case "event_level_id": |
|---|
| | | EventLevelIdColumn := field.(*entity.ColumnInt64Array).Data() |
|---|
| | | if len(EventLevelIdColumn) > 0 { |
|---|
| | | record.EventLevelId = EventLevelIdColumn[i] |
|---|
| | | } |
|---|
| | | case "video_point_id": |
|---|
| | | VideoPointIdColumn := field.(*entity.ColumnInt64).Data() |
|---|
| | | if len(VideoPointIdColumn) > 0 { |
|---|
| | | record.VideoPointId = VideoPointIdColumn[i] |
|---|
| | | } |
|---|
| | | case "detect_id": |
|---|
| | | DetectIdColumn := field.(*entity.ColumnInt64Array).Data() |
|---|
| | | if len(DetectIdColumn) > 0 { |
|---|
| | | record.DetectId = DetectIdColumn[i] |
|---|
| | | } |
|---|
| | | case "image_path": |
|---|
| | | ImagePathColumn := field.(*entity.ColumnVarChar).Data() |
|---|
| | | if len(ImagePathColumn) > 0 { |
|---|
| | | record.ImagePath = ImagePathColumn[i] |
|---|
| | | } |
|---|
| | | case "video_path": |
|---|
| | | VideoPathColumn := field.(*entity.ColumnVarChar).Data() |
|---|
| | | if len(VideoPathColumn) > 0 { |
|---|
| | | record.VideoPath = VideoPathColumn[i] |
|---|
| | | } |
|---|
| | | case "zh_desc_class": |
|---|
| | | zhColumn := field.(*entity.ColumnVarChar).Data() |
|---|
| | | if len(zhColumn) > 0 { |
|---|
| | | record.ZhDescClass = zhColumn[i] |
|---|
| | | } |
|---|
| | | case "task_name": |
|---|
| | | TNColumn := field.(*entity.ColumnVarChar).Data() |
|---|
| | | if len(TNColumn) > 0 { |
|---|
| | | record.TaskName = TNColumn[i] |
|---|
| | | } |
|---|
| | | case "event_level_name": |
|---|
| | | EVColumn := field.(*entity.ColumnVarChar).Data() |
|---|
| | | if len(EVColumn) > 0 { |
|---|
| | | record.EventLevelName = EVColumn[i] |
|---|
| | | } |
|---|
| | | case "is_desc": |
|---|
| | | descColumn := field.(*entity.ColumnInt64).Data() |
|---|
| | | if len(descColumn) > 0 { |
|---|
| | | record.IsDesc = descColumn[i] |
|---|
| | | } |
|---|
| | | case "detect_num": |
|---|
| | | DEColumn := field.(*entity.ColumnInt64).Data() |
|---|
| | | if len(DEColumn) > 0 { |
|---|
| | | record.DetectNum = DEColumn[i] |
|---|
| | | } |
|---|
| | | case "is_waning": |
|---|
| | | warnColumn := field.(*entity.ColumnInt64).Data() |
|---|
| | | if len(warnColumn) > 0 { |
|---|
| | | record.IsWarning = warnColumn[i] |
|---|
| | | } |
|---|
| | | case "detect_time": |
|---|
| | | DetectTimeColumn := field.(*entity.ColumnVarChar).Data() |
|---|
| | | if len(DetectTimeColumn) > 0 { |
|---|
| | | record.DetectTime = DetectTimeColumn[i] |
|---|
| | | } |
|---|
| | | case "knowledge_id": |
|---|
| | | KnowledgeIdColumn := field.(*entity.ColumnInt64Array).Data() |
|---|
| | | if len(KnowledgeIdColumn) > 0 { |
|---|
| | | record.KnowledgeId = KnowledgeIdColumn[i] |
|---|
| | | } |
|---|
| | | case "suggestion": |
|---|
| | | SuggestionColumn := field.(*entity.ColumnVarChar).Data() |
|---|
| | | if len(SuggestionColumn) > 0 { |
|---|
| | | record.Suggestion = SuggestionColumn[i] |
|---|
| | | } |
|---|
| | | case "risk_description": |
|---|
| | | RiskDescriptionColumn := field.(*entity.ColumnVarChar).Data() |
|---|
| | | if len(RiskDescriptionColumn) > 0 { |
|---|
| | | record.RiskDescription = RiskDescriptionColumn[i] |
|---|
| | | } |
|---|
| | | } |
|---|
| | | } |
|---|
| | | records = append(records, record) |
|---|
| | | } |
|---|
| | | return records |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 分页查询 |
|---|
| | | func (do *DataOperator) queryWithPagination(collectionName string, filterExpr string, pageNum int64, pageSize int64) (*PaginatedResult, error) { |
|---|
| | | ctx := context.Background() |
|---|
| | | |
|---|
| | | // 计算偏移量 |
|---|
| | | //offset := (pageNum - 1) * pageSize |
|---|
| | | |
|---|
| | | // 构建查询表达式（Milvus 2.1+ 支持 limit/offset） |
|---|
| | | //queryExpr := fmt.Sprintf("%s limit %d offset %d", filterExpr, pageSize, offset) |
|---|
| | | |
|---|
| | | total, err := do.getTotalCount(collectionName, filterExpr) |
|---|
| | | if err != nil { |
|---|
| | | return nil, err |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 计算总页数 |
|---|
| | | totalPages := int(math.Ceil(float64(total) / float64(pageSize))) |
|---|
| | | |
|---|
| | | // 执行查询 |
|---|
| | | result, err := do.client.client.Query( |
|---|
| | | ctx, |
|---|
| | | collectionName, |
|---|
| | | []string{}, // 所有分区 |
|---|
| | | filterExpr, |
|---|
| | | []string{"rule_id", "task_id", "is_waning", "zh_desc_class", "task_name", "event_level_name", "detect_num", |
|---|
| | | "event_level_id", "video_point_id", "detect_id", "image_path", |
|---|
| | | "video_path", "detect_time", "knowledge_id", "risk_description", "suggestion", "id", "is_desc"}, // 返回所有字段 |
|---|
| | | // client.WithLimit(pageSize), |
|---|
| | | // client.WithOffset(offset), |
|---|
| | | ) |
|---|
| | | if err != nil { |
|---|
| | | return nil, fmt.Errorf("分页查询失败: %v", err) |
|---|
| | | } |
|---|
| | | lists := convertResultToMap(result) |
|---|
| | | layout := "2006-01-02 15:04:05.999999" |
|---|
| | | |
|---|
| | | 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) |
|---|
| | | }) |
|---|
| | | |
|---|
| | | items := Paginate(lists, int(pageNum), int(pageSize)) |
|---|
| | | return &PaginatedResult{ |
|---|
| | | Items: items, |
|---|
| | | Pagination: Pagination{ |
|---|
| | | Page: int(pageNum), |
|---|
| | | PageSize: int(pageSize), |
|---|
| | | Total: int(total), |
|---|
| | | TotalPage: totalPages, |
|---|
| | | }, |
|---|
| | | }, nil |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 分页数据 |
|---|
| | | func Paginate(data []StaticRecord, page, pageSize int) []StaticRecord { |
|---|
| | | start := (page - 1) * pageSize |
|---|
| | | if start >= len(data) { |
|---|
| | | return nil |
|---|
| | | } |
|---|
| | | end := start + pageSize |
|---|
| | | if end > len(data) { |
|---|
| | | end = len(data) |
|---|
| | | } |
|---|
| | | return data[start:end] |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 统计数量 |
|---|
| | | func (do *DataOperator) getTotalCount(collectionName string, filterExpr string) (int64, error) { |
|---|
| | | result, err := do.client.client.Query( |
|---|
| | | context.Background(), |
|---|
| | | collectionName, |
|---|
| | | []string{}, |
|---|
| | | filterExpr, |
|---|
| | | []string{"count(*)"}, |
|---|
| | | ) |
|---|
| | | if err != nil { |
|---|
| | | return 0, err |
|---|
| | | } |
|---|
| | | return result[0].(*entity.ColumnInt64).Data()[0], nil |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 根据id获取任务 |
|---|
| | | func GetTaskByIds(ids []int64) (items []TaskOption, err error) { |
|---|
| | | placeholders := make([]string, len(ids)) |
|---|
| | | args := make([]interface{}, len(ids)) |
|---|
| | | for i, id := range ids { |
|---|
| | | placeholders[i] = "?" |
|---|
| | | args[i] = id |
|---|
| | | } |
|---|
| | | sqlStr := `select task_id,task_name from mal_smart_task where task_id in (` + strings.Join(placeholders, ",") + `)` |
|---|
| | | if err := db.Raw(sqlStr).Scan(&items).Error; err != nil { |
|---|
| | | return nil, err |
|---|
| | | } |
|---|
| | | return |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 根据id获取视频 |
|---|
| | | func GetVideoById(vid int64) (video Camera, err error) { |
|---|
| | | // 如果返回的是指针,需要初始化 |
|---|
| | | sqlStr := "select id,name,alias,type,addr,rtsp,is_running,run_type,run_enable FROM cameras where video_id=?" |
|---|
| | | if err := db.Raw(sqlStr, vid).Scan(&video).Error; err != nil { |
|---|
| | | return Camera{}, err |
|---|
| | | } |
|---|
| | | return |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 根据id获取检测内容 |
|---|
| | | func GetCheckByIds(ids []int64) (items []CheckOption, err error) { |
|---|
| | | placeholders := make([]string, len(ids)) |
|---|
| | | args := make([]interface{}, len(ids)) |
|---|
| | | for i, id := range ids { |
|---|
| | | placeholders[i] = "?" |
|---|
| | | args[i] = id |
|---|
| | | } |
|---|
| | | sqlStr := `select check_id,file_name from mal_check_content where check_id in (` + strings.Join(placeholders, ",") + `)` |
|---|
| | | if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil { |
|---|
| | | return nil, err |
|---|
| | | } |
|---|
| | | return |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 根据id获取规则 |
|---|
| | | func GetRuleByIds(ids []int64) (items []RuleOption, err error) { |
|---|
| | | placeholders := make([]string, len(ids)) |
|---|
| | | args := make([]interface{}, len(ids)) |
|---|
| | | for i, id := range ids { |
|---|
| | | placeholders[i] = "?" |
|---|
| | | args[i] = id |
|---|
| | | } |
|---|
| | | sqlStr := `select rule_id,file_name from mal_warning_rule where rule_id in (` + strings.Join(placeholders, ",") + `)` |
|---|
| | | if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil { |
|---|
| | | return nil, err |
|---|
| | | } |
|---|
| | | return |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 获取事件 |
|---|
| | | func GetEventByIds(ids []int64) (items []DictOption, err error) { |
|---|
| | | placeholders := make([]string, len(ids)) |
|---|
| | | args := make([]interface{}, len(ids)) |
|---|
| | | for i, id := range ids { |
|---|
| | | placeholders[i] = "?" |
|---|
| | | args[i] = 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 { |
|---|
| | | return nil, err |
|---|
| | | } |
|---|
| | | |
|---|
| | | return |
|---|
| | | |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 根据知识库id查数据 |
|---|
| | | func GetKnowledgeDocumentByIds(ids []int64) (items []KnowledgeDocumentOption, err error) { |
|---|
| | | placeholders := make([]string, len(ids)) |
|---|
| | | args := make([]interface{}, len(ids)) |
|---|
| | | for i, id := range ids { |
|---|
| | | placeholders[i] = "?" |
|---|
| | | args[i] = id |
|---|
| | | } |
|---|
| | | sqlStr := `SELECT id, know_id, file_name as title FROM mal_knowledge_document where id in (` + strings.Join(placeholders, ",") + `)` |
|---|
| | | if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil { |
|---|
| | | return nil, err |
|---|
| | | } |
|---|
| | | return |
|---|
| | | } |
|---|
| | | |
|---|
| | | // 根据知识库id查数据 |
|---|
| | | func GetCameraIds(ids []string) (items []CameraDto, err error) { |
|---|
| | | placeholders := make([]string, len(ids)) |
|---|
| | | args := make([]interface{}, len(ids)) |
|---|
| | | for i, id := range ids { |
|---|
| | | placeholders[i] = "?" |
|---|
| | | args[i] = id |
|---|
| | | } |
|---|
| | | sqlStr := `select id,name,alias,type,addr,rtsp,is_running,run_type,run_enable,video_id FROM cameras where id in (` + strings.Join(placeholders, ",") + `)` |
|---|
| | | if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil { |
|---|
| | | return nil, err |
|---|
| | | } |
|---|
| | | return |
|---|
| | | } |
|---|
| | | package models
|
|---|
| | |
|
|---|
| | | import (
|
|---|
| | | "context"
|
|---|
| | | "fmt"
|
|---|
| | | "math"
|
|---|
| | | "sort"
|
|---|
| | | "strconv"
|
|---|
| | | "strings"
|
|---|
| | | "time"
|
|---|
| | |
|
|---|
| | | "github.com/milvus-io/milvus-sdk-go/v2/entity"
|
|---|
| | | )
|
|---|
| | |
|
|---|
| | | // DataOperator 数据操作相关方法
|
|---|
| | | type DataOperator struct {
|
|---|
| | | client *MilvusClient
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // NewDataOperator 创建数据操作器
|
|---|
| | | func NewDataOperator(client *MilvusClient) *DataOperator {
|
|---|
| | | return &DataOperator{client: client}
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type PicWidHei struct {
|
|---|
| | | PicW int `json:"picW"`
|
|---|
| | | PicH int `json:"picH"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 统计查询
|
|---|
| | | type StaticRecord struct {
|
|---|
| | | Id string `json:"id"`
|
|---|
| | | TaskId []int64 `json:"task_id"`
|
|---|
| | | RuleId []int64 `json:"rule_id"`
|
|---|
| | | EventLevelId []int64 `json:"event_level_id"`
|
|---|
| | | VideoPointId int64 `json:"video_point_id"`
|
|---|
| | | DetectId []int64 `json:"detect_id"`
|
|---|
| | | VideoName string `json:"video_name"`
|
|---|
| | | TaskNames []TaskOption `json:"task_names"`
|
|---|
| | | CheckNames []CheckOption `json:"check_names"`
|
|---|
| | | RuleNames []RuleOption `json:"rule_names"`
|
|---|
| | | EventLevels []DictOption `json:"event_levels"`
|
|---|
| | | KnowledgeDocuments []KnowledgeDocumentOption `json:"knowledge_documents"`
|
|---|
| | | ImagePath string `json:"image_path"`
|
|---|
| | | VideoPath string `json:"video_path"`
|
|---|
| | | DetectTime string `json:"detect_time"`
|
|---|
| | | IsWarning int64 `json:"is_warning"`
|
|---|
| | | ZhDescClass string `json:"zh_desc_class"`
|
|---|
| | | TaskName string `json:"task_name"`
|
|---|
| | | EventLevelName string `json:"event_level_name"`
|
|---|
| | | DetectNum int64 `json:"detect_num"`
|
|---|
| | | Suggestion string `json:"suggestion"`
|
|---|
| | | RiskDescription string `json:"risk_description"`
|
|---|
| | | KnowledgeId []int64 `json:"knowledge_id"`
|
|---|
| | | IsDesc int64 `json:"is_desc"`
|
|---|
| | | CameraId string `json:"cameraId"`
|
|---|
| | | CameraName string `json:"cameraName"`
|
|---|
| | | CameraAddr string `json:"cameraAddr"`
|
|---|
| | | PicDate string `json:"picDate"`
|
|---|
| | | PicId int64 `json:"picId"`
|
|---|
| | | PicMaxUrl []string `json:"picMaxUrl"`
|
|---|
| | | PicSrcUrl []string `json:"picSrcUrl"`
|
|---|
| | | PicWH PicWidHei `json:"picWH"`
|
|---|
| | | SdkName string `json:"sdkName"`
|
|---|
| | | Content string `json:"content"`
|
|---|
| | | AlarmRules []AlarmRule `json:"alarmRules"`
|
|---|
| | | LikeDate string `json:"likeDate"`
|
|---|
| | | ShowLabels string `json:"showLabels"`
|
|---|
| | | OtherLabels string `json:"otherLabels"`
|
|---|
| | | VideoUrl string `json:"videoUrl"`
|
|---|
| | | AnalyServerId string `json:"analyServerId"`
|
|---|
| | | AnalyServerName string `json:"analyServerName"`
|
|---|
| | | AnalyServerIp string `json:"analyServerIp"`
|
|---|
| | | ClusterId string `json:"clusterId"`
|
|---|
| | | IsAlarm bool `json:"isAlarm"`
|
|---|
| | | IsAckAlarm bool `json:"isAckAlarm"`
|
|---|
| | | IsCollect bool `json:"isCollect"`
|
|---|
| | | IsDelete bool `json:"isDelete"`
|
|---|
| | | BaseInfo []*BaseCompareInfo `json:"baseInfo"`
|
|---|
| | | TargetInfo []TargetInfo `json:"targetInfo"`
|
|---|
| | | FileId string `json:"fileId"` // 数据栈文件id
|
|---|
| | | DataSource string `json:"dataSource"` // 数据来源:摄像机, 数据栈
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type Point struct {
|
|---|
| | | X float64 `json:"x"`
|
|---|
| | | Y float64 `json:"y"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type Points struct {
|
|---|
| | | TopLeft Point `json:"topLeft"`
|
|---|
| | | BottomRight Point `json:"bottomRight"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type BaseCompareInfo struct {
|
|---|
| | | TableId string `json:"tableId"`
|
|---|
| | | TableName string `json:"tableName"`
|
|---|
| | | BwType string `json:"bwType"`
|
|---|
| | | CompareScore float64 `json:"compareScore"`
|
|---|
| | | TargetId string `json:"targetId"`
|
|---|
| | | TargetName string `json:"targetName"`
|
|---|
| | | TargetPicUrl string `json:"targetPicUrl"`
|
|---|
| | | MonitorLevel string `json:"monitorLevel"`
|
|---|
| | | Content string `json:"content"`
|
|---|
| | | DbLabel string `json:"labels"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type TargetInfo struct {
|
|---|
| | | TargetId string `json:"targetId"`
|
|---|
| | | AreaId string `json:"areaId"`
|
|---|
| | | AreaName string `json:"areaName"`
|
|---|
| | | TargetScore float64 `json:"targetScore"`
|
|---|
| | | TargetType string `json:"targetType"`
|
|---|
| | | Feature string `json:"feature"`
|
|---|
| | | PicSmUrl string `json:"picSmUrl"`
|
|---|
| | | TargetLocation Points `json:"targetLocation"`
|
|---|
| | | BelongsTargetID string `json:"belongsTargetId"`
|
|---|
| | | Attribute string `json:"attribute"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type AlarmRule struct {
|
|---|
| | | AlarmLevel string `db:"alarmLevel" json:"alarmLevel"`
|
|---|
| | | DefenceState bool `db:"defenceState" json:"defenceState"`
|
|---|
| | | GroupId string `db:"groupId" json:"groupId"`
|
|---|
| | | LinkInfo string `db:"linkInfo" json:"linkInfo"`
|
|---|
| | | RuleText string `db:"ruleText" json:"ruleText"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type TaskOption struct {
|
|---|
| | | TaskId int64 `db:"task_id" json:"taskId"`
|
|---|
| | | TaskName string `db:"task_name" json:"taskName"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type CheckOption struct {
|
|---|
| | | TaskId int64 `db:"task_id" json:"taskId,omitempty"`
|
|---|
| | | CheckId int64 `db:"check_id" json:"checkId"`
|
|---|
| | | FileName string `db:"file_name" json:"fileName"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type RuleOption struct {
|
|---|
| | | TaskId int64 `db:"task_id" json:"taskId,omitempty"`
|
|---|
| | | RuleId int64 `db:"rule_id" json:"ruleId"`
|
|---|
| | | FileName string `db:"file_name" json:"fileName"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 字典
|
|---|
| | | type DictOption struct {
|
|---|
| | | DictId int64 `db:"dict_id" json:"dictId"`
|
|---|
| | | DictName string `db:"dict_name" json:"dictName"`
|
|---|
| | | DictValue string `db:"dict_value" json:"dictValue"`
|
|---|
| | | DictType string `db:"dict_type" json:"dictType"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 知识库
|
|---|
| | | type KnowledgeDocumentOption struct {
|
|---|
| | | Id string `db:"id" json:"id"`
|
|---|
| | | Title string `db:"title" json:"title"`
|
|---|
| | | KnowId int64 `db:"know_id" json:"know_id"`
|
|---|
| | | FileUrl string `db:"title" json:"file_url"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // Pagination 分页信息
|
|---|
| | | type Pagination struct {
|
|---|
| | | Page int `json:"page"` // 当前页码
|
|---|
| | | PageSize int `json:"pageSize"` // 每页大小
|
|---|
| | | Total int `json:"total"` // 总记录数
|
|---|
| | | TotalPage int `json:"totalPage"` // 总页数
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // PaginatedResult 分页结果
|
|---|
| | | type PaginatedResult struct {
|
|---|
| | | Items []StaticRecord `json:"list"`
|
|---|
| | | Pagination Pagination `json:"pagination"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | type PaginatedResult2 struct {
|
|---|
| | | Items map[string]interface{} `json:"list"`
|
|---|
| | | Pagination Pagination `json:"pagination"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 记录
|
|---|
| | | type RecordReq struct {
|
|---|
| | | Ids []string `json:"ids"`
|
|---|
| | | VideoIds []string `json:"treeNodes"`
|
|---|
| | | TaskIds []int64 `json:"taskIds`
|
|---|
| | | TaskName []string `json:"taskName"`
|
|---|
| | | Warning int64 `json:"warning`
|
|---|
| | | Page int64 `json:"page"`
|
|---|
| | | PageSize int64 `json:"pageSize"`
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 根据分页取数据
|
|---|
| | | func GetWithPage(collectionName string, pageNum int64, pageSize int64, filter string) (*PaginatedResult, error) {
|
|---|
| | |
|
|---|
| | | if pageNum < 1 {
|
|---|
| | | pageNum = 1
|
|---|
| | | }
|
|---|
| | | if pageSize < 1 {
|
|---|
| | | pageSize = 10
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | if dbClient == nil {
|
|---|
| | | return nil, nil
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | dataOp := &DataOperator{client: dbClient}
|
|---|
| | |
|
|---|
| | | fmt.Println(filter)
|
|---|
| | | records, err := dataOp.queryWithPagination(collectionName, filter, pageNum, pageSize)
|
|---|
| | | if err != nil {
|
|---|
| | | return nil, err
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | return records, nil
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 转换数据
|
|---|
| | | func convertResultToMap(result []entity.Column) []StaticRecord {
|
|---|
| | | if len(result) == 0 {
|
|---|
| | | return nil
|
|---|
| | | }
|
|---|
| | | count := result[0].Len()
|
|---|
| | | var records []StaticRecord
|
|---|
| | | for i := 0; i < count; i++ {
|
|---|
| | | record := StaticRecord{}
|
|---|
| | | for _, field := range result {
|
|---|
| | | switch field.Name() {
|
|---|
| | | case "id":
|
|---|
| | | IdColumn := field.(*entity.ColumnInt64).Data()
|
|---|
| | | if len(IdColumn) > 0 {
|
|---|
| | | record.Id = strconv.FormatInt(IdColumn[i], 10)
|
|---|
| | | }
|
|---|
| | | case "task_id":
|
|---|
| | | TaskIdColumn := field.(*entity.ColumnInt64Array).Data()
|
|---|
| | | if len(TaskIdColumn) > 0 {
|
|---|
| | | record.TaskId = TaskIdColumn[i]
|
|---|
| | | }
|
|---|
| | | case "rule_id":
|
|---|
| | | RuleIdColumn := field.(*entity.ColumnInt64Array).Data()
|
|---|
| | | if len(RuleIdColumn) > 0 {
|
|---|
| | | record.RuleId = RuleIdColumn[i]
|
|---|
| | | }
|
|---|
| | | case "event_level_id":
|
|---|
| | | EventLevelIdColumn := field.(*entity.ColumnInt64Array).Data()
|
|---|
| | | if len(EventLevelIdColumn) > 0 {
|
|---|
| | | record.EventLevelId = EventLevelIdColumn[i]
|
|---|
| | | }
|
|---|
| | | case "video_point_id":
|
|---|
| | | VideoPointIdColumn := field.(*entity.ColumnInt64).Data()
|
|---|
| | | if len(VideoPointIdColumn) > 0 {
|
|---|
| | | record.VideoPointId = VideoPointIdColumn[i]
|
|---|
| | | }
|
|---|
| | | case "detect_id":
|
|---|
| | | DetectIdColumn := field.(*entity.ColumnInt64Array).Data()
|
|---|
| | | if len(DetectIdColumn) > 0 {
|
|---|
| | | record.DetectId = DetectIdColumn[i]
|
|---|
| | | }
|
|---|
| | | case "image_path":
|
|---|
| | | ImagePathColumn := field.(*entity.ColumnVarChar).Data()
|
|---|
| | | if len(ImagePathColumn) > 0 {
|
|---|
| | | record.ImagePath = ImagePathColumn[i]
|
|---|
| | | }
|
|---|
| | | case "video_path":
|
|---|
| | | VideoPathColumn := field.(*entity.ColumnVarChar).Data()
|
|---|
| | | if len(VideoPathColumn) > 0 {
|
|---|
| | | record.VideoPath = VideoPathColumn[i]
|
|---|
| | | }
|
|---|
| | | case "zh_desc_class":
|
|---|
| | | zhColumn := field.(*entity.ColumnVarChar).Data()
|
|---|
| | | if len(zhColumn) > 0 {
|
|---|
| | | record.ZhDescClass = zhColumn[i]
|
|---|
| | | }
|
|---|
| | | case "task_name":
|
|---|
| | | TNColumn := field.(*entity.ColumnVarChar).Data()
|
|---|
| | | if len(TNColumn) > 0 {
|
|---|
| | | record.TaskName = TNColumn[i]
|
|---|
| | | }
|
|---|
| | | case "event_level_name":
|
|---|
| | | EVColumn := field.(*entity.ColumnVarChar).Data()
|
|---|
| | | if len(EVColumn) > 0 {
|
|---|
| | | record.EventLevelName = EVColumn[i]
|
|---|
| | | }
|
|---|
| | | case "is_desc":
|
|---|
| | | descColumn := field.(*entity.ColumnInt64).Data()
|
|---|
| | | if len(descColumn) > 0 {
|
|---|
| | | record.IsDesc = descColumn[i]
|
|---|
| | | }
|
|---|
| | | case "detect_num":
|
|---|
| | | DEColumn := field.(*entity.ColumnInt64).Data()
|
|---|
| | | if len(DEColumn) > 0 {
|
|---|
| | | record.DetectNum = DEColumn[i]
|
|---|
| | | }
|
|---|
| | | case "is_waning":
|
|---|
| | | warnColumn := field.(*entity.ColumnInt64).Data()
|
|---|
| | | if len(warnColumn) > 0 {
|
|---|
| | | record.IsWarning = warnColumn[i]
|
|---|
| | | }
|
|---|
| | | case "detect_time":
|
|---|
| | | DetectTimeColumn := field.(*entity.ColumnVarChar).Data()
|
|---|
| | | if len(DetectTimeColumn) > 0 {
|
|---|
| | | record.DetectTime = DetectTimeColumn[i]
|
|---|
| | | }
|
|---|
| | | case "knowledge_id":
|
|---|
| | | KnowledgeIdColumn := field.(*entity.ColumnInt64Array).Data()
|
|---|
| | | if len(KnowledgeIdColumn) > 0 {
|
|---|
| | | record.KnowledgeId = KnowledgeIdColumn[i]
|
|---|
| | | }
|
|---|
| | | case "suggestion":
|
|---|
| | | SuggestionColumn := field.(*entity.ColumnVarChar).Data()
|
|---|
| | | if len(SuggestionColumn) > 0 {
|
|---|
| | | record.Suggestion = SuggestionColumn[i]
|
|---|
| | | }
|
|---|
| | | case "risk_description":
|
|---|
| | | RiskDescriptionColumn := field.(*entity.ColumnVarChar).Data()
|
|---|
| | | if len(RiskDescriptionColumn) > 0 {
|
|---|
| | | record.RiskDescription = RiskDescriptionColumn[i]
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | }
|
|---|
| | | records = append(records, record)
|
|---|
| | | }
|
|---|
| | | return records
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 分页查询
|
|---|
| | | func (do *DataOperator) queryWithPagination(collectionName string, filterExpr string, pageNum int64, pageSize int64) (*PaginatedResult, error) {
|
|---|
| | | ctx := context.Background()
|
|---|
| | |
|
|---|
| | | // 计算偏移量
|
|---|
| | | //offset := (pageNum - 1) * pageSize
|
|---|
| | |
|
|---|
| | | // 构建查询表达式（Milvus 2.1+ 支持 limit/offset）
|
|---|
| | | //queryExpr := fmt.Sprintf("%s limit %d offset %d", filterExpr, pageSize, offset)
|
|---|
| | |
|
|---|
| | | total, err := do.getTotalCount(collectionName, filterExpr)
|
|---|
| | | if err != nil {
|
|---|
| | | return nil, err
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 计算总页数
|
|---|
| | | totalPages := int(math.Ceil(float64(total) / float64(pageSize)))
|
|---|
| | |
|
|---|
| | | // 执行查询
|
|---|
| | | result, err := do.client.client.Query(
|
|---|
| | | ctx,
|
|---|
| | | collectionName,
|
|---|
| | | []string{}, // 所有分区
|
|---|
| | | filterExpr,
|
|---|
| | | []string{"rule_id", "task_id", "is_waning", "zh_desc_class", "task_name", "event_level_name", "detect_num",
|
|---|
| | | "event_level_id", "video_point_id", "detect_id", "image_path",
|
|---|
| | | "video_path", "detect_time", "knowledge_id", "risk_description", "suggestion", "id", "is_desc"}, // 返回所有字段
|
|---|
| | | // client.WithLimit(pageSize),
|
|---|
| | | // client.WithOffset(offset),
|
|---|
| | | )
|
|---|
| | | if err != nil {
|
|---|
| | | return nil, fmt.Errorf("分页查询失败: %v", err)
|
|---|
| | | }
|
|---|
| | | lists := convertResultToMap(result)
|
|---|
| | | layout := "2006-01-02 15:04:05.999999"
|
|---|
| | |
|
|---|
| | | 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)
|
|---|
| | | })
|
|---|
| | |
|
|---|
| | | items := Paginate(lists, int(pageNum), int(pageSize))
|
|---|
| | | return &PaginatedResult{
|
|---|
| | | Items: items,
|
|---|
| | | Pagination: Pagination{
|
|---|
| | | Page: int(pageNum),
|
|---|
| | | PageSize: int(pageSize),
|
|---|
| | | Total: int(total),
|
|---|
| | | TotalPage: totalPages,
|
|---|
| | | },
|
|---|
| | | }, nil
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 分页数据
|
|---|
| | | func Paginate(data []StaticRecord, page, pageSize int) []StaticRecord {
|
|---|
| | | start := (page - 1) * pageSize
|
|---|
| | | if start >= len(data) {
|
|---|
| | | return nil
|
|---|
| | | }
|
|---|
| | | end := start + pageSize
|
|---|
| | | if end > len(data) {
|
|---|
| | | end = len(data)
|
|---|
| | | }
|
|---|
| | | return data[start:end]
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 统计数量
|
|---|
| | | func (do *DataOperator) getTotalCount(collectionName string, filterExpr string) (int64, error) {
|
|---|
| | | result, err := do.client.client.Query(
|
|---|
| | | context.Background(),
|
|---|
| | | collectionName,
|
|---|
| | | []string{},
|
|---|
| | | filterExpr,
|
|---|
| | | []string{"count(*)"},
|
|---|
| | | )
|
|---|
| | | if err != nil {
|
|---|
| | | return 0, err
|
|---|
| | | }
|
|---|
| | | return result[0].(*entity.ColumnInt64).Data()[0], nil
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 根据id获取任务
|
|---|
| | | func GetTaskByIds(ids []int64) (items []TaskOption, err error) {
|
|---|
| | | placeholders := make([]string, len(ids))
|
|---|
| | | args := make([]interface{}, len(ids))
|
|---|
| | | for i, id := range ids {
|
|---|
| | | placeholders[i] = "?"
|
|---|
| | | args[i] = id
|
|---|
| | | }
|
|---|
| | | sqlStr := `select task_id,task_name from mal_smart_task where task_id in (` + strings.Join(placeholders, ",") + `)`
|
|---|
| | | if err := db.Raw(sqlStr).Scan(&items).Error; err != nil {
|
|---|
| | | return nil, err
|
|---|
| | | }
|
|---|
| | | return
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 根据id获取视频
|
|---|
| | | func GetVideoById(vid int64) (video Camera, err error) {
|
|---|
| | | // 如果返回的是指针,需要初始化
|
|---|
| | | sqlStr := "select id,name,alias,type,addr,rtsp,is_running,run_type,run_enable FROM cameras where video_id=?"
|
|---|
| | | if err := db.Raw(sqlStr, vid).Scan(&video).Error; err != nil {
|
|---|
| | | return Camera{}, err
|
|---|
| | | }
|
|---|
| | | return
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 根据id获取检测内容
|
|---|
| | | func GetCheckByIds(ids []int64) (items []CheckOption, err error) {
|
|---|
| | | placeholders := make([]string, len(ids))
|
|---|
| | | args := make([]interface{}, len(ids))
|
|---|
| | | for i, id := range ids {
|
|---|
| | | placeholders[i] = "?"
|
|---|
| | | args[i] = id
|
|---|
| | | }
|
|---|
| | | sqlStr := `select check_id,file_name from mal_check_content where check_id in (` + strings.Join(placeholders, ",") + `)`
|
|---|
| | | if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil {
|
|---|
| | | return nil, err
|
|---|
| | | }
|
|---|
| | | return
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 根据id获取规则
|
|---|
| | | func GetRuleByIds(ids []int64) (items []RuleOption, err error) {
|
|---|
| | | placeholders := make([]string, len(ids))
|
|---|
| | | args := make([]interface{}, len(ids))
|
|---|
| | | for i, id := range ids {
|
|---|
| | | placeholders[i] = "?"
|
|---|
| | | args[i] = id
|
|---|
| | | }
|
|---|
| | | sqlStr := `select rule_id,file_name from mal_warning_rule where rule_id in (` + strings.Join(placeholders, ",") + `)`
|
|---|
| | | if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil {
|
|---|
| | | return nil, err
|
|---|
| | | }
|
|---|
| | | return
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 获取事件
|
|---|
| | | func GetEventByIds(ids []int64) (items []DictOption, err error) {
|
|---|
| | | placeholders := make([]string, len(ids))
|
|---|
| | | args := make([]interface{}, len(ids))
|
|---|
| | | for i, id := range ids {
|
|---|
| | | placeholders[i] = "?"
|
|---|
| | | args[i] = 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 {
|
|---|
| | | return nil, err
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | return
|
|---|
| | |
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 根据知识库id查数据
|
|---|
| | | func GetKnowledgeDocumentByIds(ids []int64) (items []KnowledgeDocumentOption, err error) {
|
|---|
| | | placeholders := make([]string, len(ids))
|
|---|
| | | args := make([]interface{}, len(ids))
|
|---|
| | | for i, id := range ids {
|
|---|
| | | placeholders[i] = "?"
|
|---|
| | | args[i] = id
|
|---|
| | | }
|
|---|
| | | sqlStr := `SELECT id, know_id, file_name as title FROM mal_knowledge_document where id in (` + strings.Join(placeholders, ",") + `)`
|
|---|
| | | if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil {
|
|---|
| | | return nil, err
|
|---|
| | | }
|
|---|
| | | return
|
|---|
| | | }
|
|---|
| | |
|
|---|
| | | // 根据知识库id查数据
|
|---|
| | | func GetCameraIds(ids []string) (items []CameraDto, err error) {
|
|---|
| | | placeholders := make([]string, len(ids))
|
|---|
| | | args := make([]interface{}, len(ids))
|
|---|
| | | for i, id := range ids {
|
|---|
| | | placeholders[i] = "?"
|
|---|
| | | args[i] = id
|
|---|
| | | }
|
|---|
| | | sqlStr := `select id,name,alias,type,addr,rtsp,is_running,run_type,run_enable,video_id FROM cameras where id in (` + strings.Join(placeholders, ",") + `)`
|
|---|
| | | if err := db.Raw(sqlStr, args...).Scan(&items).Error; err != nil {
|
|---|
| | | return nil, err
|
|---|
| | | }
|
|---|
| | | return
|
|---|
| | | }
|
|---|
