feat: add sdk download and regiter

zhangzengfei

2020-01-19 20ed84cad4fd237cca40619d0fe7ed39c5e030cd

 service/SdkDownLoad.go

@@ -4,16 +4,31 @@
   "bytes"
   "crypto/md5"
   "encoding/hex"
   "encoding/json"
   "fmt"
   "io"
   "io/ioutil"
   "net/http"
   "os"
   "path/filepath"
   "strings"
   "basic.com/valib/logger.git"

   "webserver/util"

   "basic.com/dbapi.git"
   "basic.com/valib/logger.git"
)

func DownSdk(id string) map[string]interface{} {
   // 下载算法
   if ok, err := DownSo(id); !ok {
      logger.Error("下载算法失败", err)
      return nil
   }

   return FindLocalSdkSoById(id)
}

// 从服务器上下载文件到临时目录，校验之后如果完整将其拷贝到目标目录下
func DownSo(url string)(bool,error) {
   resp, err := http.Get(url)
@@ -90,6 +105,7 @@
   md5Str := hex.EncodeToString(md5.Sum(nil))
   return md5Str,nil
}

// 从url中截取出文件名，参数是是否带后缀
func GetFileNameFromUrl(url string,withSuffix bool)string {
   fileName := strings.Split(url,"/")[len(strings.Split(url,"/"))-1]
@@ -110,3 +126,99 @@
      return false
   }
}

func GetSdkList(sdkName string) []map[string]interface{} {
   // 待返回的数据
   sdks, idsCache := queryDatabase(sdkName)

   // 查询本地已经存在的算法和远端的全部算法, 然后和已安装的比对, 返回全部可用的算法组. 未安装的算法在前端走下载流程
   localSdks := GetLocalSdks()
   for _, sdk := range localSdks {
      id := sdk["id"].(string)
      if !idsCache[id] {
         idsCache[id] = true

         sdk["enable"] = false
         sdk["installed"] = false

         // 不返回参数信息
         delete(sdk, "args")
         sdks = append(sdks, sdk)
      }

   }

   // 目前还没有远端算法库,
   // remoteSdks := getRemoteSdks()

   return sdks
}

func queryDatabase(sdkName string) ([]map[string]interface{}, map[string]bool) {
   var api dbapi.SdkApi
   ids := make(map[string]bool)
   sdks := []map[string]interface{}{}

   // 查询已经安装的算法
   data := api.FindAll(sdkName)
   for _, sdk := range data {
      ids[sdk.Id] = true
      sdks = append(sdks, map[string]interface{}{
         "id":        sdk.Id,
         "ipc_id":    sdk.IpcId,
         "sdk_type":  sdk.SdkType,
         "sdk_name":  sdk.SdkName,
         "icon":      sdk.Icon,
         "enable":    sdk.Enable,
         "installed": true,
      })
   }

   return sdks, ids
}

func GetLocalSdks() []map[string]interface{} {
   var algos = []map[string]interface{}{}

   AlgorithmFiles := "/opt/vasystem/bin/algorithm/*.json"

   files, err := filepath.Glob(AlgorithmFiles)
   if err != nil {
      fmt.Println("Cannot access algorithm json files: No such file or directory")
      return algos
   }

   for _, filename := range files {
      algo := make(map[string]interface{})
      f, err := ioutil.ReadFile(filename)
      if err != nil {
         return algos
      }

      if err := json.Unmarshal(f, &algo); err != nil {
         return algos
      }

      algos = append(algos, map[string]interface{}{
         "id":       algo["sdkId"],
         "ipc_id":   algo["ipcId"],
         "sdk_type": algo["sdkType"],
         "sdk_name": algo["sdkName"],
         "icon":     algo["icon"],
         "args":     algo["sdkArgs"],
      })
   }

   return algos
}

func FindLocalSdkSoById(id string) map[string]interface{} {
   localAlgos := GetLocalSdks()
   for _, sdk := range localAlgos {
      if sdk["id"].(string) == id {
         return sdk
      }
   }

   return nil
}