panel/app/jobs/process_task.go

package jobs

import (
	"github.com/goravel/framework/facades"

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

// 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().Tags("面板", "异步任务").With(map[string]any{
			"args": args,
		}).Infof("参数错误")
		return nil
	}

	var task models.Task
	if err := facades.Orm().Query().Where("id = ?", taskID).Get(&task); err != nil {
		facades.Log().Tags("面板", "异步任务").With(map[string]any{
			"task_id": taskID,
			"error":   err.Error(),
		}).Infof("获取任务失败")
		return nil
	}

	task.Status = models.TaskStatusRunning
	if err := facades.Orm().Query().Save(&task); err != nil {
		facades.Log().Tags("面板", "异步任务").With(map[string]any{
			"task_id": taskID,
			"error":   err.Error(),
		}).Infof("更新任务失败")
		return nil
	}

	facades.Log().Tags("面板", "异步任务").With(map[string]any{
		"task_id": taskID,
	}).Infof("开始执行任务")

	if _, err := shell.Execf(task.Shell); err != nil {
		task.Status = models.TaskStatusFailed
		if err := facades.Orm().Query().Save(&task); err != nil {
			facades.Log().Tags("面板", "异步任务").With(map[string]any{
				"task_id": taskID,
				"error":   err.Error(),
			}).Infof("更新任务失败")
			return nil
		}
		facades.Log().Tags("面板", "异步任务").With(map[string]any{
			"task_id": taskID,
			"error":   err.Error(),
		}).Infof("执行任务失败")
		return nil
	}

	task.Status = models.TaskStatusSuccess
	if err := facades.Orm().Query().Save(&task); err != nil {
		facades.Log().Tags("面板", "异步任务").With(map[string]any{
			"task_id": taskID,
			"error":   err.Error(),
		}).Infof("更新任务失败")
		return nil
	}

	facades.Log().Tags("面板", "异步任务").With(map[string]any{
		"task_id": taskID,
	}).Infof("执行任务成功")
	return nil
}