JCC-CSScheduler/executor/internal/task/storage_load_package.go

package task

import (
	"fmt"
	"time"

	"gitlink.org.cn/cloudream/common/api/storage"
	"gitlink.org.cn/cloudream/common/pkgs/logger"
	"gitlink.org.cn/cloudream/common/pkgs/task"
	"gitlink.org.cn/cloudream/scheduler/common/globals"
	exectsk "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq/executor/task"
)

type StorageLoadPackage struct {
	userID    int64
	packageID int64
	storageID int64
}

func NewStorageLoadPackage(userID int64, packageID int64, storageID int64) *StorageLoadPackage {
	return &StorageLoadPackage{
		userID:    userID,
		packageID: packageID,
		storageID: storageID,
	}
}

func (t *StorageLoadPackage) Execute(task *task.Task[TaskContext], ctx TaskContext, complete CompleteFn) {
	log := logger.WithType[StorageLoadPackage]("Task")
	log.Debugf("begin with %w", logger.FormatStruct(t))
	defer log.Debugf("end")

	err := t.do(ctx)
	if err != nil {
		//TODO 若任务失败，上报的状态failed字段根据情况修改
		ctx.reporter.Report(task.ID(), exectsk.NewLoadPackageTaskStatus("failed", err.Error()))
	} else {
		ctx.reporter.Report(task.ID(), exectsk.NewLoadPackageTaskStatus("completed", ""))
	}
	ctx.reporter.ReportNow()

	complete(err, CompleteOption{
		RemovingDelay: time.Minute,
	})
}

func (t *StorageLoadPackage) do(ctx TaskContext) error {
	stgCli, err := globals.CloudreamStoragePool.Acquire()
	if err != nil {
		return fmt.Errorf("new cloudream storage client: %w", err)
	}
	defer stgCli.Close()

	return stgCli.StorageLoadPackage(storage.StorageLoadPackageReq{
		UserID:    t.userID,
		PackageID: t.packageID,
		StorageID: t.storageID,
	})
}