qixiaoning
1 天以前 c3fd4effdef0cc5955f89b5df279b14841e857f5
camera-common/models/task.go
@@ -6,9 +6,37 @@
)
// 智查任务
type Task struct {
   TaskId   int64  `db:"task_id" json:"taskId"`
   TaskName string `db:"task_name" json:"taskName"`
}
type TaskAggregate struct {
   ID       int64  `gorm:"column:task_id" json:"id"`
   Name     string `gorm:"column:task_name"  json:"name"`
   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"
}
// 视频
@@ -36,3 +64,28 @@
   }
   return
}
// 查询列表数据
func GetAllTask() (items []TaskAggregate, err error) {
   // 主查询SQL
   sqlStr := fmt.Sprintf(`
       SELECT task_id, task_name
       FROM  mal_smart_task
       ORDER BY task_id DESC
   `)
   if err := db.Raw(sqlStr).Scan(&items).Error; err != nil {
      fmt.Println(err.Error())
      return nil, err
   }
   return
}
// 根据cid查询视频
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
}