add get buckets contrl

sunty

2020-08-20 9d88c7c467f8d93af4aab9ba0b6d6c01c2ffc546

 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"
@@ -45,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}"
@@ -71,15 +73,19 @@
// @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(sdkName)
   sdks := service.GetSdkList("", installed) //本机已安装的算法

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

@@ -87,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}"
@@ -130,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}"
@@ -150,42 +156,217 @@
   }
}

func (sc SdkController) SdkDownLoad(c *gin.Context) {
   var soApi dbapi.SdkApi
   var sdkConfig map[string]interface{}
// @Summary 算法安装包上传(分块检查)
// @Description 算法安装包上传(分块检查)
// @Accept multipart/form-data
// @Produce json
// @Tags 算法
// @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 算法
// @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,_ := 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 算法
// @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())
   }
}