数据集导入接口提交

adaptor/AIComputing/AICore/api/AICore.api

@@ -8,7 +8,6 @@ info(
 )
 /******************find datasetList start*************************/
 
-
 type DataSet {
 	DatasetId     string        `json:"datasetId"`
 	DataFormat    string        `json:"dataFormat"`
@@ -37,10 +36,23 @@ type (
 )
 
 /******************find datasetList end*************************/
+/******************ImportTask start*************************/
+type (
+	ImportTaskDataReq {
+		ProjectId  string `json:"projectId"`
+		DatasetId  string `json:"datasetId"`
+		ImportPath string `json:"importPath"`
+	}
+	ImportTaskDataResp {
+		TaskId int32 `json:"taskId"`
+	}
+)
+/******************ImportTask end*************************/
 
 service AICore-api {
 	
 	@handler listDataSetHandler
 	get /listDataSet (listDataSetReq) returns (listDataSetResp)
-	
+	@handler CreateTaskHandler
+	post /CreateTask (ImportTaskDataReq) returns (ImportTaskDataResp)
 }

adaptor/AIComputing/AICore/api/internal/handler/createtaskhandler.go

@@ -0,0 +1,28 @@
+package handler
+
+import (
+	"net/http"
+
+	"PCM/adaptor/AIComputing/AICore/api/internal/logic"
+	"PCM/adaptor/AIComputing/AICore/api/internal/svc"
+	"PCM/adaptor/AIComputing/AICore/api/internal/types"
+	"github.com/zeromicro/go-zero/rest/httpx"
+)
+
+func CreateTaskHandler(svcCtx *svc.ServiceContext) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		var req types.ImportTaskDataReq
+		if err := httpx.Parse(r, &req); err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+			return
+		}
+
+		l := logic.NewCreateTaskLogic(r.Context(), svcCtx)
+		resp, err := l.CreateTask(&req)
+		if err != nil {
+			httpx.ErrorCtx(r.Context(), w, err)
+		} else {
+			httpx.OkJsonCtx(r.Context(), w, resp)
+		}
+	}
+}

adaptor/AIComputing/AICore/api/internal/logic/createtasklogic.go

@@ -0,0 +1,49 @@
+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 CreateTaskLogic struct {
+	logx.Logger
+	ctx    context.Context
+	svcCtx *svc.ServiceContext
+}
+
+func NewCreateTaskLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateTaskLogic {
+	return &CreateTaskLogic{
+		Logger: logx.WithContext(ctx),
+		ctx:    ctx,
+		svcCtx: svcCtx,
+	}
+}
+
+func (l *CreateTaskLogic) CreateTask(req *types.ImportTaskDataReq) (resp *types.ImportTaskDataResp, err error) {
+	modelartsReq := &modelarts.ImportTaskDataReq{}
+	err = copier.CopyWithOption(modelartsReq, req, copier.Option{Converters: tool.Converters})
+	ImportTaskDataResp, err := l.svcCtx.ModelArtsRpc.CreateTask(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.ImportTaskDataResp{}
+
+	err = copier.CopyWithOption(&resp, &ImportTaskDataResp, copier.Option{Converters: tool.Converters})
+	/*if err != nil {
+		return nil, err
+	}
+	for i := range resp.DataSet {
+		resp.DataSet[i]. = "modelarts"
+	}*/
+
+	return resp, nil
+}