76 lines
2.3 KiB
Go
76 lines
2.3 KiB
Go
|
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
|
|||
|
|
}
|