mirror of
https://github.com/tnb-labs/panel.git
synced 2026-09-19 10:03:36 +08:00
119 lines
3.0 KiB
Go
119 lines
3.0 KiB
Go
package data
|
|
|
|
import (
|
|
"errors"
|
|
"log/slog"
|
|
"os"
|
|
|
|
"github.com/leonelquinteros/gotext"
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/acepanel/panel/v3/internal/biz"
|
|
"github.com/acepanel/panel/v3/pkg/types"
|
|
)
|
|
|
|
type taskRepo struct {
|
|
t *gotext.Locale
|
|
db *gorm.DB
|
|
log *slog.Logger
|
|
runner types.TaskRunner
|
|
}
|
|
|
|
func NewTaskRepo(db *gorm.DB, t *gotext.Locale, log *slog.Logger, runner types.TaskRunner) biz.TaskRepo {
|
|
return &taskRepo{
|
|
t: t,
|
|
db: db,
|
|
log: log,
|
|
runner: runner,
|
|
}
|
|
}
|
|
|
|
func (r *taskRepo) HasRunningTask() bool {
|
|
var count int64
|
|
r.db.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) {
|
|
tasks := make([]*biz.Task, 0)
|
|
var total int64
|
|
err := r.db.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 := r.db.Model(&biz.Task{}).Where("id = ?", id).First(task).Error
|
|
return task, err
|
|
}
|
|
|
|
func (r *taskRepo) Delete(id uint) error {
|
|
task, err := r.Get(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if task.Status == biz.TaskStatusWaiting || task.Status == biz.TaskStatusRunning {
|
|
return errors.New(r.t.Get("please cancel the task first"))
|
|
}
|
|
|
|
// 清理任务日志文件
|
|
if task.Log != "" {
|
|
_ = os.Remove(task.Log)
|
|
}
|
|
|
|
return r.db.Where("id = ?", id).Delete(&biz.Task{}).Error
|
|
}
|
|
|
|
func (r *taskRepo) Cancel(id uint) error {
|
|
// 等待中的任务直接原子标记取消，避免与运行器取任务竞争
|
|
result := r.db.Model(&biz.Task{}).Where("id = ? AND status = ?", id, biz.TaskStatusWaiting).Update("status", biz.TaskStatusCanceled)
|
|
if result.Error != nil {
|
|
return result.Error
|
|
}
|
|
if result.RowsAffected > 0 {
|
|
return nil
|
|
}
|
|
|
|
task, err := r.Get(id)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if task.Status != biz.TaskStatusRunning {
|
|
return errors.New(r.t.Get("task is not waiting or running"))
|
|
}
|
|
|
|
// 运行中的任务交由运行器杀死进程组
|
|
if !r.runner.Cancel(id) {
|
|
return errors.New(r.t.Get("task has already finished"))
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *taskRepo) UpdateStatus(id uint, status biz.TaskStatus) error {
|
|
return r.db.Model(&biz.Task{}).Where("id = ?", id).Update("status", status).Error
|
|
}
|
|
|
|
func (r *taskRepo) UpdateLog(id uint, log string) error {
|
|
return r.db.Model(&biz.Task{}).Where("id = ?", id).Update("log", log).Error
|
|
}
|
|
|
|
func (r *taskRepo) Push(task *biz.Task) error {
|
|
// 防止有人喜欢酒吧点炒饭，按语言无关的任务标识去重
|
|
if task.Key != "" {
|
|
var count int64
|
|
if err := r.db.Model(&biz.Task{}).Where("`key` = ? and (status = ? or status = ?)", task.Key, biz.TaskStatusWaiting, biz.TaskStatusRunning).Count(&count).Error; err != nil {
|
|
return err
|
|
}
|
|
if count > 0 {
|
|
return errors.New(r.t.Get("duplicate submission, please wait for the previous task to end"))
|
|
}
|
|
}
|
|
|
|
if err := r.db.Create(task).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
r.runner.Notify()
|
|
return nil
|
|
}
|