annzee
/
PCM
forked from JCCE/PCM
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

删除训练任务接口提交

This commit is contained in:
lee 2023-02-28 21:03:13 +08:00
parent c05a62836a
commit d903110666
2 changed files with 56 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

44
adaptor/AIComputing/modelarts/rpc/internal/logic/deletetrainingjoblogic.go Normal file
View File

@ -0,0 +1,44 @@
package logic
import (
"PCM/adaptor/AIComputing/modelarts/rpc/internal/common"
"PCM/common/tool"
"context"
"k8s.io/apimachinery/pkg/util/json"
"strings"
"PCM/adaptor/AIComputing/modelarts/rpc/internal/svc"
"PCM/adaptor/AIComputing/modelarts/rpc/modelarts"
"github.com/zeromicro/go-zero/core/logx"
)
type DeleteTrainingJobLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewDeleteTrainingJobLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteTrainingJobLogic {
return &DeleteTrainingJobLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
// DeleteTrainingJobConfig 删除训练作业
func (l *DeleteTrainingJobLogic) DeleteTrainingJob(in *modelarts.DeleteTrainingJobReq) (*modelarts.DeleteTrainingJobResp, error) {
var resp modelarts.DeleteTrainingJobResp
url := "https://modelarts.cn-north-4.myhuaweicloud.com/v2/" + in.ProjectId + "/training-jobs/" + in.TrainingJobId
token := common.GetToken()
body, err := tool.HttpClient(tool.DELETE, url, strings.NewReader(``), token)
if err != nil {
return nil, err
}
json.Unmarshal(body, &resp)
if &resp == nil {
return nil, err
}
return &resp, nil
}

15
adaptor/AIComputing/modelarts/rpc/pb/modelarts.proto
View File

@ -529,9 +529,19 @@ message CreateTrainingJobConfigResp{
string error_message =2;
string error_code =3;
}
/******************CreateTrainingJobConfig end*************************/
/******************DeleteTrainingJob start*************************/
message DeleteTrainingJobReq{
string project_id =1;
string training_job_id =2;
}
message DeleteTrainingJobResp{
}
/******************DeleteTrainingJob end*************************/
/******************DeleteTrainingJobConfig start*************************/
message DeleteTrainingJobConfigReq{
string project_id =1;
@ -1181,12 +1191,11 @@ service ModelArts {
// CreateTrainingJob 创建训练作业
rpc CreateTrainingJob(CreateTrainingJobReq) returns (CreateTrainingJobResp);
// DeleteTrainingJobConfig 删除训练作业
// rpc DeleteTrainingJob(DeleteTrainingJobReq) returns (DeleteTrainingJobResp);
rpc DeleteTrainingJob(DeleteTrainingJobReq) returns (DeleteTrainingJobResp);
// CreateTrainingJobConfig 创建训练作业参数
rpc CreateTrainingJobConfig(CreateTrainingJobConfigReq) returns (CreateTrainingJobConfigResp);
// DeleteTrainingJobConfig 删除训练作业参数
rpc DeleteTrainingJobConfig(DeleteTrainingJobConfigReq) returns (DeleteTrainingJobConfigResp);
// ListTrainingJobConfig 查询训练作业参数
rpc ListTrainingJobConfig(ListTrainingJobConfigReq) returns (ListTrainingJobConfigResp);