From 0bb9a4c37862e5098a596e121fb13fd12c2ca44f Mon Sep 17 00:00:00 2001 From: liuxiaolong <liuxiaolong@aiotlink.com> Date: 星期五, 29 五月 2020 14:14:15 +0800 Subject: [PATCH] install sdk tar.gz --- service/SdkInstall.go | 229 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 229 insertions(+), 0 deletions(-) diff --git a/service/SdkInstall.go b/service/SdkInstall.go index 6d43c33..3ad4557 100644 --- a/service/SdkInstall.go +++ b/service/SdkInstall.go @@ -1 +1,230 @@ package service + +import ( + "basic.com/dbapi.git" + "basic.com/valib/logger.git" + "encoding/json" + "errors" + "io/ioutil" + "os" + "path" + "webserver/extend/config" + "webserver/extend/util" +) + +//绠楁硶瀹夎鍖呭畨瑁� +func (sv SysService) SdkInstall(identifier string,filename string) (bool,error) { + configPatchPath := "" + if config.Server.PatchPath != "" { + configPatchPath = config.Server.PatchPath + } else { + configPatchPath = "/opt/vasystem/patch" + } + + filenameWithSuffix := path.Base(filename) + ext := path.Ext(filenameWithSuffix) + + zipFilePath := configPatchPath + "/"+identifier+ext + if util.Exists(zipFilePath) { + strMd5, e := util.FileMd5(zipFilePath) + if e !=nil || strMd5 == "" { + return false,errors.New("鑾峰彇瀹夎鍖卪d5澶辫触") + } + if strMd5 == identifier { + if !installSdk(identifier, ext) { + return false,errors.New("鎵ц瀹夎杩囩▼寮傚父,璇风‘瀹氫笂浼犵殑瀹夎鍖呮槸tar.gz鏍煎紡") + } + return true,nil + + } else { + logger.Debug("strMd5 is", strMd5,"identifier is",identifier,"not equal") + return false,errors.New("鏍¢獙瀹夎鏂囦欢澶辫触") + } + } else { + return false,errors.New("瀹夎鍖呭凡涓㈠け锛岃閲嶆柊涓婁紶") + } +} + +//瀹夎sdk +func installSdk(identifier string, ext string) bool { + configPatchPath := "" + if config.Server.PatchPath != "" { + configPatchPath = config.Server.PatchPath + } else { + configPatchPath = "/opt/vasystem/patch" + } + //1.瑙e帇缂╂洿鏂板寘 + unPackPath := configPatchPath+"/"+identifier+"_basic/" + if util.Exists(unPackPath) { + //姝ょ増鏈凡缁忔洿鏂拌繃 + rmErr := os.RemoveAll(unPackPath) + if rmErr !=nil { + return false + } + } + if !util.CreateDirectory(unPackPath) { + return false + } + + unPackFilePath := configPatchPath+"/"+identifier+ext + err := util.UnTarGz(unPackFilePath, unPackPath) + if err !=nil { + logger.Debug("UnPack err:",err,"unPackFile:",unPackFilePath) + return false + } + //瑙e帇瀹屾垚锛岃幏鍙栧畨瑁呭寘涓殑鏂囦欢锛屽紑濮嬪畨瑁� + //1.瑙f瀽瀹夎璇存槑ins.inc + incPath := unPackPath+"ins.inc" + ret := false + if util.Exists(incPath) { + if incB, err := ioutil.ReadFile(incPath);err == nil { + var ins InsInc + if err = json.Unmarshal(incB, &ins); err == nil { + //澶勭悊鎺堟潈淇℃伅 + + //2.瑙f瀽sdk.def,灏唖o鍜寊conf澶嶅埗鍒�/opt/vasystem/libs/鏂囦欢澶逛笅 + defPath := unPackPath + "sdk.def" + if util.Exists(defPath) { + if defB, err := ioutil.ReadFile(defPath); err == nil { + //3.灏嗙畻娉晄o銆佷緷璧栨枃浠躲�亃conf銆� + soM := make(map[string]SdkDef) + var skDefArr []SdkDef + if err = json.Unmarshal(defB, &skDefArr);err == nil { + var sdkApi dbapi.SdkApi + for _,skd := range skDefArr { + if _,ok := soM[skd.Def.SdkType];!ok { + soM[skd.Def.SdkType] = skd + } + srv := SdkRegisterVo{ + Id: skd.Def.Id, + SdkType: skd.Def.SdkType, + SdkName: skd.Def.SdkName, + Icon: skd.Def.Icon, + Url: skd.Def.Url, + } + for _,ag := range skd.Args { + sra := SdkRegisterArgVo{ + Scope: ag.Scope, + } + sra.SdkArg = SdkArg{ + Alias: ag.Alias, + Name: ag.Name, + Type: ag.Type, + ArgType: ag.ArgType, + Must: ag.Must, + Unit: ag.Unit, + Range: ag.Range, + DefaultValue: ag.DefaultValue, + DefaultOperator: ag.DefaultOperator, + Sort: ag.Sort, + } + srv.Args = append(srv.Args, sra) + } + paramBody := util.Struct2Map(srv) + sdkApi.Register(paramBody) //灏嗙畻娉曟敞鍐屽埌鏁版嵁搴撲腑 + } + + zconfPath := "/opt/vasystem/bin/zconf/" + libPath := "/opt/vasystem/libs/" + if !util.DirExists(zconfPath) { + os.MkdirAll(zconfPath, 0777) + } + if !util.DirExists(libPath) { + os.MkdirAll(libPath, 0777) + } + for sdkType,_ := range soM { + //澶嶅埗json鍚姩鏂囦欢 + util.CopyFile(unPackPath+sdkType+"/"+sdkType+".json", zconfPath) + util.CopyDir(unPackPath+sdkType+"/"+sdkType, libPath) + if util.DirExists(unPackPath+sdkType+"models") { + util.CopyDir(unPackPath+sdkType+"models", "/opt/vasystem/bin/") + } + } + ret = true + } + } + } + } + } + } + + return ret +} + + +type InsInc struct { + OrderId string `json:"orderId"` + ProductId string `json:"productId"` + ActivateCode string `json:"activateCode"` + MachineCode string `json:"machineCode"` + UserId string `json:"userId"` + SdkIds []string `json:"sdkIds"` + ModuleIds []string `json:"moduleIds"` + ChCount int `json:"chCount"` + AuthCount int `json:"authCount"` + ServeYear int `json:"serveYear"` +} + +//绠楁硶鍜屽弬鏁板畾涔� +type SdkDef struct { + Def Sdk `json:"def"` + Args []SdkArgEntity `json:"args"` +} + +type Sdk struct { + Id string `gorm:"column:id;primary_key;type:varchar(50);unique;not null;" json:"id"` + IpcId string `gorm:"column:ipc_id" json:"ipc_id"` + SdkType string `gorm:"column:sdk_type" json:"sdk_type"` + SdkName string `gorm:"column:sdk_name" json:"sdk_name" example:"浜鸿劯妫�娴�"` + Icon string `gorm:"column:icon" json:"icon,omitempty"` + Url string `gorm:"column:url" json:"url,omitempty" example:"http://ip:port/govideo/sdk/1"` + CreateTime string `gorm:"column:create_time" json:"create_time"` + CreateBy string `gorm:"column:create_by" json:"create_by"` + UpdateTime string `gorm:"column:update_time" json:"update_time"` + Enable bool `gorm:"column:enable;default:1" json:"enable"` + DelFlag int `gorm:"column:del_flag;default:0" json:"del_flag"` + Env string `gorm:"column:env" json:"env"` //杩愯鐜鍙婄粨鏋滆鏄庯紝json鏍煎紡锛屽寘鍚玸o_file_path,runtime,param,depends(cuda鐗堟湰锛宱pencv鐗堟湰锛宼ensorflow鐗堟湰绛�) +} + +type SdkArgEntity struct { + Id string `gorm:"primary_key;column:id" json:"id"` + SdkId string `gorm:"column:sdk_id" json:"sdk_id"` + Scope string `gorm:"column:scope" json:"scope"` + SdkArg +} + +type SdkArg struct { + Alias string `gorm:"column:alias" json:"alias"` //鍙傛暟鐨勫埆鍚� + Name string `gorm:"column:name" json:"name"` //鍙傛暟鍚嶇О + Type string `gorm:"column:type" json:"type"` //鍙傛暟绫诲瀷(鏁存暟锛屽瓧绗︿覆鎴栨暟缁�) + ArgType string `gorm:"column:arg_type" json:"arg_type"` + Must bool `gorm:"column:must" json:"must"` //鏄惁蹇呭~ + Unit string `gorm:"column:unit" json:"unit"` //鍗曚綅 + Range string `gorm:"column:range" json:"range"` //鍊肩殑鑼冨洿锛宔g锛�0,100琛ㄧず浠�0鍒�100 + DefaultValue string `gorm:"column:default_value" json:"default_value"` + DefaultOperator string `gorm:"column:default_operator" json:"default_operator"` + Sort int `gorm:"column:sort;default:1" json:"sort"` +} + + +type SdkRegisterVo struct { + Id string `json:"id"` + SdkType string `json:"sdk_type"`//浜鸿劯妫�娴嬶細FaceDetect,浜鸿劯鎻愬彇锛欶aceExtract,浜鸿劯姣斿锛欶aceCompare,琛屼负锛歒olo + SdkName string `json:"sdk_name"` //绠楁硶鍚嶇О + Args []SdkRegisterArgVo `json:"args"` //绠楁硶鍙傛暟 + Icon string `json:"icon"` //绠楁硶鍥炬爣 + Url string `json:"url"` //绠楁硶涓嬭浇鍦板潃 +} + +type SdkRegisterArgVo struct { + Scope string `json:"scope"` + SdkArg + + Dics []SdkArgDic `json:"dics"` //濡傛灉姝ょ畻娉曞弬鏁版槸琚�夐」锛岄渶瑕佸皢鍙傛暟鏋氫妇鍊煎啓鍏ュ埌瀛楀吀琛ㄤ腑 +} + +type SdkArgDic struct { + Value string `json:"value"` + Name string `json:"name"` + Sort int `json:"sort"` +} \ No newline at end of file -- Gitblit v1.8.0