add get buckets contrl

sunty

2020-08-20 9d88c7c467f8d93af4aab9ba0b6d6c01c2ffc546

 controllers/sdk.go

@@ -4,7 +4,6 @@
   "net/http"
   "strconv"
   "webserver/extend/code"
   "webserver/extend/config"
   "webserver/extend/util"
   "webserver/middlewares/auth"
   "webserver/service"
@@ -48,7 +47,7 @@
// @Description 算法保存
// @Accept json
// @Produce json
// @Tags sdk
// @Tags 算法
// @Param reqMap body controllers.SdkVo true "人脸检测"
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"",success:true}"
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
@@ -74,16 +73,18 @@
// @Summary 查找所有算法
// @Description 查找所有算法
// @Produce json
// @Tags sdk
// @Tags 算法
// @Param sdkName query string false "可选参数"
// @Param installed query bool true "查已安装或者查所有"
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"算法列表",success:true}"
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"[]",success:false}"
// @Router /data/api-v/sdk/findAllSdk [GET]
func (sc SdkController) FindAllSdk(c *gin.Context) {
   sdkName := c.Query("sdkName")
   installed, _ := strconv.ParseBool(c.Query("installed"))
   logger.Debug("FindAllSdk sdkName:", sdkName)

   sdks := service.GetSdkList("") //本机已安装的算法
   sdks := service.GetSdkList("", installed) //本机已安装的算法

   util.ResponseFormat(c, code.Success, sdks)
}
@@ -92,7 +93,7 @@
// @Summary 根据id获取算法信息
// @Description 根据id获取算法信息
// @Produce json
// @Tags sdk
// @Tags 算法
// @Param id query string true "算法id，必填"
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"算法信息",success:true}"
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
@@ -135,7 +136,7 @@
// @Summary 根据taskId获取算法信息
// @Description 根据taskId获取算法信息
// @Produce json
// @Tags sdk
// @Tags 算法
// @Param taskId query string true "taskId，必填"
// @Success 200 {string} json "{"code":200, msg:"请求处理成功",data:"算法信息",success:true}"
// @Failure 500 {string} json "{"code":500, msg:"请求失败",data:"",success:false}"
@@ -159,7 +160,7 @@
// @Description 算法安装包上传(分块检查)
// @Accept multipart/form-data
// @Produce json
// @Tags sdk
// @Tags 算法
// @Param chunkNumber formData int true "当前是第几个分块"
// @Param chunkSize formData int true "每一块的大小"
// @Param currentChunkSize formData int true "当前块的大小"
@@ -211,7 +212,7 @@
// @Description 算法安装包上传
// @Accept multipart/form-data
// @Produce json
// @Tags sysset
// @Tags 算法
// @Param chunkNumber formData int true "当前是第几个分块"
// @Param chunkSize formData int true "每一块的大小"
// @Param currentChunkSize formData int true "当前块的大小"
@@ -258,59 +259,114 @@
      Header:           header,
   }
   var sv service.SysService
   if b,isComplete := sv.PatchUpload(&arg); b {
      if isComplete { //上传完毕需要自动触发安装
         var sv service.SdkInstallService
         if b,err := sv.SdkInstall(identifier, filename);b {
            util.ResponseFormat(c,&code.Code{http.StatusOK, true, "算法安装成功"},"算法安装成功")
            return
         } else {
            util.ResponseFormat(c,&code.Code{http.StatusBadRequest, false, "算法安装失败"},err.Error())
            return
         }
      }
      util.ResponseFormat(c, code.Success, "")
   if b,_ := sv.PatchUpload(&arg); b {
      //if isComplete { //上传完毕需要自动触发安装
      //   var sv service.SdkInstallService
      //   if b,err := sv.SdkInstall(identifier, filename);b {
      //      util.ResponseFormat(c,&code.Code{http.StatusOK, true, "算法安装成功"},"算法安装成功")
      //      return
      //   } else {
      //      util.ResponseFormat(c,&code.Code{http.StatusBadRequest, false, "算法安装失败"},err.Error())
      //      return
      //   }
      //}
      util.ResponseFormat(c, code.Success, "上传完成")
   } else {
      util.ResponseFormat(c, code.ComError, "")
   }
}

func (sc SdkController) SdkDownLoad(c *gin.Context) {
   var soApi dbapi.SdkApi
   var sdkConfig map[string]interface{}
// @Security ApiKeyAuth
// @Summary 上传完成获取安装信息
// @Description 上传完成获取安装信息
// @Produce json
// @Tags 算法
// @Param identifier formData string true "整个文件的唯一标识，目前是md5"
// @Param filename formData string true "文件名称"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sdk/showInstallInfo [post]
func (sc SdkController) ShowInstallInfo(c *gin.Context) {
   identifier := c.Request.FormValue("identifier")
   filename := c.Request.FormValue("filename")
   var sv service.SdkInstallService
   b,d, err := sv.ShowInstallInfo(identifier, filename)
   if b {
      util.ResponseFormat(c,code.Success, d)
   } else {
      util.ResponseFormat(c,&code.Code{http.StatusInternalServerError, false, err.Error()}, err.Error())
   }
}

// @Security ApiKeyAuth
// @Summary 上传完成开始安装
// @Description 上传完成开始安装
// @Produce json
// @Tags 算法
// @Param identifier formData string true "整个文件的唯一标识，目前是md5"
// @Param filename formData string true "文件名称"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sdk/install [post]
func (sc SdkController) Install(c *gin.Context) {
   identifier := c.Request.FormValue("identifier")
   filename := c.Request.FormValue("filename")

   var sv service.SdkInstallService
   if b,err := sv.SdkInstall(identifier, filename);b {
      util.ResponseFormat(c,&code.Code{http.StatusOK, true, "算法安装成功"},"算法安装成功")
      return
   } else {
      util.ResponseFormat(c,&code.Code{http.StatusBadRequest, false, err.Error()},err.Error())
      return
   }
}

// @Security ApiKeyAuth
// @Summary 从商城下载或升级算法
// @Description 从商城下载或升级算法
// @Produce json
// @Tags 算法
// @Param path query string true "算法id"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sdk/sdkDownload [get]
func (sc SdkController) SdkDownLoad(c *gin.Context) {
   sdkId, exist := c.GetQuery("path")
   if !exist {
      util.ResponseFormat(c, code.ComError, "下载的算法参数有误")
      util.ResponseFormat(c, code.RequestParamError, "下载的算法参数有误")
      return
   }

   logger.Info(sdkId)

   // 先查找本地目录内是否已存在算法so, 没有走云端下载
   if sdkConfig = service.FindLocalSdkSoById(sdkId); sdkConfig == nil {
      sdkConfig = service.DownSdk(sdkId)
   if b,err := service.DownloadOrUpgrade(sdkId);b {
      util.ResponseFormat(c, code.Success, "算法安装成功")
   } else {
      util.ResponseFormat(c, &code.Code{http.StatusBadRequest, false, err.Error()}, err.Error())
   }

   if sdkConfig != nil {
      if regSdkSuccess, _ := soApi.Register(sdkConfig); regSdkSuccess {
         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")
      if err != nil {
         logger.Info(err)
      }
      if !flag1 {
         util.ResponseFormat(c, code.ComError, "请重新下载算法")
      }
   }

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

// @Security ApiKeyAuth
// @Summary 使用激活码激活安装算法
// @Description 使用激活码激活安装算法
// @Produce json
// @Tags 算法
// @Param code query string true "激活码"
// @Param sdkId query string true "算法id"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sdk/active [get]
func (sc SdkController) Active(c *gin.Context) {
   cod := c.GetString("code")
   sdkId := c.GetString("sdkId")
   if cod == "" || sdkId == "" {
      util.ResponseFormat(c, code.RequestParamError, "激活码和算法id不能为空")
      return
   }
   var sv service.SdkInstallService
   d, err := sv.Active(cod, sdkId)
   if err == nil {
      util.ResponseFormat(c,code.Success,  d)
   } else {
      util.ResponseFormat(c,code.ComError, err.Error())
   }
}