fix ret

liuxiaolong

2019-12-23 3c1c1467ebc4356a9ad5441d52ab62a431cb3df5

 sdkApi.go

@@ -1 +1,159 @@
package dbapi

import (
   "basic.com/pubsub/protomsg.git"
   "encoding/json"
   "fmt"
   "strconv"
)

type SdkApi struct{
   Ip string
   Port int
}

func (api SdkApi) getBasicUrl() string {
   if api.Ip == "" {
      return BASIC_URL
   }
   if api.Ip == "" {
      api.Ip = DEFAULT_IP
   }
   if api.Port == 0 {
      api.Port = DEFAULT_PORT
   }
   return "http://"+api.Ip+":"+strconv.Itoa(api.Port)
}

//保存算法，新增(算法id为空)，更新（算法id不为空）
func (api SdkApi)Save(paramBody map[string]interface{}) (bool,interface{}) {
   flag := false
   url := api.getBasicUrl() + DATA_URL_PREFIX + "/sdk/saveSdk"
   client := NewClient()
   respBody, err := client.DoPostRequest(url, CONTENT_TYPE_JSON, paramBody, nil, nil)
   if err != nil {
      return flag,nil
   }
   var res Result
   if err = json.Unmarshal(respBody, &res); err != nil {
      fmt.Println(err)
      return flag,nil
   }
   return res.Success,res.Data
}

//查找所有算法
func (api SdkApi) FindAll(sdkName string)[]protomsg.Sdk{
   var sdkArr []protomsg.Sdk
   url := api.getBasicUrl() + DATA_URL_PREFIX + "/sdk/findAllSdk"
   client := NewClient()
   paramMap := make(map[string]string, 0)
   paramMap["sdkName"] = sdkName
   respBody, err := client.DoGetRequest(url, paramMap, nil)
   if err !=nil {
      return sdkArr
   }
   var res Result
   if err = json.Unmarshal(respBody, &res); err != nil {
      fmt.Println(err)
      return sdkArr
   }
   bytes, _ := json.Marshal(res.Data)
   err = json.Unmarshal(bytes, &sdkArr)
   return sdkArr
}

func (api SdkApi) FindAllSdkRun()(sdkRuns []protomsg.SdkRun){
   sdkArr := api.FindAll("")
   if len(sdkArr) == 0 {
      return nil
   }
   idMap :=make(map[string]string,0)
   for _,sdk := range sdkArr{
      if sdk.IpcId !=""{
         if _,ok :=idMap[sdk.IpcId];!ok{
            idMap[sdk.IpcId] = sdk.IpcId
            sdkRuns = append(sdkRuns,protomsg.SdkRun{
               IpcId:sdk.IpcId,
               SdkType:sdk.SdkType,
            })
         }
      }
   }
   return sdkRuns
}

//查找算法id的slice
func (api SdkApi) GetAllSdkIds() (ipcIds []string){
   sdkArr := api.FindAll("")
   if len(sdkArr) == 0 {
      return nil
   }
   idMap :=make(map[string]string,0)
   for _,sdk := range sdkArr{
      if sdk.IpcId !=""{
         if _,ok :=idMap[sdk.IpcId];!ok{
            idMap[sdk.IpcId] = sdk.IpcId
            ipcIds = append(ipcIds,sdk.IpcId)
         }
      }
   }
   return ipcIds
}

//通过id获取算法信息
func (api SdkApi) GetById(id string) (bool,interface{}){
   url := api.getBasicUrl() + DATA_URL_PREFIX + "/sdk/getById"
   client := NewClient()
   paramMap := make(map[string]string)
   paramMap["id"] = id
   respBody, err := client.DoGetRequest(url, paramMap, nil)
   if err !=nil {
      return false,nil
   }
   var res Result
   if err = json.Unmarshal(respBody, &res); err != nil {
      fmt.Println(err)
      return false,nil
   }
   return res.Success,res.Data
}

func (api SdkApi) FindByTaskId(taskId string) (flag bool,sdks []protomsg.Sdk) {
   url := api.getBasicUrl() + DATA_URL_PREFIX + "/sdk/findByTaskId"
   client := NewClient()
   paramMap := make(map[string]string)
   paramMap["taskId"] = taskId
   respBody, err := client.DoGetRequest(url, paramMap, nil)
   if err !=nil {
      return false,nil
   }
   var res Result
   if err = json.Unmarshal(respBody, &res); err != nil {
      fmt.Println(err)
      return false,nil
   }
   bytes, _ := json.Marshal(res.Data)
   err = json.Unmarshal(bytes, &sdks)
   return true,sdks
}

func (api SdkApi) GetSdkArgs(sdkId string, scope string) (b bool,args []protomsg.SdkArg) {
   url := api.getBasicUrl() + DATA_URL_PREFIX + "/sdkArg/getSdkArgs"
   client := NewClient()
   paramMap := make(map[string]string)
   paramMap["sdkId"] = sdkId
   paramMap["scope"] = scope
   respBody, err := client.DoGetRequest(url, paramMap, nil)
   if err !=nil {
      return false,nil
   }
   var res Result
   if err = json.Unmarshal(respBody, &res); err != nil {
      fmt.Println(err)
      return false,nil
   }
   bytes, _ := json.Marshal(res.Data)
   err = json.Unmarshal(bytes, &args)
   return true,args
}