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

package task

import (
	"fmt"
	"strconv"
	"time"

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

type StorageLoadPackage struct {
	*exectsk.StorageLoadPackage
}

func NewStorageLoadPackage(info *exectsk.StorageLoadPackage) *StorageLoadPackage {
	return &StorageLoadPackage{
		StorageLoadPackage: info,
	}
}

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

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

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

	path := strconv.FormatInt(int64(t.PackageID), 10) + "_" + strconv.FormatInt(time.Now().Unix(), 10)
	_, err = stgCli.StorageLoadPackage(cdsapi.StorageLoadPackageReq{
		UserID:    t.UserID,
		PackageID: t.PackageID,
		StorageID: t.StorageID,
		RootPath:  path,
	})
	if err != nil {
		return "", err
	}

	return path, nil
}

func init() {
	Register(NewStorageLoadPackage)
}