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

package task

import (
	"fmt"

	"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 StorageCreatePackage struct {
	*exectsk.StorageCreatePackage
}

func NewStorageCreatePackage(info *exectsk.StorageCreatePackage) *StorageCreatePackage {
	return &StorageCreatePackage{
		StorageCreatePackage: info,
	}
}

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

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

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

	resp, err := stgCli.StorageCreatePackage(cdsapi.StorageCreatePackageReq{
		UserID:    t.UserID,
		StorageID: t.StorageID,
		Path:      t.Path,
		BucketID:  t.BucketID,
		Name:      t.Name,
	})
	if err != nil {
		return err
	}

	// TODO 根据接口result返回情况修改
	task.SendStatus(exectsk.NewStorageCreatePackageStatus("completed", "", resp.Package.PackageID))
	return nil
}

func init() {
	Register(NewStorageCreatePackage)
}