package controllers
|
|
import (
|
"bytes"
|
"encoding/json"
|
"strconv"
|
"vamicro/appcenter-service/models"
|
"vamicro/appcenter-service/service"
|
"vamicro/appcenter-service/vo"
|
|
"basic.com/pubsub/protomsg.git"
|
"basic.com/valib/bhomeclient.git"
|
"basic.com/valib/logger.git"
|
)
|
|
type SdkController struct {
|
}
|
|
// @Summary 算法注册
|
// @Description 算法注册
|
// @Accept json
|
// @Produce json
|
// @Tags sdk
|
// @Param reqMap body vo.SdkVo 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/register [POST]
|
func (sc SdkController) Register(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var reqBody vo.SdkRegisterVo
|
|
c.BindJSON(&reqBody)
|
if reqBody.Id == "" || reqBody.SdkType == "" || reqBody.SdkName == "" || reqBody.Icon == "" {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
logger.Debug("sdk register paramBody:", reqBody)
|
sv := service.NewSdkService(h.Bk)
|
if sv.Register(&reqBody) {
|
return &bhomeclient.Reply{Success: true, Msg: "算法注册成功"}
|
} else {
|
return &bhomeclient.Reply{Success: false, Msg: "注册失败"}
|
}
|
}
|
|
// @Summary 查找所有算法
|
// @Description 查找所有算法
|
// @Produce json
|
// @Tags sdk
|
// @Param sdkName query string false "可选参数"
|
// @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(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
sdkName := c.Query("sdkName")
|
insStr := c.Query("installed") //如果未传此参数，则表示查所有
|
userId := c.Header("Login_user_id")
|
ok := false
|
if insStr != "" {
|
ok = true
|
}
|
logger.Debugf("FindAllSdk sdkName=%v, insStr=%v, ok=%v, userId=%v", sdkName, insStr, ok, userId)
|
installed, _ := strconv.ParseBool(insStr)
|
|
sdks := service.GetSdkList(sdkName, insStr, userId) //本机已安装的算法
|
if ok && !installed { //传了installed参数
|
nSdks := make([]service.SdkInsOrUpgrade, 0)
|
for _, s := range sdks {
|
if !s.Installed {
|
nSdks = append(nSdks, s)
|
}
|
}
|
return &bhomeclient.Reply{Success: true, Data: nSdks}
|
} else {
|
return &bhomeclient.Reply{Success: true, Data: sdks}
|
}
|
}
|
|
// @Summary 根据id获取算法信息
|
// @Description 根据id获取算法信息
|
// @Produce json
|
// @Tags sdk
|
// @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}"
|
// @Router /data/api-v/sdk/getById [GET]
|
func (sc SdkController) GetById(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var sdk models.Sdk
|
|
id := c.Query("id")
|
if id == "" {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
rows, err := sdk.SelectById(id)
|
if err != nil || rows == 0 {
|
return &bhomeclient.Reply{Success: false, Msg: "记录不存在"}
|
}
|
|
var sdkVo = vo.SdkVo{}
|
sdkVo.Sdk = sdk
|
sdkVo.Args = sdk.GetSdkArgs(id, models.SCOPE_CAMERARULE)
|
return &bhomeclient.Reply{Success: true, Data: sdkVo}
|
}
|
|
// @Summary 根据taskId获取算法信息
|
// @Description 根据taskId获取算法信息
|
// @Produce json
|
// @Tags sdk
|
// @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}"
|
// @Router /data/api-v/sdk/findByTaskId [GET]
|
func (sc SdkController) FindByTaskId(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
taskId := c.Query("taskId")
|
if taskId == "" {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
var sdk models.Sdk
|
sdks, err := sdk.FindByTaskId(taskId)
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "查询失败"}
|
}
|
var sdkVoArr []vo.SdkVo
|
for _, sdk := range sdks {
|
sdkVo := vo.SdkVo{}
|
sdkVo.Sdk = sdk
|
sdkVo.Args = sdk.GetSdkArgs(sdk.Id, models.SCOPE_CAMERARULE)
|
sdkVoArr = append(sdkVoArr, sdkVo)
|
}
|
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "记录不存在"}
|
} else {
|
return &bhomeclient.Reply{Success: true, Data: sdkVoArr}
|
}
|
}
|
|
// @Summary 获取所有算法通道数量设置
|
// @Description 获取所有算法通道数量设置
|
// @Produce json
|
// @Tags sdk
|
// @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/getAllSdkChanSet [GET]
|
func (sc SdkController) GetAllSdkChanSet(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var scs models.SdkChanSet
|
list, err := scs.FindAll()
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "内部错误"}
|
} else {
|
return &bhomeclient.Reply{Success: true, Data: list}
|
}
|
}
|
|
type SdkChanUpdateArg struct {
|
SdkType string `json:"sdk_type" binding:"required"`
|
ChanNum int `json:"chan_num" binding:"required"`
|
}
|
|
// @Summary 更新算法通道数量设置
|
// @Description 更新算法通道数量设置
|
// @Produce json
|
// @Tags sdk
|
// @Param reqBody body controllers.SdkChanUpdateArg 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/updateSdkChanSet [POST]
|
func (sc SdkController) UpdateSdkChanSet(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var reqBody SdkChanUpdateArg
|
err := c.BindJSON(&reqBody)
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
var scs = models.SdkChanSet{
|
SdkType: reqBody.SdkType,
|
ChanNum: reqBody.ChanNum,
|
}
|
b, _ := scs.Update()
|
if b {
|
pMsg := protomsg.DbChangeMessage{
|
Table: protomsg.TableChanged_T_SdkChanSet,
|
Action: protomsg.DbAction_Update,
|
}
|
pb, _ := json.Marshal(pMsg)
|
h.Bk.Publish("sdk-service", pb)
|
return &bhomeclient.Reply{Success: true, Msg: "更新成功"}
|
} else {
|
return &bhomeclient.Reply{Success: false, Msg: "更新失败"}
|
}
|
}
|
|
// @Summary 更新算法默认场景参数
|
// @Description 更新算法默认场景参数
|
// @Produce json
|
// @Tags sdk
|
// @Param reqBody body vo.SdkArgDefault 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/updateSdkArgConfig [POST]
|
func (sc SdkController) UpdateSdkArgConfig(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var reqBody vo.SdkArgDefault
|
err := c.BindJSON(&reqBody)
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
var sdk models.Sdk
|
rows, err := sdk.SelectById(reqBody.SdkId)
|
if err != nil || rows == 0 {
|
return &bhomeclient.Reply{Success: false, Msg: "算法id不存在"}
|
}
|
bf := bytes.NewBuffer([]byte{})
|
jsonEncoder := json.NewEncoder(bf)
|
jsonEncoder.SetEscapeHTML(false)
|
jsonEncoder.Encode(reqBody.DefArgs)
|
sdk.ArgDef = bf.String()
|
b, err := sdk.Update()
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: err.Error()}
|
}
|
if b {
|
pMsg := protomsg.DbChangeMessage{
|
Table: protomsg.TableChanged_T_Sdk,
|
Action: protomsg.DbAction_Update,
|
}
|
pb, _ := json.Marshal(pMsg)
|
h.Bk.Publish("sdk-service", pb)
|
return &bhomeclient.Reply{Success: true, Msg: "更新成功"}
|
} else {
|
return &bhomeclient.Reply{Success: false, Msg: "更新失败"}
|
}
|
}
|
|
// @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(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
chunkNumber, e1 := strconv.Atoi(c.Query("chunkNumber"))
|
chunkSize, e2 := strconv.Atoi(c.Query("chunkSize"))
|
currentChunkSize, e3 := strconv.Atoi(c.Query("currentChunkSize"))
|
identifier := c.Query("identifier")
|
filename := c.Query("filename")
|
relativePath := c.Query("relativePath")
|
totalChunks, e5 := strconv.Atoi(c.Query("totalChunks"))
|
if identifier == "" || e1 != nil || e2 != nil || e3 != nil || e5 != nil {
|
return &bhomeclient.Reply{Msg: "参数有误"}
|
}
|
userId := c.Header("Login_user_id")
|
var arg = service.FileChunkCheckVo{
|
UserId: userId,
|
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")
|
return &bhomeclient.Reply{Success: true, Data: "found"}
|
} else {
|
//c.String(http.StatusNoContent,"")
|
return &bhomeclient.Reply{Success: true, Data: ""}
|
}
|
}
|
|
// @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(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
chunkNumber, e1 := strconv.Atoi(c.PostForm("chunkNumber"))
|
chunkSize, e2 := strconv.Atoi(c.PostForm("chunkSize"))
|
currentChunkSize, e3 := strconv.Atoi(c.PostForm("currentChunkSize"))
|
totalSize, e4 := strconv.ParseInt(c.PostForm("totalSize"), 10, 64)
|
identifier := c.PostForm("identifier")
|
filename := c.PostForm("filename")
|
relativePath := c.PostForm("relativePath")
|
totalChunks, e5 := strconv.Atoi(c.PostForm("totalChunks"))
|
file, e6 := c.FormFile()
|
//file, header, e6 := c.Request.FormFile("file")
|
if identifier == "" || e1 != nil || e2 != nil || e3 != nil || e4 != nil || e5 != nil || e6 != nil {
|
return &bhomeclient.Reply{Msg: "参数有误"}
|
}
|
userId := c.Header("Login_user_id")
|
var arg = service.FileUploadVo{
|
UserId: userId,
|
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, _, msg := 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
|
// }
|
//}
|
return &bhomeclient.Reply{Success: true, Msg: "上传完成"}
|
} else {
|
return &bhomeclient.Reply{Msg: msg}
|
}
|
}
|
|
// @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(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
identifier := c.Query("identifier")
|
filename := c.Query("filename")
|
logger.Debug("showInstallInfo identifier:", identifier, "filename:", filename)
|
if identifier == "" || filename == "" {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
var sv service.SdkInstallService
|
b, d, err := sv.ShowInstallInfo(identifier, filename)
|
if b {
|
return &bhomeclient.Reply{Success: true, Data: d}
|
} else {
|
return &bhomeclient.Reply{Msg: 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(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
identifier := c.Query("identifier")
|
filename := c.Query("filename")
|
logger.Debug("showInstallInfo identifier:", identifier, "filename:", filename)
|
if identifier == "" || filename == "" {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
|
var sv service.SdkInstallService
|
if b, err := sv.PackInstall(identifier, filename, h); b {
|
return &bhomeclient.Reply{Success: true, Msg: "安装成功"}
|
} else {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
}
|
|
// @Security ApiKeyAuth
|
// @Summary 卸载算法
|
// @Description 卸载算法
|
// @Produce json
|
// @Tags 算法
|
// @Param reqBody body controllers.DelSdk true "卸载参数"
|
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
|
// @Router /data/api-v/sdk/unInstall [post]
|
func (sc SdkController) UnInstall(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
var reqBody DelSdk
|
err := c.BindJSON(&reqBody)
|
if err != nil {
|
return &bhomeclient.Reply{Success: false, Msg: "参数有误"}
|
}
|
var sdkE models.Sdk
|
if sdkE.DeleteById(reqBody.SdkId) {
|
pMsg := protomsg.DbChangeMessage{
|
Table: protomsg.TableChanged_T_Sdk,
|
Action: protomsg.DbAction_Delete,
|
}
|
pb, _ := json.Marshal(pMsg)
|
h.Bk.Publish(service.ProcName, pb)
|
return &bhomeclient.Reply{Success: true, Msg: "卸载成功"}
|
} else {
|
return &bhomeclient.Reply{Success: false, Msg: "卸载失败"}
|
}
|
}
|
|
// @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(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
id := c.Query("path")
|
if id == "" {
|
return &bhomeclient.Reply{Msg: "下载算法参数有误"}
|
}
|
|
if service.IsInstalling(id) {
|
return &bhomeclient.Reply{Success: true, Msg: "安装中，请稍候..."}
|
}
|
|
if b, err := service.DownloadOrUpgrade(id, h); b {
|
return &bhomeclient.Reply{Success: true, Msg: "算法安装成功"}
|
} else {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
}
|
|
// @Security ApiKeyAuth
|
// @Summary 应用中心中未购买的算法列表
|
// @Description 应用中心中未购买的算法列表
|
// @Produce json
|
// @Tags 算法
|
// @Success 200 {string} json "{"code":200, msg:"", success:true}"
|
// @Failure 500 {string} json "{"code":500, msg:"", success:false}"
|
// @Router /data/api-v/sdk/buyList [get]
|
func (sc SdkController) BuyList(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
sdkName := c.Query("sdkName")
|
logger.Debug("BuyList sdkName:", sdkName)
|
var sv service.SdkInstallService
|
list := sv.GetBuyList(sdkName)
|
if list == nil {
|
list = make([]service.SdkWithArg, 0)
|
}
|
|
return &bhomeclient.Reply{Success: true, Data: list}
|
}
|
|
// @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(h *bhomeclient.WrapperHandler, c *bhomeclient.Request) *bhomeclient.Reply {
|
cod := c.Query("code")
|
sdkId := c.Query("sdkId")
|
if sdkId == "" {
|
return &bhomeclient.Reply{Msg: "算法id不能为空"}
|
}
|
var sv service.SdkInstallService
|
d, err := sv.Active(cod, sdkId)
|
if err == nil {
|
// 激活成功
|
var sdk models.Sdk
|
if err := sdk.UpdateCode(sdkId, cod); err != nil {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
return &bhomeclient.Reply{Success: true, Data: d}
|
} else {
|
return &bhomeclient.Reply{Msg: err.Error()}
|
}
|
}
|
|
type DelSdk struct {
|
SdkId string `json:"sdkId" binding:"required"`
|
}
|
