add get buckets contrl

sunty

2020-08-20 9d88c7c467f8d93af4aab9ba0b6d6c01c2ffc546

 service/SdkDownLoad.go

@@ -1,15 +1,18 @@
package service

import (
   "basic.com/valib/licence.git"
   "bytes"
   "crypto/md5"
   "encoding/hex"
   "encoding/json"
   "errors"
   "fmt"
   "io"
   "io/ioutil"
   "net/http"
   "os"
   "path"
   "path/filepath"
   "strings"
   "webserver/extend/config"
@@ -130,61 +133,83 @@
   }
}

func GetSdkList(sdkName string) []SdkInsOrUpgrade {
//installedFlag 如果是true,表示只查本地的，不需要比对查看未安装的算法
func GetSdkList(sdkName string, installedFlag bool) []SdkInsOrUpgrade {
   var api dbapi.SdkApi
   localSdks := api.FindAll(sdkName) //本地已安装所有算法
   localSdks := api.FindAll(sdkName) //本地已安装所有算法,带顺序
   if localSdks == nil {
      return []SdkInsOrUpgrade{}
   }
   installedSdks := make([]SdkInsOrUpgrade, len(localSdks))

   localSdkM := make(map[string]SdkInsOrUpgrade)
   for _,ls :=range localSdks {
   for idx,ls :=range localSdks {
      siou := SdkInsOrUpgrade{
         Installed: true,
      }
      siou.Sdk = Sdk{
         Id: ls.Id,
         IpcId: ls.IpcId,
         SdkType: ls.SdkType,
         SdkName: ls.SdkName,
         Icon: ls.Icon,
         Url: ls.Url,
      sdk := Sdk{
         CreateTime: ls.CreateTime,
         CreateBy: ls.CreateBy,
         UpdateTime: ls.UpdateTime,
         Enable: ls.Enable,
         DelFlag: int(ls.DelFlag),
         IconBlob: ls.IconBlob,
         Version: ls.Version,
      }
      var sc SdkCommon
      sc.copyFromProto(ls)
      sdk.SdkCommon = sc

      for _,arg := range ls.Args {
         sdk.Args = append(sdk.Args, SdkArg{
            Alias: arg.Alias,
            Name: arg.Name,
            Type: arg.Type,
            Must: arg.Must,
            Unit: arg.Unit,
            Range: arg.Range,
            DefaultValue: arg.DefaultValue,
            DefaultOperator: arg.DefaultOperator,
            Sort: int(arg.Sort),
         })
      }
      siou.Sdk = sdk
      localSdkM[siou.Id] = siou
      installedSdks[idx] = siou
   }

   //远端可用的算法
   remoteSdks := findAllMySdk()
   logger.Debug("remoteSdks:", remoteSdks)
   nInsSdks := make([]SdkInsOrUpgrade,0)
   if !installedFlag {
      //远端可用的算法
      remoteSdks := findAllMySdk()
      //logger.Debug("remoteSdks:", remoteSdks)

   for sdkId,sdk := range remoteSdks {
      if v,ok := localSdkM[sdkId];ok { //本地已安装
         ls := v
         if shouldVersionBeUpgrade(ls.Version, sdk.Version) { //判断版本号是否需要升级
            ls.Installed = true
            ls.IsUpgrade = true //需要升级
            ls.RemoteVersion = sdk.Version //远端的版本号
      for _,sdk := range remoteSdks {
         if v,ok := localSdkM[sdk.Id];ok { //本地已安装
            ls := v
            if shouldVersionBeUpgrade(ls.Version, sdk.Version) { //判断版本号是否需要升级
               ls.Installed = true
               ls.IsUpgrade = true //需要升级
               ls.RemoteVersion = sdk.Version //远端的版本号
            }
            localSdkM[sdk.Id] = ls
            for index,es := range installedSdks {
               if es.Id == sdk.Id {
                  installedSdks[index] = ls
                  break
               }
            }
         } else { //本地未安装，需要安装
            bIns := SdkInsOrUpgrade{
               RemoteVersion: sdk.Version,
               IsUpgrade: false,
               Installed: false,
            }
            bIns.Sdk = sdk
            nInsSdks = append(nInsSdks, bIns)
         }
         localSdkM[sdkId] = ls
      } else { //本地未安装，需要安装
         bIns := SdkInsOrUpgrade{
            RemoteVersion: sdk.Version,
            IsUpgrade: false,
            Installed: false,
         }
         bIns.Sdk = sdk
         localSdkM[sdkId] = bIns
      }
   }

   sdks := make([]SdkInsOrUpgrade, 0)
   for _,v := range localSdkM {
      sdks = append(sdks, v)
   }
   return sdks
   return append(installedSdks, nInsSdks...)
}

func queryDatabase(sdkName string) ([]map[string]interface{}, map[string]bool) {
@@ -215,35 +240,39 @@
   Sdks []Sdk `json:"sdks"`
}

func findAllMySdk() map[string]Sdk {
   token := "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE1OTEzODA3ODksInVzZXIiOiJ7XCJpZFwiOlwiZTZjY2QzNmQtNGYxNi00NmZjLTg4ZDUtMDczNjU4NjZkMjA1XCIsXCJwZXJtaXNzaW9uc1wiOltcIm9yZGVyTWFuZ2U6ZG93bmxvYWRcIixcImRldmljZU1hbmFnZTp1bmJpbmRcIixcImNvZGVNYW5nZVwiLFwidmlwTWFuYWdlOmFkZFwiLFwiYWRtaW5NYW5hZ2U6ZWRpdFwiLFwiY29kZU1hbmdlOnZpZXdcIixcImNvZGVNYW5nZTphZGRcIixcImFkbWluTWFuYWdlXCIsXCJvcmRlck1hbmdlOmNhbmNlbFwiLFwicHJvZHVjdENlbnRlcjpidXlcIixcInByb2R1Y3RDZW50ZXJcIixcInByb2R1Y3RNYW5nZTp2aWV3XCIsXCJ2aXBNYW5hZ2U6dmlld1wiLFwib3JkZXJNYW5nZVwiLFwicHJvZHVjdENlbnRlcjp2aWV3XCIsXCJkZXZpY2VNYW5hZ2VcIixcImFkbWluTWFuYWdlOmFkZFwiLFwicHJvZHVjdE1hbmdlOmFkZFwiLFwiYWRtaW5NYW5hZ2U6dmlld1wiLFwicHJvZHVjdE1hbmdlXCIsXCJvcmRlck1hbmdlOnZpZXdcIixcImRldmljZU1hbmFnZTp2aWV3XCIsXCJvcmRlck1hbmdlOnBheVwiLFwidmlwTWFuYWdlOmVkaXRcIixcInZpcE1hbmFnZVwiLFwicHJvZHVjdE1hbmdlOmVkaXRcIixcInByb2R1Y3RNYW5nZTpwdWJsaXNoXCIsXCJkZXZpY2VNYW5hZ2U6YWRkXCIsXCJhcGlcIixcImhvbWVcIixcInByb2R1Y3RNYW5nZTpvZmZTYWxlXCIsXCJwcm9kdWN0Q2VudGVyOmRvd25sb2FkXCJdLFwidXNlcm5hbWVcIjpcImJhc2ljXCJ9In0.HwRobdFLtMK7ni5OKk4_NAyqpKGuUlUbqF3HBJMJuOk"
   url := "http://192.168.20.10:7004/data/api-s/sdk/findAllMySdk"
   fmt.Println("token:", token, "url:", url)
const (
   token = "Bearer eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjQ3NDUwMjU5MjMsInVzZXIiOiJ7XCJpZFwiOlwiZTZjY2QzNmQtNGYxNi00NmZjLTg4ZDUtMDczNjU4NjZkMjA1XCIsXCJwZXJtaXNzaW9uc1wiOltcInByb2R1Y3RNYW5nZTpwdWJsaXNoXCIsXCJjb2RlTWFuZ2U6dmlld1wiLFwiZGV2aWNlTWFuYWdlOmFkZFwiLFwiYWRtaW5NYW5hZ2VcIixcIm9yZGVyTWFuZ2VcIixcImRldmljZU1hbmFnZTp2aWV3XCIsXCJwcm9kdWN0TWFuZ2U6YWRkXCIsXCJhZG1pbk1hbmFnZTp2aWV3XCIsXCJjb2RlTWFuZ2U6YWRkXCIsXCJwcm9kdWN0TWFuZ2U6b2ZmU2FsZVwiLFwib3JkZXJNYW5nZTpjYW5jZWxcIixcInByb2R1Y3RDZW50ZXI6ZG93bmxvYWRcIixcInByb2R1Y3RDZW50ZXI6YnV5XCIsXCJwcm9kdWN0TWFuZ2U6dmlld1wiLFwiYXBpXCIsXCJob21lXCIsXCJvcmRlck1hbmdlOnBheVwiLFwiYWRtaW5NYW5hZ2U6YWRkXCIsXCJvcmRlck1hbmdlOmRvd25sb2FkXCIsXCJwcm9kdWN0Q2VudGVyXCIsXCJkZXZpY2VNYW5hZ2U6dW5iaW5kXCIsXCJvcmRlck1hbmdlOnZpZXdcIixcImFkbWluTWFuYWdlOmVkaXRcIixcImRldmljZU1hbmFnZVwiLFwidmlwTWFuYWdlOmFkZFwiLFwidmlwTWFuYWdlOnZpZXdcIixcInByb2R1Y3RDZW50ZXI6dmlld1wiLFwidmlwTWFuYWdlOmVkaXRcIixcInZpcE1hbmFnZVwiLFwicHJvZHVjdE1hbmdlOmVkaXRcIixcImNvZGVNYW5nZVwiLFwicHJvZHVjdE1hbmdlXCJdLFwidXNlcm5hbWVcIjpcImJhc2ljXCJ9In0.vwjAFkWuEyadRLvIOGK8LFE3MjpY3SQ7j6AlTXnQDG8"
)

func findAllMySdk() []Sdk {
   url := "http://"+config.ShopConf.Url+"/data/api-s/sdk/findAllMySdk"
   machineCode := licence.GetMachineCode()
   fmt.Println("url:", url, "machineCode:", machineCode)
   paramBody := map[string]interface{} {
      "serverId": config.Server.AnalyServerId,
      "machineCode": "",
      "machineCode": machineCode,
   }
   header := map[string]string {
      "Authorization": token,
   }
   respBody, err := reqUtil.DoPostRequest(url, reqUtil.CONTENT_TYPE_JSON, paramBody, nil, header)
   if err != nil {
      logger.Debug("DoPostRequest err:", err)
      return nil
   }
   var res dbapi.Result
   if err = json.Unmarshal(respBody, &res); err != nil {
      logger.Debug("unmarshal err:", err)
      return nil
   }
   bytes, _ := json.Marshal(res.Data)
   var ss shopSdks
   if err := json.Unmarshal(bytes, &ss);err != nil {
      logger.Debug("unmarshal err:", err)
      return nil
   }
   m := make(map[string]Sdk)
   for _,s := range ss.Sdks {
      m[s.Id] = s
   }
   return m

   return ss.Sdks
}

func GetLocalSdks() []map[string]interface{} {
@@ -291,3 +320,79 @@

   return nil
}

type downOrUpResp struct {
   Url string `json:"url"`
   Md5 string `json:"md5"`
}

//下载或者升级算法
func DownloadOrUpgrade(sdkId string) (bool,error) {
   url := "http://"+config.ShopConf.Url+"/data/api-s/sdk/downloadOrUpgrade"
   machineCode := licence.GetMachineCode()
   if machineCode == "" {
      logger.Debug("获取机器码失败")
      return false, errors.New("获取机器码失败")
   }
    paramBody := map[string]interface{}{
       "sdkId": sdkId,
       "machineCode": machineCode,
   }
   header := map[string]string {
      "Authorization": token,
   }
   respBody, err := reqUtil.DoPostRequest(url, reqUtil.CONTENT_TYPE_JSON, paramBody, nil, header)
   if err != nil {
      logger.Debug("DoPostRequest err:", err)
      return false, err
   }
   var res dbapi.Result
   if err = json.Unmarshal(respBody, &res); err != nil {
      logger.Debug("unmarshal err:", err)
      return false, err
   }
   if !res.Success {
      logger.Debug("res.Data:", res.Data)
      return false, errors.New("请求商城失败")
   }
   bytes, _ := json.Marshal(res.Data)
   var resp downOrUpResp
   if err := json.Unmarshal(bytes, &resp);err != nil {
      logger.Debug("unmarshal err:", err)
      return false, err
   }
   logger.Debug("res.Data:", res.Data, "resp:", resp)
   if resp.Url == "" || resp.Md5 == "" {
      return false, errors.New("获取下载安装包失败")
   }

   //2.下载安装（升级）包，验证md5后安装
   configPatchPath := ""
   if config.Server.PatchPath != "" {
      configPatchPath = config.Server.PatchPath
   } else {
      configPatchPath = "../patch"
   }
   if !reqUtil.DirExists(configPatchPath) {
      os.Mkdir(configPatchPath, 0777)
   }
   filenameWithSuffix := path.Base(resp.Url)
   ext := path.Ext(filenameWithSuffix)
   downUrl := "http://"+config.ShopConf.Url+"/files/"+resp.Url

   gzFilePath := configPatchPath + "/" + resp.Md5+ext
   err = reqUtil.DownloadFile(gzFilePath, downUrl)
   if err != nil {
      logger.Debug("DownloadFile err:", err)
      return false, err
   }
   if unPackB,unPackErr := unPackPatchPackage(resp.Md5, ext); !unPackB { //解压失败
      return false, unPackErr
   }
   b, err := installSdk(resp.Md5, ext)
   if b {
      return true, nil
   }

   return false, err
}