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

查询数据集导出任务的状态合并分支

This commit is contained in:
Diva123456 2023-03-01 15:30:24 +08:00
parent b489458159
commit 35727486b0
2 changed files with 77 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
adaptor/AIComputing/modelarts/rpc/internal/logic/getexporttaskstatusofdatasetlogic.go Normal file
View File

@ -0,0 +1,47 @@
package logic
/*
desc: "AI core微服务"
author: "xie"
*/
import (
"PCM/adaptor/AIComputing/modelarts/rpc/internal/common"
"PCM/common/tool"
"context"
"k8s.io/apimachinery/pkg/util/json"
"PCM/adaptor/AIComputing/modelarts/rpc/internal/svc"
"PCM/adaptor/AIComputing/modelarts/rpc/modelarts"
"github.com/zeromicro/go-zero/core/logx"
)
type GetExportTaskStatusOfDatasetLogic struct {
ctx context.Context
svcCtx *svc.ServiceContext
logx.Logger
}
func NewGetExportTaskStatusOfDatasetLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetExportTaskStatusOfDatasetLogic {
return &GetExportTaskStatusOfDatasetLogic{
ctx: ctx,
svcCtx: svcCtx,
Logger: logx.WithContext(ctx),
}
}
func (l *GetExportTaskStatusOfDatasetLogic) GetExportTaskStatusOfDataset(in *modelarts.GetExportTaskStatusOfDatasetReq) (*modelarts.GetExportTaskStatusOfDatasetResp, error) {
// todo: add your logic here and delete this line
var resp modelarts.GetExportTaskStatusOfDatasetResp
url := "https://modelarts.cn-north-4.myhuaweicloud.com/v2/" + in.ProjectId + "/datasets/" + in.ResourceId + "/export-tasks/" + in.TaskId
token := common.GetToken()
body, err := tool.HttpClient(tool.GET, url, nil, token)
if err != nil {
return nil, err
}
// var resp Pppp
json.Unmarshal(body, &resp)
return &resp, nil
}

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

@ -728,6 +728,35 @@ message ExportTaskStatus {
}
/******************Get Export Tasks Of Dataset End*************************/
/******************Get Export Task Status Of Dataset Start*************************/
message GetExportTaskStatusOfDatasetReq{
string resource_id = 1;
string project_id = 2;
string task_id = 3;
}
message GetExportTaskStatusOfDatasetResp{
uint32 code = 1; // @gotags: copier:"Code"
uint32 create_time = 2;
string error_code = 3;
string error_msg = 4;
int32 export_format = 5;
ExportParams export_params = 6;
repeated ExportTaskStatus export_tasks = 7;
int32 export_type = 8;
int32 finished_sample_count = 9;
string path = 10;
float progress = 11;
string status = 12;
string task_id = 13;
int64 total_count = 14;
int64 total_sample = 15;
uint32 update_time = 16;
string version_format = 17;
string version_id = 18;
}
/******************Get Export Tasks Of Dataset End*************************/
/******************Create Service Start*************************/
message CreateServiceReq{
string workspace_id = 1;
@ -910,11 +939,9 @@ message ListServices{
map<string,string> additional_properties = 24;
}
/******************List Services End*************************/
/******************find datasetList start*************************/
message datasetReq{
string project_id = 1;
}
@ -1283,7 +1310,7 @@ service ModelArts {
//export task
rpc ExportTask(ExportTaskReq) returns (ExportTaskResp);
rpc GetExportTasksOfDataset(GetExportTasksOfDatasetReq) returns (GetExportTasksOfDatasetResp);
rpc GetExportTaskStatusOfDataset(GetExportTaskStatusOfDatasetReq) returns (GetExportTaskStatusOfDatasetResp);
//processor task
rpc CreateProcessorTask(CreateProcessorTaskReq) returns (CreateProcessorTaskResp);
rpc DescribeProcessorTask(DescribeProcessorTaskReq) returns (DescribeProcessorTaskResp);