From 7c811247ecf143e08c576986a884bedadc57dd66 Mon Sep 17 00:00:00 2001 From: liuxiaolong <liuxiaolong@aiotlink.com> Date: 星期五, 05 六月 2020 18:29:41 +0800 Subject: [PATCH] add refresh token to resp --- service/SdkInstall.go | 387 +++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 387 insertions(+), 0 deletions(-) diff --git a/service/SdkInstall.go b/service/SdkInstall.go index 6d43c33..43f29a3 100644 --- a/service/SdkInstall.go +++ b/service/SdkInstall.go @@ -1 +1,388 @@ package service + +import ( + "basic.com/dbapi.git" + "basic.com/valib/logger.git" + "encoding/json" + "errors" + "io/ioutil" + "os" + "path" + "path/filepath" + "strconv" + "strings" + "webserver/extend/config" + "webserver/extend/util" +) + +type SdkInstallService struct { + +} + +//绠楁硶瀹夎鍖呭畨瑁� +func (sv SdkInstallService) SdkInstall(identifier string,filename string) (bool,error) { + configPatchPath := "" + if config.Server.PatchPath != "" { + configPatchPath = config.Server.PatchPath + } else { + configPatchPath = "../patch" + } + + filenameWithSuffix := path.Base(filename) + ext := path.Ext(filenameWithSuffix) + + zipFilePath := configPatchPath + "/"+identifier+ext + unPackTargetPath := configPatchPath+"/"+identifier+"_basic/" + if util.Exists(zipFilePath) { + strMd5, e := util.FileMd5(zipFilePath) + if e !=nil || strMd5 == "" { + go os.Remove(zipFilePath) + if util.DirExists(unPackTargetPath) { + go os.RemoveAll(unPackTargetPath) + } + return false,errors.New("鑾峰彇瀹夎鍖卪d5澶辫触") + } + if strMd5 == identifier { + if _,insE := installSdk(identifier, ext); insE != nil { + return false, insE + } + return true,nil + + } else { + go os.Remove(zipFilePath) + if util.DirExists(unPackTargetPath) { + go os.RemoveAll(unPackTargetPath) + } + 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, error) { + configPatchPath := "" + if config.Server.PatchPath != "" { + configPatchPath = config.Server.PatchPath + } else { + configPatchPath = "../patch" + } + //1.瑙e帇缂╂洿鏂板寘 + unPackTargetPath := configPatchPath+"/"+identifier+"_basic/" + unPackFilePath := configPatchPath+"/"+identifier+ext + defer func() { + if util.Exists(unPackTargetPath) { + os.RemoveAll(unPackTargetPath) + } + if util.Exists(unPackFilePath) { + os.RemoveAll(unPackFilePath) + } + }() + + if util.Exists(unPackTargetPath) { + //姝ょ増鏈凡缁忔洿鏂拌繃 + rmErr := os.RemoveAll(unPackTargetPath) + if rmErr !=nil { + return false,rmErr + } + } + if !util.CreateDirectory(unPackTargetPath) { + return false, errors.New("鍒涘缓鍘嬬缉鏂囦欢澶瑰け璐�") + } + + logger.Debug("unPackFilePath:", unPackFilePath, "unPackPath:", unPackTargetPath) + _,err := util.UnTarGzByCmd(unPackFilePath, unPackTargetPath) + if err !=nil { + logger.Debug("UnPack err:",err,"unPackFile:",unPackFilePath) + return false, err + } + targetFileName := "" + err = filepath.Walk(unPackTargetPath, func(path string, f os.FileInfo, err error) error { + if f == nil { + return err + } + if f.IsDir() { + targetFileName = f.Name() + } + return nil + }) + if err != nil { + return false, err + } + unPackPath := unPackTargetPath + targetFileName +"/" + //瑙e帇瀹屾垚锛岃幏鍙栧畨瑁呭寘涓殑鏂囦欢锛屽紑濮嬪畨瑁� + //1.瑙f瀽瀹夎璇存槑ins.inc + incPath := unPackPath+"ins.inc" + if util.Exists(incPath) { + if incB, err := ioutil.ReadFile(incPath);err == nil { + var ins InsInc + if err = json.Unmarshal(incB, &ins); err == nil { + //澶勭悊鎺堟潈淇℃伅 + curMc := "" + if ins.MachineCode != curMc { + + } + //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 + nInsM := make(map[string]string) + aInsM := make(map[string]string) + if err = json.Unmarshal(defB, &skDefArr);err == nil { + var sdkApi dbapi.SdkApi + for _,skd := range skDefArr { + //娉ㄥ唽绠楁硶淇℃伅鍜岀畻娉曞弬鏁板埌dbserver + //鍏堢湅姝ょ畻娉曟湁娌℃湁娉ㄥ唽锛屽鏋滃凡瀹夎锛屾瘮瀵圭増鏈彿 + bf, curDbSdk := sdkApi.GetById(skd.Def.Id) + if bf { + bytes, _ := json.Marshal(curDbSdk) + var sm map[string]interface{} + unE := json.Unmarshal(bytes, &sm) + if unE != nil { + continue + } + sdkVersion, ok := sm["version"] + if ok { //鏈夌増鏈彿 + vStr := sdkVersion.(string) + if shouldVersionBeUpgrade(vStr, skd.Def.Version) { + nInsM[skd.Def.SdkType] = skd.Def.SdkType + aInsM[skd.Def.Id] = skd.Def.Id + } + } else { + nInsM[skd.Def.SdkType] = skd.Def.SdkType + aInsM[skd.Def.Id] = skd.Def.Id + } + } else { + nInsM[skd.Def.SdkType] = skd.Def.SdkType + aInsM[skd.Def.Id] = skd.Def.Id + } + + + if _,ok := soM[skd.Def.SdkType];!ok { + soM[skd.Def.SdkType] = skd + } + if _,ok := aInsM[skd.Def.Id]; ok { + srv := SdkRegisterVo{ + Id: skd.Def.Id, + SdkType: skd.Def.SdkType, + SdkName: skd.Def.SdkName, + Icon: skd.Def.Icon, + Url: skd.Def.Url, + IconBlob: skd.Def.IconBlob, + Version: skd.Def.Version, + } + 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) + logger.Debug("sdkApi.Register paramBody:", paramBody) + sdkApi.Register(paramBody) //灏嗙畻娉曟敞鍐屽埌鏁版嵁搴撲腑 + } + } + zconfPath := "./zconf/" + libPath := "../libs/" + if !util.DirExists(zconfPath) { + os.MkdirAll(zconfPath, 0777) + } + if !util.DirExists(libPath) { + os.MkdirAll(libPath, 0777) + } + for sdkType,_ := range soM { + if _,iOk := nInsM[sdkType];iOk { + //鍏堣В鍘婼dkType.tar.gz鏂囦欢 + if unTarGzE := util.UnTarGz(unPackPath+sdkType+".tar.gz", unPackPath);unTarGzE == nil { + //澶嶅埗json鍚姩鏂囦欢 + if util.Exists(unPackPath+sdkType+"/zconf/"+sdkType+".json") { + util.CopyFile(unPackPath+sdkType+"/zconf/"+sdkType+".json", zconfPath+sdkType+".json") + } + if util.DirExists(unPackPath+sdkType) { + if _,cE := util.CopyDirByCmd(unPackPath+sdkType, libPath);cE != nil { + return false, cE + } + } + if util.DirExists(unPackPath+sdkType+"/models") { + if _,cE := util.CopyDirByCmd(unPackPath+sdkType+"/models", "./zconf"); cE != nil { + return false, cE + } + } + } else { + logger.Debug("unTarGzE sdkType.tar.gz err:", unTarGzE) + } + } else { + logger.Debug("upper version has been installed") + } + } + + } else { + return false, errors.New("鍙嶅簭鍒楀寲绠楁硶瀹氫箟淇℃伅澶辫触") + } + } else { + return false, errors.New("璇诲彇绠楁硶瀹氫箟淇℃伅澶辫触") + } + } else { + return false, errors.New("绠楁硶瀹氫箟淇℃伅涓㈠け") + } + } else { + return false, errors.New("鍙嶅簭鍒楀寲鎺堟潈淇℃伅澶辫触") + } + } else { + return false, errors.New("璇诲彇鎺堟潈鏂囦欢澶辫触") + } + } else { + return false, errors.New("鎺堟潈鏂囦欢涓嶅瓨鍦�") + } + + return true, nil +} + +func shouldVersionBeUpgrade(curVersion,dstVersion string) bool { + if curVersion == "" { + return true + } + if dstVersion == "" { + return false + } + curIVArr := strings.Split(curVersion, ".") + dstIVArr := strings.Split(dstVersion, ".") + if len(curIVArr) !=3 || len(dstIVArr) != 3{ + return false + } + cH,cEH := strconv.Atoi(curIVArr[0]) + cM,cEM := strconv.Atoi(curIVArr[1]) + cL,cEL := strconv.Atoi(curIVArr[2]) + + dH,dEH := strconv.Atoi(curIVArr[0]) + dM,dEM := strconv.Atoi(curIVArr[1]) + dL,dEL := strconv.Atoi(curIVArr[2]) + if cEH !=nil || cEM != nil || cEL != nil || dEH != nil ||dEM !=nil || dEL !=nil { + return false + } + if cH > dH { + return false + } else if cH == dH { + if cM > dM { + return false + } else if cM == dM { + if cL > dL { + return false + } else if cL == dL { + return false + } else { + return true + } + } else { + return true + } + } else { + return true + } +} + + +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鐗堟湰绛�) + IconBlob string `gorm:"column:iconBlob;type:text;" json:"iconBlob"` //鍥剧墖 + Version string `gorm:"column:version;type:varchar(50);default:'1.0';" json:"version"` //鐗堟湰鍙� +} + +type SdkInsOrUpgrade struct { + Sdk + RemoteVersion string `json:"remoteVersion"` //鍟嗗煄浠撳簱鐗堟湰鍙� + Installed bool `json:"installed"` //鏄惁宸插畨瑁� + IsUpgrade bool `json:"isUpgrade"` //鏄惁闇�瑕佸崌绾� +} + +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"` //绠楁硶涓嬭浇鍦板潃 + IconBlob string `json:"iconBlob"` //鍥剧墖 + Version string `json:"version"` //鐗堟湰鍙� +} + +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