panel/internal/data/task.go

package data

import (
	"github.com/TheTNB/panel/internal/biz"
	"github.com/TheTNB/panel/internal/job"
	"github.com/TheTNB/panel/internal/panel"
)

type taskRepo struct{}

func NewTaskRepo() biz.TaskRepo {
	return &taskRepo{}
}

func (r *taskRepo) HasRunningTask() bool {
	var count int64
	panel.Orm.Model(&biz.Task{}).Where("status = ?", biz.TaskStatusRunning).Or("status = ?", biz.TaskStatusWaiting).Count(&count)
	return count > 0
}

func (r *taskRepo) List(page, limit uint) ([]*biz.Task, int64, error) {
	var tasks []*biz.Task
	var total int64
	err := panel.Orm.Model(&biz.Task{}).Order("id desc").Count(&total).Offset(int((page - 1) * limit)).Limit(int(limit)).Find(&tasks).Error
	return tasks, total, err
}

func (r *taskRepo) Get(id uint) (*biz.Task, error) {
	task := new(biz.Task)
	err := panel.Orm.Model(&biz.Task{}).Where("id = ?", id).First(task).Error
	return task, err
}

func (r *taskRepo) Delete(id uint) error {
	return panel.Orm.Model(&biz.Task{}).Where("id = ?", id).Delete(&biz.Task{}).Error
}

func (r *taskRepo) UpdateStatus(id uint, status biz.TaskStatus) error {
	return panel.Orm.Model(&biz.Task{}).Where("id = ?", id).Update("status", status).Error
}

func (r *taskRepo) Push(task *biz.Task) error {
	if err := panel.Orm.Create(task).Error; err != nil {
		return err
	}
	return panel.Queue.Push(job.NewProcessTask(r), []any{
		task.ID,
	})
}