feat: add rollback and virtual list

Author: samuelncui

apis/job_create.go

@@ -16,9 +16,5 @@ func (api *API) JobCreate(ctx context.Context, req *entity.JobCreateRequest) (*e
 		return nil, err
 	}
 
-	if err := api.exe.Start(ctx, job); err != nil {
-		return nil, err
-	}
-
 	return &entity.JobCreateReply{Job: convertJobs(job)[0]}, nil
 }

apis/job_edit_state.go

@@ -0,0 +1,44 @@
+package apis
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/samuelncui/yatm/entity"
+)
+
+func (api *API) JobEditState(ctx context.Context, req *entity.JobEditStateRequest) (*entity.JobEditStateReply, error) {
+	job, err := api.exe.GetJob(ctx, req.Id)
+	if err != nil {
+		return nil, err
+	}
+	if job == nil {
+		return nil, fmt.Errorf("job not found, id= %d", req.Id)
+	}
+
+	if job.Status == entity.JobStatus_PROCESSING {
+		return nil, fmt.Errorf("job status 'PROCESSING' is unexpected")
+	}
+	if req.Status != nil {
+		if *req.Status == entity.JobStatus_PROCESSING {
+			return nil, fmt.Errorf("job target status 'PROCESSING' is unexpected")
+		}
+		job.Status = *req.Status
+	}
+
+	job.State = req.State
+	if _, err := api.exe.SaveJob(ctx, job); err != nil {
+		return nil, fmt.Errorf("save job fail, %w", err)
+	}
+
+	executor, err := api.exe.GetJobExecutor(ctx, job.ID)
+	if err != nil {
+		return nil, fmt.Errorf("get job executor fail, %w", err)
+	}
+
+	if err := executor.Close(ctx); err != nil {
+		return nil, fmt.Errorf("close job executor fail, %w", err)
+	}
+
+	return &entity.JobEditStateReply{}, nil
+}

apis/job_next.go

@@ -6,15 +6,10 @@ import (
 	"github.com/samuelncui/yatm/entity"
 )
 
-func (api *API) JobNext(ctx context.Context, req *entity.JobNextRequest) (*entity.JobNextReply, error) {
-	job, err := api.exe.GetJob(ctx, req.Id)
-	if err != nil {
+func (api *API) JobDispatch(ctx context.Context, req *entity.JobDispatchRequest) (*entity.JobDispatchReply, error) {
+	if err := api.exe.Dispatch(ctx, req.Id, req.Param); err != nil {
 		return nil, err
 	}
 
-	if err := api.exe.Submit(ctx, job, req.Param); err != nil {
-		return nil, err
-	}
-
-	return &entity.JobNextReply{Job: convertJobs(job)[0]}, nil
+	return &entity.JobDispatchReply{}, nil
 }

entity/job.pb.go

@@ -41,10 +41,10 @@ message JobState {
     }
 }
 
-message JobNextParam {
+message JobDispatchParam {
     oneof param {
-        job_archive.JobArchiveNextParam archive = 1;
-        job_restore.JobRestoreNextParam restore = 2;
+        job_archive.JobArchiveDispatchParam archive = 1;
+        job_restore.JobRestoreDispatchParam restore = 2;
     }
 }
 

entity/job.proto

@@ -16,7 +16,7 @@ message JobArchiveParam {
     repeated source.Source sources = 1;
 }
 
-message JobArchiveNextParam {
+message JobArchiveDispatchParam {
     oneof param {
         JobArchiveWaitForTapeParam wait_for_tape = 1;
         JobArchiveCopyingParam copying = 2;

entity/job_archive.pb.go

@@ -16,7 +16,7 @@ message JobRestoreParam {
     repeated int64 file_ids = 1;
 }
 
-message JobRestoreNextParam {
+message JobRestoreDispatchParam {
     oneof param {
         JobRestoreWaitForTapeParam wait_for_tape = 1;
         JobRestoreCopyingParam copying = 2;