feat: add sdk download and regiter

zhangzengfei

2020-01-19 20ed84cad4fd237cca40619d0fe7ed39c5e030cd

 controllers/sdk.go

@@ -1,14 +1,14 @@
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 {
@@ -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)
}

@@ -152,11 +151,31 @@
}

func (sc SdkController) SdkDownLoad(c *gin.Context) {
   var soApi dbapi.SoApi
   var sdkConfig map[string]interface{}

   path,exist := c.GetQuery("path")
   if !exist {
      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" {
@@ -170,23 +189,6 @@
         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, "算法下载失败,请重试")
}