Merge pull request '调整提交任务集，上传任务集文件的流程' (#17) from feature_gxh into master

api/scheduler/jobset.go

@@ -16,7 +16,8 @@ type JobSetSumbitReq struct {
 }
 
 type JobSetSumbitResp struct {
-	JobSetID string `json:"jobSetID"`
+	JobSetID          string                         `json:"jobSetID"`
+	FilesUploadScheme models.JobSetFilesUploadScheme `json:"filesUploadScheme"`
 }
 
 func (c *Client) JobSetSumbit(req JobSetSumbitReq) (*JobSetSumbitResp, error) {
@@ -49,14 +50,15 @@ func (c *Client) JobSetSumbit(req JobSetSumbitReq) (*JobSetSumbitResp, error) {
 	return nil, fmt.Errorf("unknow response content type: %s", contType)
 }
 
-type JobSetSetLocalFileReq struct {
+type JobSetLocalFileUploadedReq struct {
 	JobSetID  string `json:"jobSetID"`
 	LocalPath string `json:"localPath"`
+	Error     string `json:"error"`
 	PackageID int64  `json:"packageID"`
 }
 
-func (c *Client) JobSetSetLocalFile(req JobSetSetLocalFileReq) error {
-	url, err := url.JoinPath(c.baseURL, "/jobSet/setLocalFile")
+func (c *Client) JobSetLocalFileUploaded(req JobSetLocalFileUploadedReq) error {
+	url, err := url.JoinPath(c.baseURL, "/jobSet/localFileUploaded")
 	if err != nil {
 		return err
 	}

api/scheduler/scheduler_test.go

@@ -40,7 +40,7 @@ func Test_JobSet(t *testing.T) {
 		So(err, ShouldBeNil)
 		So(id.JobSetID, ShouldNotBeEmpty)
 
-		err = cli.JobSetSetLocalFile(JobSetSetLocalFileReq{
+		err = cli.JobSetLocalFileUploaded(JobSetLocalFileUploadedReq{
 			JobSetID:  id.JobSetID,
 			LocalPath: "code",
 			PackageID: 1,

models/job.go → models/scheduler.go

@@ -116,3 +116,43 @@ func JobSetInfoFromJSON(data []byte) (*JobSetInfo, error) {
 
 	return &ret, nil
 }
+
+const (
+	FileScheduleActionNo   = "No"
+	FileScheduleActionMove = "Move"
+	FileScheduleActionLoad = "Load"
+)
+
+type FileScheduleScheme struct {
+	Action          string `json:"action"`
+	TargetStorageID int64  `json:"targetStorageID"`
+}
+
+const (
+	ImageScheduleActionNo     = "No"
+	ImageScheduleActionImport = "Import"
+)
+
+type ImageScheduleScheme struct {
+	Action string `json:"action"`
+}
+
+type JobScheduleScheme struct {
+	TargetSlwNodeID int64               `json:"targetSlwNodeID"`
+	Dataset         FileScheduleScheme  `json:"dataset"`
+	Code            FileScheduleScheme  `json:"code"`
+	Image           ImageScheduleScheme `json:"image"`
+}
+
+type JobSetPreScheduleScheme struct {
+	JobSchemes map[string]JobScheduleScheme `json:"jobSchemes"` // 任务的预调度方案。Key为LocalJobIDs
+}
+
+type JobSetFilesUploadScheme struct {
+	LocalFileSchemes []LocalFileUploadScheme `json:"localFileUploadSchemes"`
+}
+
+type LocalFileUploadScheme struct {
+	LocalPath         string `json:"localPath"`
+	UploadToStgNodeID *int64 `json:"uploadToStgNodeID"`
+}