mirror of
https://github.com/tnb-labs/panel.git
synced 2026-09-01 14:55:12 +08:00
耗子
443516a2cf
依赖注入：
- 移除 google/wire（含 wire.go/wire_gen.go 与全部 ProviderSet），
改用 samber/do v2 单注入器 + 双入口惰性构建
- 贡献模型替代命令式注册：路由 routes:、命令 commands:、
任务 jobs: 前缀经 internal/registry 收集与校验
- 构造函数统一 func NewXxx(i do.Injector) (T, error)
三层架构：
- 补全用例层：每个 biz.XxxRepo 配 XxxUsecase，
service/command/job 表现层只依赖用例，不再直接引用仓储
路由与文档：
- route/http.go 按域拆为声明式 Endpoint 贡献，
Endpoint 承载登录白名单与端点限流语义
- 调试模式下提供 OpenAPI 3.1 文档：/openapi.json 与 /docs（Scalar），
从 validate 标签生成
目录与依赖：
- internal/http/{middleware,request,rule} 拍平至 internal/*
- CLI 命令拆至 internal/command
- 日志轮转 timberjack 换为 libtnb/logrotate
- 升级 validator/cron/sessions/gormstore/sqlite/securecookie
- 数据库迁移保持 gormigrate 不变
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
79 lines
2.1 KiB
Go
79 lines
2.1 KiB
Go
package data
|
|
|
|
import (
|
|
"errors"
|
|
"log/slog"
|
|
|
|
"github.com/leonelquinteros/gotext"
|
|
"github.com/samber/do/v2"
|
|
"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(i do.Injector) (biz.TaskRepo, error) {
|
|
return &taskRepo{
|
|
t: do.MustInvoke[*gotext.Locale](i),
|
|
db: do.MustInvoke[*gorm.DB](i),
|
|
log: do.MustInvoke[*slog.Logger](i),
|
|
runner: do.MustInvoke[types.TaskRunner](i),
|
|
}, nil
|
|
}
|
|
|
|
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 {
|
|
return r.db.Model(&biz.Task{}).Where("id = ?", id).Delete(&biz.Task{}).Error
|
|
}
|
|
|
|
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 {
|
|
// 防止有人喜欢酒吧点炒饭
|
|
var count int64
|
|
if err := r.db.Model(&biz.Task{}).Where("shell = ? and (status = ? or status = ?)", task.Shell, 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
|
|
}
|