定义执行历史和执行队列model, 以及基础的增删改查功能
This commit is contained in:
75
internal/infra/models/execution.go
Normal file
75
internal/infra/models/execution.go
Normal file
@@ -0,0 +1,75 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"time"
|
||||
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// 定义系统任务的特殊ID
|
||||
const (
|
||||
SystemTaskIDResolvePlan int = -1 // 代表“解析计划”的系统任务
|
||||
)
|
||||
|
||||
type ExecutionStatus string
|
||||
|
||||
const (
|
||||
ExecutionStatusStarted ExecutionStatus = "started" // 开始执行
|
||||
ExecutionStatusCompleted ExecutionStatus = "completed" // 执行完成
|
||||
ExecutionStatusFailed ExecutionStatus = "failed" // 执行失败
|
||||
ExecutionStatusCancelled ExecutionStatus = "cancelled" // 执行取消
|
||||
ExecutionStatusWaiting ExecutionStatus = "waiting" // 等待执行 (用于预写日志)
|
||||
)
|
||||
|
||||
// PlanExecutionLog 记录整个计划的一次执行历史
|
||||
type PlanExecutionLog struct {
|
||||
gorm.Model
|
||||
PlanID uint `gorm:"index"`
|
||||
Status ExecutionStatus
|
||||
StartedAt time.Time
|
||||
EndedAt time.Time
|
||||
Error string
|
||||
}
|
||||
|
||||
// TableName 自定义 GORM 使用的数据库表名
|
||||
func (PlanExecutionLog) TableName() string {
|
||||
return "plan_execution_logs"
|
||||
}
|
||||
|
||||
// TaskExecutionLog 记录单个任务的一次执行历史
|
||||
type TaskExecutionLog struct {
|
||||
gorm.Model
|
||||
PlanExecutionLogID uint `gorm:"index"` // 关联到某次计划执行
|
||||
|
||||
// TaskID 使用 int 类型以容纳特殊的负数ID,代表系统任务
|
||||
TaskID int `gorm:"index"`
|
||||
|
||||
// 关键改动:移除了 OnDelete 约束。
|
||||
Task Task `gorm:"foreignKey:TaskID;constraint:OnUpdate:CASCADE;"`
|
||||
|
||||
Status ExecutionStatus
|
||||
Output string // 任务执行的输出或错误信息
|
||||
StartedAt time.Time
|
||||
EndedAt time.Time
|
||||
}
|
||||
|
||||
// TableName 自定义 GORM 使用的数据库表名
|
||||
func (TaskExecutionLog) TableName() string {
|
||||
return "task_execution_logs"
|
||||
}
|
||||
|
||||
// AfterFind 是 GORM 的一个钩子,在查询数据后自动执行
|
||||
// 我们用它来优雅地处理系统任务的“虚拟”Task定义
|
||||
func (log *TaskExecutionLog) AfterFind(tx *gorm.DB) (err error) {
|
||||
// 检查是否是我们的“解析计划”系统任务
|
||||
if log.TaskID == SystemTaskIDResolvePlan {
|
||||
// 如果是,手动创建一个写死的 Task 定义并绑定上去
|
||||
// 这使得上层服务在处理日志时,无需关心TaskID是否为负数
|
||||
log.Task = Task{
|
||||
// 注意:这里不能设置 ID,否则 GORM 可能会混淆
|
||||
Name: "系统:解析并启动计划",
|
||||
Description: "这是一个由系统自动触发的内部任务,用于准备计划的执行。",
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
@@ -3,6 +3,7 @@ package models
|
||||
import (
|
||||
"fmt"
|
||||
"sort"
|
||||
"time"
|
||||
|
||||
"gorm.io/datatypes"
|
||||
"gorm.io/gorm"
|
||||
@@ -133,7 +134,11 @@ func (SubPlan) TableName() string {
|
||||
|
||||
// Task 代表计划中的一个任务,具有执行顺序
|
||||
type Task struct {
|
||||
gorm.Model
|
||||
// 手动定义字段以将 ID 类型设置为 int,以匹配 TaskExecutionLog 中的 TaskID
|
||||
ID int `gorm:"primarykey"`
|
||||
CreatedAt time.Time
|
||||
UpdatedAt time.Time
|
||||
DeletedAt gorm.DeletedAt `gorm:"index"` // 保持软删除功能
|
||||
|
||||
PlanID uint `gorm:"not null;index" json:"plan_id"` // 此任务所属计划的ID
|
||||
Name string `gorm:"not null" json:"name"`
|
||||
|
||||
32
internal/infra/models/schedule.go
Normal file
32
internal/infra/models/schedule.go
Normal file
@@ -0,0 +1,32 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"time"
|
||||
)
|
||||
|
||||
// PendingTask 是一个待执行任务队列, 里面会储存待执行的Task以及这个Task什么时候执行
|
||||
// 它是一个纯粹的工作队列,任务被认领后即被删除。
|
||||
type PendingTask struct {
|
||||
// 手动填充必须字段以实现硬删除,不内嵌 gorm.Model
|
||||
ID uint `gorm:"primarykey"`
|
||||
CreatedAt time.Time
|
||||
UpdatedAt time.Time
|
||||
|
||||
// 业务字段
|
||||
|
||||
// TaskID 使用 int 类型以容纳特殊的负数ID,代表系统任务
|
||||
TaskID int `gorm:"index"`
|
||||
|
||||
ExecuteAt time.Time `gorm:"index"` // 任务执行时间
|
||||
TaskExecutionLogID uint `gorm:"unique;not null"` // 对应的执行历史记录ID
|
||||
|
||||
// 关联关系定义
|
||||
// 通过 TaskExecutionLogID 关联到唯一的 TaskExecutionLog 记录
|
||||
// ON DELETE CASCADE 确保如果日志被删除,这个待办任务也会被自动清理
|
||||
TaskExecutionLog TaskExecutionLog `gorm:"foreignKey:TaskExecutionLogID;references:ID;constraint:OnUpdate:CASCADE,OnDelete:CASCADE;"`
|
||||
}
|
||||
|
||||
// TableName 自定义 GORM 使用的数据库表名
|
||||
func (PendingTask) TableName() string {
|
||||
return "pending_tasks"
|
||||
}
|
||||
Reference in New Issue
Block a user