add get buckets contrl

sunty

2020-08-20 9303b69ea569bcb5e581147543a3fd58e90d0d25

 controllers/sdk.go

@@ -1,9 +1,11 @@
package controllers

import (
   "net/http"
   "strconv"
   "webserver/extend/code"
   "webserver/extend/config"
   "webserver/extend/util"
   "webserver/middlewares/auth"
   "webserver/service"

   "basic.com/dbapi.git"
@@ -59,7 +61,7 @@
      return
   }
   paramBody := util.Struct2Map(sdk)
   flag, data := api.Save(paramBody)
   flag, data := api.Register(paramBody)
   if flag {
      util.ResponseFormat(c, code.Success, data)
   } else {
@@ -78,8 +80,10 @@
// @Router /data/api-v/sdk/findAllSdk [GET]
func (sc SdkController) FindAllSdk(c *gin.Context) {
   sdkName := c.Query("sdkName")
   logger.Debug("FindAllSdk sdkName:", sdkName)

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

   util.ResponseFormat(c, code.Success, sdks)
}

@@ -150,45 +154,145 @@
   }
}

func (sc SdkController) SdkDownLoad(c *gin.Context) {
   var soApi dbapi.SoApi
   var sdkConfig map[string]interface{}
// @Summary 算法安装包上传(分块检查)
// @Description 算法安装包上传(分块检查)
// @Accept multipart/form-data
// @Produce json
// @Tags sdk
// @Param chunkNumber formData int true "当前是第几个分块"
// @Param chunkSize formData int true "每一块的大小"
// @Param currentChunkSize formData int true "当前块的大小"
// @Param identifier formData string true "整个文件的唯一标识，目前是md5"
// @Param filename formData string true "文件名称"
// @Param relativePath formData string true "文件在客户端电脑的路径"
// @Param totalChunks formData int true "文件总块数"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sdk/upload [get]
func (sc SdkController) Upload(c *gin.Context) {
   authDriver := auth.GenerateAuthDriver()
   userM := (*authDriver).User(c)
   if userM ==nil {
      util.ResponseFormat(c,code.TokenNotFound,"登录过期，请登录")
      return
   }
   chunkNumber, e1 := strconv.Atoi(c.Request.FormValue("chunkNumber"))
   chunkSize, e2 := strconv.Atoi(c.Request.FormValue("chunkSize"))
   currentChunkSize, e3 := strconv.Atoi(c.Request.FormValue("currentChunkSize"))
   identifier := c.Request.FormValue("identifier")
   filename := c.Request.FormValue("filename")
   relativePath := c.Request.FormValue("relativePath")
   totalChunks, e5 := strconv.Atoi(c.Request.FormValue("totalChunks"))
   if identifier == "" || e1 != nil || e2 != nil || e3 != nil || e5 !=nil {
      util.ResponseFormat(c,code.RequestParamError,"参数有误")
      return
   }
   var arg = service.FileChunkCheckVo {
      UserId: userM["id"].(string),
      FileName: filename,
      Identifier: identifier,
      RelativePath: relativePath,
      TotalChunks: totalChunks,
      ChunkNumber: chunkNumber,
      ChunkSize: chunkSize,
      CurrentChunkSize:currentChunkSize,
   }
   var sv service.SysService
   if b := sv.CheckUpdateFile(&arg);b {
      c.String(http.StatusOK,"found")
   } else {
      c.String(http.StatusNoContent,"")
   }
}

// @Security ApiKeyAuth
// @Summary 算法安装包上传
// @Description 算法安装包上传
// @Accept multipart/form-data
// @Produce json
// @Tags sysset
// @Param chunkNumber formData int true "当前是第几个分块"
// @Param chunkSize formData int true "每一块的大小"
// @Param currentChunkSize formData int true "当前块的大小"
// @Param totalSize formData string true "文件总大小"
// @Param identifier formData string true "整个文件的唯一标识，目前是md5"
// @Param filename formData string true "文件名称"
// @Param relativePath formData string true "文件在客户端电脑的路径"
// @Param totalChunks formData int true "文件总块数"
// @Param file formData file true "文件分片内容"
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
// @Router /data/api-v/sdk/upload [post]
func (sc SdkController) UploadPack(c *gin.Context) {
   authDriver := auth.GenerateAuthDriver()
   userM := (*authDriver).User(c)
   if userM == nil {
      util.ResponseFormat(c, code.TokenNotFound, "登录过期，请登录")
      return
   }
   chunkNumber, e1 := strconv.Atoi(c.Request.FormValue("chunkNumber"))
   chunkSize, e2 := strconv.Atoi(c.Request.FormValue("chunkSize"))
   currentChunkSize, e3 := strconv.Atoi(c.Request.FormValue("currentChunkSize"))
   totalSize, e4 := strconv.ParseInt(c.Request.FormValue("totalSize"), 10, 64)
   identifier := c.Request.FormValue("identifier")
   filename := c.Request.FormValue("filename")
   relativePath := c.Request.FormValue("relativePath")
   totalChunks, e5 := strconv.Atoi(c.Request.FormValue("totalChunks"))
   file, header, e6 := c.Request.FormFile("file")
   if identifier == "" || e1 != nil || e2 != nil || e3 != nil || e4 != nil || e5 != nil || e6 != nil {
      util.ResponseFormat(c, code.RequestParamError, "参数有误")
      return
   }
   var arg = service.FileUploadVo{
      UserId:           userM["id"].(string),
      FileName:         filename,
      Identifier:       identifier,
      RelativePath:     relativePath,
      TotalSize:        totalSize,
      TotalChunks:      totalChunks,
      ChunkNumber:      chunkNumber,
      ChunkSize:        chunkSize,
      CurrentChunkSize: currentChunkSize,
      File:             &file,
      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, "")
   } else {
      util.ResponseFormat(c, code.ComError, "")
   }
}

// @Security ApiKeyAuth
// @Summary 从商城下载或升级算法
// @Description 从商城下载或升级算法
// @Produce json
// @Tags sysset
// @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 {
      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")
      if err != nil {
         logger.Info(err)
      }
      if !flag1 {
         util.ResponseFormat(c, code.ComError, "请重新下载算法")
      }
   }

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