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

package task

import (
	"fmt"
	"gitlink.org.cn/cloudream/common/pkgs/logger"
	pcmsdk "gitlink.org.cn/cloudream/common/sdks/pcm"
	schglb "gitlink.org.cn/cloudream/scheduler/common/globals"
	exectsk "gitlink.org.cn/cloudream/scheduler/common/pkgs/mq/executor/task"
)

type PCMSubmitTask struct {
	*exectsk.SubmitTask
}

func NewPCMSubmitTask(info *exectsk.SubmitTask) *PCMSubmitTask {
	return &PCMSubmitTask{
		SubmitTask: info,
	}
}

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

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

func (t *PCMSubmitTask) do(task *Task, ctx TaskContext) error {
	log := logger.WithType[PCMSubmitTask]("Task")

	pcmCli, err := schglb.PCMPool.Acquire()
	if err != nil {
		return fmt.Errorf("new pcm client: %w", err)
	}
	defer schglb.PCMPool.Release(pcmCli)

	resp, err := pcmCli.SubmitTask(pcmsdk.SubmitTaskReq{
		PartID:     t.PCMParticipantID,
		ImageID:    t.PCMImageID,
		ResourceID: t.PCMResourceID,
		CMD:        t.CMD,
		Envs:       t.Envs,
		Params:     t.Params,
	})

	if err != nil {
		return err
	}

	log.Infof("pcm submit task success, task id: %v", resp.TaskID)

	var prevStatus pcmsdk.TaskStatus
	retryCount := 0
	for {
		tsResp, err := pcmCli.GetTask(pcmsdk.GetTaskReq{
			PartID: t.PCMParticipantID,
			TaskID: resp.TaskID,
		})
		if err != nil {
			retryCount++
			if retryCount > 3 {
				return err
			}
			continue
		}

		if tsResp.TaskStatus != prevStatus {
			task.SendStatus(exectsk.NewSubmitTaskStatus(tsResp.TaskStatus, ""))
		}

		prevStatus = tsResp.TaskStatus

		// 根据返回的result判定任务是否完成，若完成 跳出循环，结束任务
		if tsResp.TaskStatus == "succeeded" || tsResp.TaskStatus == "Completed" {
			return nil
		}

		if tsResp.TaskStatus == pcmsdk.TaskStatusFailed {
			// TODO 返回更详细的信息
			return fmt.Errorf("task failed")
		}
	}
}

func init() {
	Register(NewPCMSubmitTask)
}