From f88a44f58396fb36a979cfe5640a36d6ac2d7289 Mon Sep 17 00:00:00 2001 From: liuxiaolong <736321739@qq.com> Date: 星期四, 25 七月 2019 14:50:56 +0800 Subject: [PATCH] add discovery --- controllers/task.go | 32 ++++++++++++++++++++++++++++++-- 1 files changed, 30 insertions(+), 2 deletions(-) diff --git a/controllers/task.go b/controllers/task.go index 07420dc..b70e24b 100644 --- a/controllers/task.go +++ b/controllers/task.go @@ -2,11 +2,13 @@ import ( "basic.com/dbapi.git" + "basic.com/pubsub/protomsg.git" + "encoding/json" "time" + "github.com/gin-gonic/gin" "webserver/extend/code" "webserver/extend/util" - "github.com/gin-gonic/gin" ) type TaskController struct{} @@ -22,6 +24,16 @@ 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"//缁撴灉杈撳嚭鐨勭畻娉昳d +) + // @Summary 鏌ユ壘鎵�鏈変换鍔★紝鍖呭惈浠诲姟淇℃伅鍜屽搴旂殑绠楁硶淇℃伅 // @Description 鏌ユ壘鎵�鏈変换鍔� // @Produce json @@ -34,7 +46,23 @@ var taskApi dbapi.TaskApi taskInfos := taskApi.FindAll() - util.ResponseFormat(c, code.Success, taskInfos) + 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 娣诲姞浠诲姟 -- Gitblit v1.8.0