| | |
| | | |
| | | 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" |
| | | ) |
| | | |
| | | type SdkController struct { |
| | |
| | | Name string `json:"name"` //参数名称 |
| | | Type string `json:"type"` //参数类型(整数,字符串或数组) |
| | | Must bool `json:"must"` //是否必填 |
| | | Unit string `json:"unit"` |
| | | Range string `json:"range"` //值的范围,eg:0,100表示从0到100 |
| | | DefaultValue string `json:"default_value"` |
| | | Sort int `json:"sort"` //参数顺序 |
| | | } |
| | | |
| | | // @Security ApiKeyAuth |
| | | // @Summary 算法保存 |
| | | // @Description 算法保存 |
| | | // @Accept json |
| | | // @Produce json |
| | | // @Tags sdk |
| | | // @Param reqMap body controllers.SdkVo true "人脸检测" |
| | |
| | | } |
| | | } |
| | | |
| | | // @Security ApiKeyAuth |
| | | // @Summary 查找所有算法 |
| | | // @Description 查找所有算法 |
| | | // @Produce json |
| | |
| | | util.ResponseFormat(c, code.Success, sdks) |
| | | } |
| | | |
| | | // @Security ApiKeyAuth |
| | | // @Summary 根据id获取算法信息 |
| | | // @Description 根据id获取算法信息 |
| | | // @Produce json |
| | |
| | | |
| | | // @Router /data/api-v/sdkArg/getSdkArgs [get] |
| | | func (sc SdkController) GetSdkArgs(c *gin.Context) { |
| | | sdkId := c.Param("sdkId") |
| | | scope := c.Param("scope") |
| | | sdkId := c.Query("sdkId") |
| | | scope := c.Query("scope") |
| | | if sdkId== "" || scope == ""{ |
| | | util.ResponseFormat(c,code.RequestParamError,"参数有误") |
| | | return |
| | |
| | | } |
| | | } |
| | | |
| | | // @Security ApiKeyAuth |
| | | // @Summary 根据taskId获取算法信息 |
| | | // @Description 根据taskId获取算法信息 |
| | | // @Produce json |
| | |
| | | } else { |
| | | util.ResponseFormat(c,code.ComError,sdks) |
| | | } |
| | | } |
| | | |
| | | func (sc SdkController) SdkDownLoad(c *gin.Context) { |
| | | path,exist := c.GetQuery("path") |
| | | if !exist { |
| | | util.ResponseFormat(c,code.ComError,"下载的算法参数有误") |
| | | } |
| | | logger.Info(path) |
| | | 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,"请重新下载算法") |
| | | } |
| | | } |
| | | // 下载算法,校验,并写入到目标目录下 |
| | | 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,"下载算法成功!") |
| | | } |
| | | } |