package controllers
|
|
import (
|
"basic.com/dbapi.git"
|
"basic.com/pubsub/protomsg.git"
|
"encoding/json"
|
"time"
|
|
"github.com/gin-gonic/gin"
|
"webserver/extend/code"
|
"webserver/extend/util"
|
)
|
|
type TaskController struct{}
|
|
type TaskVo struct {
|
Taskid string `json:"taskid"`
|
Taskname string `json:"taskname" example:"任务一"`
|
CreateAt time.Time `json:"create_at"`
|
Createby string `json:"create_by"`
|
UpdateAt time.Time `json:"update_at"`
|
Enable bool `json:"enable"`
|
IsAlarm bool `json:"is_alarm"`
|
DelFlag bool `json:"del_flag"`
|
}
|
|
type TaskSdkVo struct {
|
Task TaskVo `json:"task"`
|
Sdks []SdkVo `json:"sdks"`
|
}
|
|
var (
|
FaceExtract_VirtualTaskId = "92496BDF-2BFA-98F2-62E8-96DD9866ABD2"//虚拟任务id
|
FaceExtract_VirtualSdkId = "virtual-faceextract-sdk-pull"//结果输出的算法id
|
)
|
|
// @Summary 查找所有任务，包含任务信息和对应的算法信息
|
// @Description 查找所有任务
|
// @Produce json
|
// @Tags task
|
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"",success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
|
// @Router /data/api-v/task/findAll [GET]
|
func (tc TaskController) FindAll(c *gin.Context) {
|
// 显示所有任务: 获取信息
|
var taskApi dbapi.TaskApi
|
|
taskInfos := taskApi.FindAll()
|
var arr []protomsg.TaskSdkInfo
|
for _,ti :=range taskInfos{
|
if ti.Task.Taskid != FaceExtract_VirtualTaskId{
|
arr = append(arr,ti)
|
}
|
}
|
var tasks []TaskSdkVo
|
dataBytes, err := json.Marshal(arr)
|
if err !=nil {
|
util.ResponseFormat(c,code.ComError,[]TaskSdkVo{})
|
} else {
|
if err := json.Unmarshal(dataBytes, &tasks);err !=nil {
|
util.ResponseFormat(c,code.ComError,[]TaskSdkVo{})
|
} else {
|
util.ResponseFormat(c,code.Success,tasks)
|
}
|
}
|
}
|
|
// @Summary 添加任务
|
// @Description 查找所有任务
|
// @Accept json
|
// @Produce json
|
// @Tags task
|
// @Param reqMap body controllers.TaskVo false "任务AAA"
|
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"",success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
|
// @Router /data/api-v/task/add [POST]
|
func (tc TaskController) AddTask(c *gin.Context) {
|
var task TaskVo
|
var taskApi dbapi.TaskApi
|
err := c.BindJSON(&task)
|
if err !=nil {
|
util.ResponseFormat(c,code.RequestParamError,"参数有误")
|
return
|
}
|
paramBody := util.Struct2Map(task)
|
flag, data := taskApi.Add(paramBody)
|
if !flag {
|
util.ResponseFormat(c, code.ComError, data)
|
return
|
}
|
|
util.ResponseFormat(c, code.Success, data)
|
}
|
|
// @Summary 给任务添加算法
|
// @Description 任务添加算法
|
// @Produce json
|
// @Tags task
|
// @Param taskId query string true "任务id"
|
// @Param sdkId query string true "算法id"
|
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"",success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
|
// @Router /data/api-v/task/addTaskSdk [GET]
|
func (tc TaskController) AddTaskSdk(c *gin.Context) {
|
var taskApi dbapi.TaskApi
|
taskId := c.Query("taskId")
|
sdkId := c.Query("sdkId")
|
if taskId == "" || sdkId == "" {
|
util.ResponseFormat(c,code.RequestParamError,"参数有误")
|
return
|
}
|
flag,data := taskApi.AddTaskSdk(taskId,sdkId)
|
if flag {
|
util.ResponseFormat(c, code.ComError, data)
|
} else {
|
util.ResponseFormat(c,code.Success,data)
|
}
|
}
|
|
// @Summary 任务删除算法
|
// @Description 根据taskid和sdkid删除
|
// @Produce json
|
// @Tags task
|
// @Param taskId path string true "任务id"
|
// @Param sdkId path string true "算法id"
|
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"",success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
|
// @Router /data/api-v/task/delTaskSdk [GET]
|
func (tc TaskController) DeleteTaskSdk(c *gin.Context) {
|
var taskApi dbapi.TaskApi
|
taskId := c.Query("taskId")
|
sdkId := c.Query("sdkId")
|
if taskId == "" || sdkId == "" {
|
util.ResponseFormat(c,code.RequestParamError,"参数有误")
|
return
|
}
|
flag,data := taskApi.DeleteTaskSdk(taskId,sdkId)
|
if flag {
|
util.ResponseFormat(c, code.ComError, data)
|
} else {
|
util.ResponseFormat(c,code.Success,data)
|
}
|
}
|
|
// @Summary 更新任务状态
|
// @Description （算法不变,只更新任务状态）
|
// @Produce json
|
// @Tags task
|
// @Param taskId path string true "任务id"
|
// @Param taskName path string true "任务名称"
|
// @Param isAlarm path bool true "是否报警"
|
// @Param isEnable path bool true "是否停用"
|
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"",success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
|
// @Router /data/api-v/task/updateTaskStatus [POST]
|
func (tc TaskController) UpdateTaskStatus(c *gin.Context) {
|
var taskApi dbapi.TaskApi
|
paramBody :=make(map[string]string,0)
|
paramBody["taskId"] = c.Query("taskId")
|
paramBody["taskName"] = c.Query("taskName")
|
paramBody["isAlarm"] = c.Query("isAlarm")
|
paramBody["isEnable"] = c.Query("isEnable")
|
flag, data := taskApi.UpdateTaskStatus(paramBody)
|
if flag {
|
util.ResponseFormat(c,code.Success,data)
|
} else {
|
util.ResponseFormat(c,code.ComError,data)
|
}
|
}
|
|
// @Summary 删除任务
|
// @Description 根据任务id删除任务
|
// @Produce json
|
// @Tags task
|
// @Param taskId path string true "任务id"
|
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"",success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
|
// @Router /data/api-v/task/delete [GET]
|
func (tc TaskController) DeleteTask(c *gin.Context) {
|
var taskApi dbapi.TaskApi
|
taskId := c.Query("taskId")
|
if taskId == "" {
|
util.ResponseFormat(c,code.RequestParamError,"参数有误")
|
return
|
}
|
flag, data := taskApi.Delete(taskId)
|
if flag {
|
util.ResponseFormat(c,code.Success,data)
|
} else {
|
util.ResponseFormat(c,code.ComError,data)
|
}
|
|
}
|
