From 9303b69ea569bcb5e581147543a3fd58e90d0d25 Mon Sep 17 00:00:00 2001 From: sunty <1172534965@qq.com> Date: 星期四, 20 八月 2020 20:05:23 +0800 Subject: [PATCH] add get buckets contrl --- controllers/cameraTask.go | 56 +++++++++++++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 53 insertions(+), 3 deletions(-) diff --git a/controllers/cameraTask.go b/controllers/cameraTask.go index 52a817c..7ae01fb 100644 --- a/controllers/cameraTask.go +++ b/controllers/cameraTask.go @@ -2,9 +2,14 @@ import ( "basic.com/dbapi.git" + "basic.com/pubsub/esutil.git" + "encoding/json" "fmt" "github.com/gin-gonic/gin" + "strconv" + "webserver/cache" "webserver/extend/code" + "webserver/extend/config" "webserver/extend/util" "webserver/models" ) @@ -90,7 +95,52 @@ var api dbapi.CameraTaskApi flag, data := api.FindTasksByCameraIds(mulCamera.CameraIds) if flag { - util.ResponseFormat(c,code.Success,data) + bytes, _ := json.Marshal(data) + var camTInfo []CameraTaskInfoVo + json.Unmarshal(bytes, &camTInfo) + + //澶勭悊宸茶鍒犻櫎鐨勪换鍔� + + var delTasks []models.Task + var taskApi dbapi.TaskApi + taskInfos := taskApi.FindAll() + m := make(map[string]string) + for _,ti :=range taskInfos{ + if ti.Task.Taskid != FaceExtract_VirtualTaskId{ + m[ti.Task.Taskid] = ti.Task.Taskid + } + } + localConf, _ := cache.GetServerInfo() + if localConf.AlarmIp != "" && localConf.ServerId != "" && localConf.AlarmPort>0 { + indexName := config.EsInfo.EsIndex.AiOcean.IndexName + esTaskM, e := esutil.AggregateTaskList(localConf.AlarmIp, strconv.Itoa(int(localConf.AlarmPort)), indexName, localConf.ServerId) + if e==nil && esTaskM !=nil { + for _,tM :=range esTaskM { + if _,ok := m[tM["taskId"].(string)];!ok {//琛ㄧず姝や换鍔″凡琚垹闄� + taskId := tM["taskId"].(string) + m[taskId] = taskId + delTasks = append(delTasks, models.Task{ + Taskid: taskId, + Taskname: tM["taskName"].(string), + DelFlag: true, + }) + } + } + } + } + var result []CameraTaskInfoVo + if camTInfo != nil { + for _, camIn := range camTInfo { + nIfn := CameraTaskInfoVo{ + Camera: camIn.Camera, + Polygon: camIn.Polygon, + Tasks: append(camIn.Tasks, delTasks...), + } + result = append(result, nIfn) + } + } + + util.ResponseFormat(c,code.Success, result) } else { util.ResponseFormat(c,code.ComError,data) } @@ -160,7 +210,7 @@ // @Success 200 {string} json "{"code":200, success:true, msg:"璇锋眰澶勭悊鎴愬姛", data:"娣诲姞鐨勪换鍔′俊鎭�"}" // @Failure 500 {string} json "{"code":500, success:false, msg:"",data:"閿欒淇℃伅鍐呭"}" // @Router /data/api-v/camera/saveTask [post] -func (ac CameraController) CameraTaskSave(c *gin.Context) { +func (cc CameraController) CameraTaskSave(c *gin.Context) { var saveBody CameraTaskSaveArg err := c.BindJSON(&saveBody) if err != nil { @@ -218,7 +268,7 @@ // @Success 200 {string} json "{"code":200, success:true, msg:"璇锋眰澶勭悊鎴愬姛", data:"鍒犻櫎鐨勬憚鍍忔満淇℃伅"}" // @Failure 500 {string} json "{"code":500, success:false, msg:"",data:"閿欒淇℃伅鍐呭"}" // @Router /data/api-v/camera/delTask/{cameraId}/{taskId} [delete] -func (ac CameraController) CameraDelTask(c *gin.Context) { +func (cc CameraController) CameraDelTask(c *gin.Context) { cameraId := c.Param("cameraId") taskId := c.Param("taskId") if cameraId == "" || taskId == "" { -- Gitblit v1.8.0