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 }