add discovery

liuxiaolong

2019-07-25 f88a44f58396fb36a979cfe5640a36d6ac2d7289

 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"//结果输出的算法id
)

// @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 添加任务