package dbapi
|
|
import (
|
"basic.com/pubsub/protomsg.git"
|
"encoding/json"
|
"fmt"
|
)
|
|
type SdkApi struct{
|
}
|
|
//保存算法，新增(算法id为空)，更新（算法id不为空）
|
func (api SdkApi)Save(paramBody map[string]interface{}) (bool,interface{}) {
|
flag := false
|
url := BASIC_URL + 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 := BASIC_URL + 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 := BASIC_URL + 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 := BASIC_URL + 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 := BASIC_URL + 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
|
}
|
