feat: add sdk download and regiter

zhangzengfei

2020-01-19 20ed84cad4fd237cca40619d0fe7ed39c5e030cd

 controllers/sdk.go

@@ -1,43 +1,43 @@
package controllers

import (
   "basic.com/dbapi.git"
   uuid "github.com/satori/go.uuid"
   "webserver/extend/code"
   "webserver/extend/config"
   "basic.com/valib/logger.git"
   "webserver/extend/util"
   "github.com/gin-gonic/gin"
   "webserver/service"

   "basic.com/dbapi.git"
   "basic.com/valib/logger.git"
   "github.com/gin-gonic/gin"
)

type SdkController struct {
}

type SdkVo struct {
   Id       string `json:"id"`
   SdkType string `json:"sdk_type"`//人脸检测：FaceDetect,人脸提取：FaceExtract,人脸比对：FaceCompare,行为：Yolo
   SdkName string `json:"sdk_name"`    //算法名称
   Args    []SdkArgVo `json:"args"` //算法参数
   Icon    string `json:"icon"`       //算法图标
   Url     string `json:"url"`                       //算法下载地址
   CreateTime string `json:"create_time"`
   CreateBy string `json:"create_by"`
   UpdateTime string `json:"update_time"`
   Enable bool `json:"enable"`//是否启用
   DelFlag bool `json:"del_flag"`//逻辑删除
   Id         string     `json:"id"`
   SdkType    string     `json:"sdk_type"` //人脸检测：FaceDetect,人脸提取：FaceExtract,人脸比对：FaceCompare,行为：Yolo
   SdkName    string     `json:"sdk_name"` //算法名称
   Args       []SdkArgVo `json:"args"`     //算法参数
   Icon       string     `json:"icon"`     //算法图标
   Url        string     `json:"url"`      //算法下载地址
   CreateTime string     `json:"create_time"`
   CreateBy   string     `json:"create_by"`
   UpdateTime string     `json:"update_time"`
   Enable     bool       `json:"enable"`   //是否启用
   DelFlag    bool       `json:"del_flag"` //逻辑删除
}

//算法参数定义
type SdkArgVo struct {
   Alias   string `json:"alias"`   //参数的别名
   Name  string `json:"name"`  //参数名称
   Type  string `json:"type"`  //参数类型(整数，字符串或数组)
   Must  bool   `json:"must"`  //是否必填
   Unit  string `json:"unit"`
   Range string `json:"range"` //值的范围，eg：0,100表示从0到100
   Alias        string `json:"alias"` //参数的别名
   Name         string `json:"name"`  //参数名称
   Type         string `json:"type"`  //参数类型(整数，字符串或数组)
   Must         bool   `json:"must"`  //是否必填
   Unit         string `json:"unit"`
   Range        string `json:"range"` //值的范围，eg：0,100表示从0到100
   DefaultValue string `json:"default_value"`
   Sort  int    `json:"sort"`  //参数顺序
   Sort         int    `json:"sort"` //参数顺序
}

// @Security ApiKeyAuth
@@ -54,13 +54,13 @@
   var sdk SdkVo
   var api dbapi.SdkApi

   if err := c.BindJSON(&sdk);err !=nil {
      util.ResponseFormat(c,code.RequestParamError,"参数错误")
   if err := c.BindJSON(&sdk); err != nil {
      util.ResponseFormat(c, code.RequestParamError, "参数错误")
      return
   }
   paramBody := util.Struct2Map(sdk)
   flag, data := api.Save(paramBody)
   if flag{
   if flag {
      util.ResponseFormat(c, code.Success, data)
   } else {
      util.ResponseFormat(c, code.ComError, data)
@@ -77,10 +77,9 @@
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"[]",success:false}"
// @Router /data/api-v/sdk/findAllSdk [GET]
func (sc SdkController) FindAllSdk(c *gin.Context) {
   var api dbapi.SdkApi

   sdkName := c.Query("sdkName")
   sdks := api.FindAll(sdkName)

   sdks := service.GetSdkList(sdkName)
   util.ResponseFormat(c, code.Success, sdks)
}

@@ -96,17 +95,17 @@
func (sc SdkController) GetById(c *gin.Context) {
   var api dbapi.SdkApi
   sdkId := c.Query("id")
   if sdkId == ""{
      util.ResponseFormat(c,code.RequestParamError,"sdkId不能为空")
   if sdkId == "" {
      util.ResponseFormat(c, code.RequestParamError, "sdkId不能为空")
      return
   }

   flag, sdk := api.GetById(sdkId)

   if flag {
      util.ResponseFormat(c,code.Success,sdk)
      util.ResponseFormat(c, code.Success, sdk)
   } else {
      util.ResponseFormat(c,code.ComError,sdk)
      util.ResponseFormat(c, code.ComError, sdk)
   }
}

@@ -114,16 +113,16 @@
func (sc SdkController) GetSdkArgs(c *gin.Context) {
   sdkId := c.Query("sdkId")
   scope := c.Query("scope")
   if sdkId== "" || scope == ""{
      util.ResponseFormat(c,code.RequestParamError,"参数有误")
   if sdkId == "" || scope == "" {
      util.ResponseFormat(c, code.RequestParamError, "参数有误")
      return
   }
   var api dbapi.SdkApi
   b,d := api.GetSdkArgs(sdkId, scope)
   if b{
      util.ResponseFormat(c,code.Success,d)
   b, d := api.GetSdkArgs(sdkId, scope)
   if b {
      util.ResponseFormat(c, code.Success, d)
   } else {
      util.ResponseFormat(c,code.ComError,"查询失败")
      util.ResponseFormat(c, code.ComError, "查询失败")
   }
}

@@ -140,53 +139,56 @@
   var api dbapi.SdkApi
   taskId := c.Query("taskId")
   if taskId == "" {
      util.ResponseFormat(c,code.ComError,"任务id不能为空")
      util.ResponseFormat(c, code.ComError, "任务id不能为空")
      return
   }
   flag,sdks := api.FindByTaskId(taskId)
   flag, sdks := api.FindByTaskId(taskId)
   if flag {
      util.ResponseFormat(c,code.Success,sdks)
      util.ResponseFormat(c, code.Success, sdks)
   } else {
      util.ResponseFormat(c,code.ComError,sdks)
      util.ResponseFormat(c, code.ComError, sdks)
   }
}

func (sc SdkController) SdkDownLoad(c *gin.Context) {
   path,exist := c.GetQuery("path")
   var soApi dbapi.SoApi
   var sdkConfig map[string]interface{}

   path, exist := c.GetQuery("path")
   if !exist {
      util.ResponseFormat(c,code.ComError,"下载的算法参数有误")
      util.ResponseFormat(c, code.ComError, "下载的算法参数有误")
   }

   logger.Info(path)

   // 先查找本地目录内是否已存在算法so, 没有走云端下载
   if sdkConfig = service.FindLocalSdkSoById(path); sdkConfig == nil {
      sdkConfig = service.DownSdk(path)
   }

   if sdkConfig != nil {
      regSdkSuccess, _ := soApi.Add(sdkConfig)
      regArgsSuccess, _ := soApi.Add(sdkConfig)

      if regSdkSuccess && regArgsSuccess {
         util.ResponseFormat(c, code.Success, "下载算法成功！")
         return
      }
   }

   flag := c.Query("needUpdateMiddle")
   // 下载算法（有时候也需要把中间件一起下载下来）
   if flag == "true" {
      // 下载并更新中间件,带上MD5校验
      flag1,err := service.DownSo("http://"+config.SoPath.Ip+":"+config.SoPath.Port+"/"+"middleware.so")
      logger.Debug("中间件路径：","http://"+config.SoPath.Ip+":"+config.SoPath.Port+"/"+"middleware.so")
      flag1, err := service.DownSo("http://" + config.SoPath.Ip + ":" + config.SoPath.Port + "/" + "middleware.so")
      logger.Debug("中间件路径：", "http://"+config.SoPath.Ip+":"+config.SoPath.Port+"/"+"middleware.so")
      if err != nil {
         logger.Info(err)
      }
      if !flag1 {
         util.ResponseFormat(c,code.ComError,"请重新下载算法")
         util.ResponseFormat(c, code.ComError, "请重新下载算法")
      }
   }
   // 下载算法，校验，并写入到目标目录下
   flag2,err2 := service.DownSo(path)
   if err2 != nil {
      logger.Info(err2)
   }
   if !flag2 {
      util.ResponseFormat(c,code.ComError,"请重新下载算法")
   }
   // 将算法和so名称存到规则私有的注册表
   var soApi dbapi.SoApi
   param := make(map[string]interface{})
   param["id"] = uuid.NewV4().String()
   sdkId := uuid.NewV4().String()
   param["sdkId"] = sdkId
   param["soName"] = service.GetFileNameFromUrl(path,true)
   flag3,_ := soApi.Add(param)
   if flag3 {
      util.ResponseFormat(c,code.Success,"下载算法成功！")
   }
}

   util.ResponseFormat(c, code.ComError, "算法下载失败,请重试")
}