qixiaoning
2025-10-29 9b17a8dcbc1f94eb117a37b3b24ca9dae0d2d588
camera-common/models/task.go
@@ -18,11 +18,25 @@
   IsDelete int8   `json:"isDelete"`
}
type TaskIdM struct {
   ID int64 `gorm:"column:task_id" json:"id"`
}
type TaskM struct {
   TaskId          int64  `db:"task_id" json:"taskId"`
   TaskName        string `db:"task_name" json:"taskName"`
   TaskDescription string `db:"task_description" json:"taskDescription"`
   EventName       string `db:"event_name" json:"eventName"`
}
// 训练模型标签
type TrainTags struct {
   Id         int64  `gorm:"primaryKey;autoIncrement;column:id" json:"id"`
   Createtime string `gorm:"type:varchar(50);column:create_time;default:''" json:"create_time"`
}
func (TrainTags) TableName() string {
   return "mal_train_tags"
}
// 视频
@@ -67,11 +81,11 @@
}
// 根据cid查询视频
func GetTaskTotal(cid string) (total int, err error) {
   sql := "select count(1) as total from mal_task_video_link where video_id='" + cid + "'"
   err = db.Raw(sql).Count(&total).Error
   if err != nil {
      return 0, err
func GetTaskTotal(cid string) (items []TaskIdM, err error) {
   sqlStr := "select task_id from mal_task_video_link where video_id='" + cid + "'"
   if err := db.Raw(sqlStr).Scan(&items).Error; err != nil {
      fmt.Println(err.Error())
      return nil, err
   }
   return total, nil
   return
}