fix:删除接口修改
Former-commit-id: 56be61056087aa8d025b1409bfdc12da4f60320b
This commit is contained in:
parent
1efe5406e5
commit
f3719f2ab2
|
@ -6,10 +6,14 @@ package logic
|
||||||
*/
|
*/
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"APIGW-go-sdk/core"
|
||||||
"PCM/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/common"
|
"PCM/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/common"
|
||||||
"PCM/common/tool"
|
"PCM/common/tool"
|
||||||
"context"
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"io/ioutil"
|
||||||
"k8s.io/apimachinery/pkg/util/json"
|
"k8s.io/apimachinery/pkg/util/json"
|
||||||
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"PCM/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/svc"
|
"PCM/adaptor/PCM-AI/PCM-MODELARTS/rpc/internal/svc"
|
||||||
|
@ -35,17 +39,69 @@ func NewDeleteModelLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Delet
|
||||||
func (l *DeleteModelLogic) DeleteModel(in *modelarts.DeleteModelReq) (*modelarts.DeleteModelResp, error) {
|
func (l *DeleteModelLogic) DeleteModel(in *modelarts.DeleteModelReq) (*modelarts.DeleteModelResp, error) {
|
||||||
// todo: add your logic here and delete this line
|
// todo: add your logic here and delete this line
|
||||||
var resp modelarts.DeleteModelResp
|
var resp modelarts.DeleteModelResp
|
||||||
url := "https://modelarts.cn-north-4.myhuaweicloud.com/v1/" + in.ProjectId + "/models/" + in.ModelId
|
//根据智算类型判断走华为智算还是南京智算
|
||||||
|
modelArtsType := in.ModelArtsType
|
||||||
|
if modelArtsType == l.svcCtx.Config.HaweiModelArtsType {
|
||||||
|
modelArtsUrl := l.svcCtx.Config.ModelArtsUrl
|
||||||
|
url := modelArtsUrl + "v1/" + in.ProjectId + "/models/" + in.ModelId
|
||||||
reqByte, err := json.Marshal(in)
|
reqByte, err := json.Marshal(in)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
payload := strings.NewReader(string(reqByte))
|
payload := strings.NewReader(string(reqByte))
|
||||||
token := common.GetToken()
|
token := common.GetToken()
|
||||||
body, err := tool.HttpClient(tool.DELETE, url, payload, token)
|
statusCode, body, err := tool.HttpClientWithBodyAndCode(tool.DELETE, url, payload, token)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
|
if statusCode == 202 {
|
||||||
json.Unmarshal(body, &resp)
|
json.Unmarshal(body, &resp)
|
||||||
|
resp.Code = 200
|
||||||
|
resp.Msg = "Success"
|
||||||
|
} else if statusCode != 202 {
|
||||||
|
json.Unmarshal(body, &resp)
|
||||||
|
resp.Code = 400
|
||||||
|
resp.Msg = "Failure"
|
||||||
|
}
|
||||||
|
|
||||||
|
} else if modelArtsType == l.svcCtx.Config.NanjingModelArtsType {
|
||||||
|
AK := l.svcCtx.Config.AK
|
||||||
|
SK := l.svcCtx.Config.SK
|
||||||
|
NanjingModelArtsUrl := l.svcCtx.Config.NanjingModelArtsUrl
|
||||||
|
XProjectId := l.svcCtx.Config.XProjectId
|
||||||
|
XDomainId := l.svcCtx.Config.XDomainId
|
||||||
|
s := core.Signer{
|
||||||
|
Key: AK,
|
||||||
|
Secret: SK,
|
||||||
|
}
|
||||||
|
r, err := http.NewRequest("DELETE", NanjingModelArtsUrl+in.ProjectId+"/models/"+in.ModelId,
|
||||||
|
nil)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
//return
|
||||||
|
}
|
||||||
|
r.Header.Add("content-type", "application/json;charset=UTF-8")
|
||||||
|
r.Header.Add("X-Project-Id", XProjectId)
|
||||||
|
r.Header.Add("X-Domain-Id", XDomainId)
|
||||||
|
r.Header.Add("x-stage", "RELEASE")
|
||||||
|
s.Sign(r)
|
||||||
|
client := http.DefaultClient
|
||||||
|
res, err := client.Do(r)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Println(err)
|
||||||
|
}
|
||||||
|
defer res.Body.Close()
|
||||||
|
body, err := ioutil.ReadAll(res.Body)
|
||||||
|
statusCode := res.StatusCode
|
||||||
|
if statusCode == 202 {
|
||||||
|
json.Unmarshal(body, &resp)
|
||||||
|
resp.Code = 200
|
||||||
|
resp.Msg = "Success"
|
||||||
|
} else if statusCode != 202 {
|
||||||
|
json.Unmarshal(body, &resp)
|
||||||
|
resp.Code = 400
|
||||||
|
resp.Msg = "Failure"
|
||||||
|
}
|
||||||
|
}
|
||||||
return &resp, nil
|
return &resp, nil
|
||||||
}
|
}
|
||||||
|
|
|
@ -1332,6 +1332,7 @@ message DeleteModelResp{
|
||||||
int32 code = 1; // @gotags: copier:"Code"
|
int32 code = 1; // @gotags: copier:"Code"
|
||||||
repeated string delete_success_list = 2;
|
repeated string delete_success_list = 2;
|
||||||
repeated DeleteModelResponseFailedList delete_failed_list = 3;
|
repeated DeleteModelResponseFailedList delete_failed_list = 3;
|
||||||
|
string msg =4; // @gotags: copier:"Msg"
|
||||||
}
|
}
|
||||||
|
|
||||||
message DeleteModelResponseFailedList{
|
message DeleteModelResponseFailedList{
|
||||||
|
|
Loading…
Reference in New Issue