83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"time"
|
|
|
|
"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
// PendingTaskRepository 定义了与待执行任务队列交互的接口。
|
|
type PendingTaskRepository interface {
|
|
CreatePendingTasksInBatch(tasks []*models.PendingTask) error
|
|
ClaimNextDueTask() (*models.TaskExecutionLog, error)
|
|
}
|
|
|
|
// pendingTaskRepository 是使用 GORM 的具体实现。
|
|
type pendingTaskRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewPendingTaskRepository 创建一个新的待执行任务队列仓库。
|
|
// 它接收一个 GORM DB 实例作为依赖。
|
|
func NewPendingTaskRepository(db *gorm.DB) PendingTaskRepository {
|
|
return &pendingTaskRepository{db: db}
|
|
}
|
|
|
|
// CreatePendingTasksInBatch 在一次数据库调用中创建多个待执行任务条目。
|
|
func (r *pendingTaskRepository) CreatePendingTasksInBatch(tasks []*models.PendingTask) error {
|
|
return r.db.Create(&tasks).Error
|
|
}
|
|
|
|
// ClaimNextDueTask 以原子方式认领下一个到期任务。
|
|
// 它在一个事务中完成三件事：
|
|
// 1. 查找并锁定一个到期的待办任务。
|
|
// 2. 从待办队列中将其删除。
|
|
// 3. 将其在执行日志表中的状态更新为 'running'。
|
|
// 最后返回更新后的执行日志对象，作为执行的唯一凭证。
|
|
func (r *pendingTaskRepository) ClaimNextDueTask() (*models.TaskExecutionLog, error) {
|
|
var log models.TaskExecutionLog
|
|
|
|
err := r.db.Transaction(func(tx *gorm.DB) error {
|
|
var pendingTask models.PendingTask
|
|
|
|
// 1. 查找并用 FOR UPDATE 锁定一个到期的待办任务
|
|
if err := tx.Clauses(clause.Locking{Strength: "UPDATE"}).
|
|
Where("execute_at <= ?", time.Now()).
|
|
Order("execute_at ASC").
|
|
First(&pendingTask).Error; err != nil {
|
|
return err // 没找到是正常情况，事务将回滚
|
|
}
|
|
|
|
// 2. 在同一个事务中，立刻硬删除这个待办任务
|
|
if err := tx.Unscoped().Delete(&pendingTask).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// 3. 在同一个事务中，更新其在日志表中的孪生兄弟的状态
|
|
updates := map[string]interface{}{
|
|
"status": models.ExecutionStatusStarted,
|
|
"started_at": time.Now(),
|
|
}
|
|
if err := tx.Model(&models.TaskExecutionLog{}).Where("id = ?", pendingTask.TaskExecutionLogID).Updates(updates).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
// 4. 获取更新后的完整日志对象，以返回给调用者
|
|
if err := tx.First(&log, pendingTask.TaskExecutionLogID).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
})
|
|
|
|
if err != nil {
|
|
// 如果错误是 `gorm.ErrRecordNotFound`，它仅表示当前没有到期的任务。
|
|
// 服务层应该优雅地处理这种情况，而不是将其视为需要立即处理的严重错误。
|
|
return nil, err
|
|
}
|
|
|
|
return &log, nil
|
|
}
|