package models
|
|
import (
|
"fmt"
|
"strings"
|
"vamicro/camera-common/models"
|
)
|
|
// 查询列表数据
|
func GetListTask(ids []string) (items []models.TaskM, err error) {
|
|
//idList := strings.Join(ids, ",")
|
idList := "'" + strings.Join(ids, "','") + "'"
|
// 主查询SQL
|
//sqlStr := `select c.task_id, c.task_name from mal_task_video_link a left join mal_smart_task c on a.task_id = c.task_id where a.video_id in(?) ORDER BY a.task_id desc`
|
sqlStr := fmt.Sprintf(`
|
SELECT c.task_id, c.task_name,c.task_description,b.dict_value as event_name
|
FROM mal_task_video_link a
|
LEFT JOIN mal_smart_task c ON a.task_id = c.task_id
|
LEFT JOIN mal_dict_type b ON b.dict_id = c.event_level
|
WHERE a.video_id IN (%s)
|
ORDER BY a.task_id DESC
|
`, idList)
|
if err := db2.Raw(sqlStr).Scan(&items).Error; err != nil {
|
fmt.Println(err.Error())
|
return nil, err
|
}
|
return
|
}
|
|
func GetAllRule() map[int64]string {
|
var items []models.RuleOption
|
sqlStr := `
|
SELECT a.file_name,a.rule_id,b.task_id
|
FROM mal_warning_rule a
|
LEFT JOIN mal_task_rule_link b ON a.rule_id = b.rule_id
|
WHERE task_id>0
|
group BY a.rule_id
|
`
|
if err := db2.Raw(sqlStr).Scan(&items).Error; err != nil {
|
fmt.Println(err.Error())
|
return nil
|
}
|
|
ruleMap := make(map[int64]string)
|
for _, d2 := range items {
|
if d2.TaskId > 0 {
|
ruleMap[d2.TaskId] = d2.FileName
|
}
|
}
|
return ruleMap
|
}
|
|
type VideoLink struct {
|
ID int `gorm:"column:id" json:"id"`
|
VideoId string `gorm:"column:video_id" json:"videoId"`
|
TaskId int `gorm:"column:task_id" json:"taskId"`
|
}
|
|
func GetTasks() map[string][]interface{} {
|
|
var lists []VideoLink
|
sqlStr := `select id,video_id,task_id from mal_task_video_link`
|
if err := db2.Raw(sqlStr).Scan(&lists).Error; err != nil {
|
fmt.Println(err.Error())
|
return nil
|
}
|
|
checkMap := make(map[string][]interface{})
|
for _, d2 := range lists {
|
if d2.VideoId != "" {
|
checkMap[d2.VideoId] = append(checkMap[d2.VideoId], d2)
|
}
|
}
|
return checkMap
|
}
|
