From ca4ac988b6f6e0810ec5a310a2a6a9045bba94b2 Mon Sep 17 00:00:00 2001 From: lee <320844511@qq.com> Date: Fri, 3 Mar 2023 12:26:25 +0800 Subject: [PATCH] =?UTF-8?q?=E6=95=B0=E6=8D=AE=E9=9B=86=E5=AF=BC=E5=85=A5?= =?UTF-8?q?=E6=8E=A5=E5=8F=A3=E6=8F=90=E4=BA=A4?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../internal/logic/createtrainingjoblogic.go | 30 +++++++++++ .../api/internal/logic/listimportlogic.go | 53 +++++++++++++++++++ .../logic/listtrainingjobsreqlogic.go | 53 +++++++++++++++++++ 3 files changed, 136 insertions(+) create mode 100644 adaptor/AIComputing/AICore/api/internal/logic/createtrainingjoblogic.go create mode 100644 adaptor/AIComputing/AICore/api/internal/logic/listimportlogic.go create mode 100644 adaptor/AIComputing/AICore/api/internal/logic/listtrainingjobsreqlogic.go diff --git a/adaptor/AIComputing/AICore/api/internal/logic/createtrainingjoblogic.go b/adaptor/AIComputing/AICore/api/internal/logic/createtrainingjoblogic.go new file mode 100644 index 0000000..facb18d --- /dev/null +++ b/adaptor/AIComputing/AICore/api/internal/logic/createtrainingjoblogic.go @@ -0,0 +1,30 @@ +package logic + +import ( + "context" + + "PCM/adaptor/AIComputing/AICore/api/internal/svc" + "PCM/adaptor/AIComputing/AICore/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type CreateTrainingJobLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewCreateTrainingJobLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateTrainingJobLogic { + return &CreateTrainingJobLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *CreateTrainingJobLogic) CreateTrainingJob(req *types.CreateTrainingJobReq) (resp *types.CreateTrainingJobResp, err error) { + // todo: add your logic here and delete this line + + return +} diff --git a/adaptor/AIComputing/AICore/api/internal/logic/listimportlogic.go b/adaptor/AIComputing/AICore/api/internal/logic/listimportlogic.go new file mode 100644 index 0000000..3c51630 --- /dev/null +++ b/adaptor/AIComputing/AICore/api/internal/logic/listimportlogic.go @@ -0,0 +1,53 @@ +package logic + +import ( + "PCM/adaptor/AIComputing/modelarts/rpc/modelarts" + "PCM/common/tool" + "PCM/common/xerr" + "context" + "github.com/jinzhu/copier" + "github.com/pkg/errors" + + "PCM/adaptor/AIComputing/AICore/api/internal/svc" + "PCM/adaptor/AIComputing/AICore/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type ListImportLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewListImportLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ListImportLogic { + return &ListImportLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *ListImportLogic) ListImport(req *types.ListImportTasksReq) (resp *types.ListImportTasksResp, err error) { + modelartsReq := &modelarts.ListImportTasksReq{} + err = copier.CopyWithOption(modelartsReq, req, copier.Option{Converters: tool.Converters}) + listDataSetResp, err := l.svcCtx.ModelArtsRpc.GetImportTaskList(l.ctx, modelartsReq) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("Failed to get db DataSet list"), "Failed to get db DataSet list err : %v ,req:%+v", err, req) + } + resp = &types.ListImportTasksResp{} + for i := 0; i < len(listDataSetResp.ImportTasks); i++ { + historyJob := types.ImportTasks{} + resp.ImportTasks = append(resp.ImportTasks, historyJob) + } + + err = copier.CopyWithOption(&resp, &listDataSetResp, copier.Option{Converters: tool.Converters}) + /*if err != nil { + return nil, err + } + for i := range resp.DataSet { + resp.DataSet[i]. = "modelarts" + }*/ + + return resp, nil +} diff --git a/adaptor/AIComputing/AICore/api/internal/logic/listtrainingjobsreqlogic.go b/adaptor/AIComputing/AICore/api/internal/logic/listtrainingjobsreqlogic.go new file mode 100644 index 0000000..50950ee --- /dev/null +++ b/adaptor/AIComputing/AICore/api/internal/logic/listtrainingjobsreqlogic.go @@ -0,0 +1,53 @@ +package logic + +import ( + "PCM/adaptor/AIComputing/modelarts/rpc/modelarts" + "PCM/common/tool" + "PCM/common/xerr" + "context" + "github.com/jinzhu/copier" + "github.com/pkg/errors" + + "PCM/adaptor/AIComputing/AICore/api/internal/svc" + "PCM/adaptor/AIComputing/AICore/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type ListTrainingJobsreqLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewListTrainingJobsreqLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ListTrainingJobsreqLogic { + return &ListTrainingJobsreqLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *ListTrainingJobsreqLogic) ListTrainingJobsreq(req *types.ListTrainingJobsreq) (resp *types.ListTrainingJobsresp, err error) { + modelartsReq := &modelarts.ListTrainingJobsreq{} + err = copier.CopyWithOption(modelartsReq, req, copier.Option{Converters: tool.Converters}) + listDataSetResp, err := l.svcCtx.ModelArtsRpc.GetListTrainingJobs(l.ctx, modelartsReq) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("Failed to get db DataSet list"), "Failed to get db DataSet list err : %v ,req:%+v", err, req) + } + resp = &types.ListTrainingJobsresp{} + for i := 0; i < len(listDataSetResp.Items); i++ { + historyJob := types.Items{} + resp.Items = append(resp.Items, historyJob) + } + + err = copier.CopyWithOption(&resp, &listDataSetResp, copier.Option{Converters: tool.Converters}) + /*if err != nil { + return nil, err + } + for i := range resp.DataSet { + resp.DataSet[i]. = "modelarts" + }*/ + + return resp, nil +}