panel/app/jobs/process_task.go

package jobs

import (
	"os/exec"

	"github.com/goravel/framework/facades"

	"github.com/TheTNB/panel/app/models"
)

// ProcessTask 处理面板任务
type ProcessTask struct {
}

// Signature The name and signature of the job.
func (receiver *ProcessTask) Signature() string {
	return "process_task"
}

// Handle Execute the job.
func (receiver *ProcessTask) Handle(args ...any) error {
	taskID, ok := args[0].(uint)
	if !ok {
		facades.Log().Info("[面板][ProcessTask] 任务ID参数错误")
		return nil
	}

	var task models.Task
	if err := facades.Orm().Query().Where("id = ?", taskID).Get(&task); err != nil {
		facades.Log().Infof("[面板][ProcessTask] 获取任务%d失败: %s", taskID, err.Error())
		return nil
	}

	task.Status = models.TaskStatusRunning
	if err := facades.Orm().Query().Save(&task); err != nil {
		facades.Log().Infof("[面板][ProcessTask] 更新任务%d失败: %s", taskID, err.Error())
		return nil
	}

	facades.Log().Infof("[面板][ProcessTask] 开始执行任务%d", taskID)
	cmd := exec.Command("bash", "-c", task.Shell)
	err := cmd.Run()
	if err != nil {
		task.Status = models.TaskStatusFailed
		if err := facades.Orm().Query().Save(&task); err != nil {
			facades.Log().Infof("[面板][ProcessTask] 更新任务%d失败: %s", taskID, err.Error())
			return nil
		}
		facades.Log().Infof("[面板][ProcessTask] 任务%d执行失败: %s", taskID, err.Error())
		return nil
	}

	task.Status = models.TaskStatusSuccess
	if err := facades.Orm().Query().Save(&task); err != nil {
		facades.Log().Infof("[面板][ProcessTask] 更新任务%d失败: %s", taskID, err.Error())
		return nil
	}

	facades.Log().Infof("[面板][ProcessTask] 任务%d执行成功", taskID)
	return nil
}