114 lines
3.4 KiB
Go
114 lines
3.4 KiB
Go
|
package entities
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/google/uuid"
|
|||
|
|
"gorm.io/gorm"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// TaskStatus 任务状态枚举
|
|||
|
|
type TaskStatus string
|
|||
|
|
|
|||
|
|
const (
|
|||
|
|
TaskStatusPending TaskStatus = "pending" // 等待执行
|
|||
|
|
TaskStatusRunning TaskStatus = "running" // 正在执行
|
|||
|
|
TaskStatusCompleted TaskStatus = "completed" // 已完成
|
|||
|
|
TaskStatusFailed TaskStatus = "failed" // 执行失败
|
|||
|
|
TaskStatusCancelled TaskStatus = "cancelled" // 已取消
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// ScheduledTask 定时任务状态管理实体
|
|||
|
|
type ScheduledTask struct {
|
|||
|
|
// 基础标识
|
|||
|
|
ID string `gorm:"primaryKey;type:varchar(36)" json:"id" comment:"任务唯一标识"`
|
|||
|
|
TaskID string `gorm:"type:varchar(100);not null;uniqueIndex" json:"task_id" comment:"Asynq任务ID"`
|
|||
|
|
TaskType string `gorm:"type:varchar(50);not null" json:"task_type" comment:"任务类型"`
|
|||
|
|
|
|||
|
|
// 关联信息
|
|||
|
|
ArticleID string `gorm:"type:varchar(36);not null;index" json:"article_id" comment:"关联的文章ID"`
|
|||
|
|
|
|||
|
|
// 任务状态
|
|||
|
|
Status TaskStatus `gorm:"type:varchar(20);not null;default:'pending'" json:"status" comment:"任务状态"`
|
|||
|
|
|
|||
|
|
// 时间信息
|
|||
|
|
ScheduledAt time.Time `gorm:"not null" json:"scheduled_at" comment:"计划执行时间"`
|
|||
|
|
StartedAt *time.Time `json:"started_at" comment:"开始执行时间"`
|
|||
|
|
CompletedAt *time.Time `json:"completed_at" comment:"完成时间"`
|
|||
|
|
|
|||
|
|
// 执行结果
|
|||
|
|
Error string `gorm:"type:text" json:"error" comment:"错误信息"`
|
|||
|
|
RetryCount int `gorm:"default:0" json:"retry_count" comment:"重试次数"`
|
|||
|
|
|
|||
|
|
// 时间戳字段
|
|||
|
|
CreatedAt time.Time `gorm:"autoCreateTime" json:"created_at" comment:"创建时间"`
|
|||
|
|
UpdatedAt time.Time `gorm:"autoUpdateTime" json:"updated_at" comment:"更新时间"`
|
|||
|
|
DeletedAt gorm.DeletedAt `gorm:"index" json:"-" comment:"软删除时间"`
|
|||
|
|
|
|||
|
|
// 关联关系
|
|||
|
|
Article *Article `gorm:"foreignKey:ArticleID" json:"article,omitempty" comment:"关联的文章"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TableName 指定表名
|
|||
|
|
func (ScheduledTask) TableName() string {
|
|||
|
|
return "scheduled_tasks"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BeforeCreate GORM钩子：创建前自动生成UUID
|
|||
|
|
func (st *ScheduledTask) BeforeCreate(tx *gorm.DB) error {
|
|||
|
|
if st.ID == "" {
|
|||
|
|
st.ID = uuid.New().String()
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MarkAsRunning 标记任务为正在执行
|
|||
|
|
func (st *ScheduledTask) MarkAsRunning() {
|
|||
|
|
st.Status = TaskStatusRunning
|
|||
|
|
now := time.Now()
|
|||
|
|
st.StartedAt = &now
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MarkAsCompleted 标记任务为已完成
|
|||
|
|
func (st *ScheduledTask) MarkAsCompleted() {
|
|||
|
|
st.Status = TaskStatusCompleted
|
|||
|
|
now := time.Now()
|
|||
|
|
st.CompletedAt = &now
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MarkAsFailed 标记任务为执行失败
|
|||
|
|
func (st *ScheduledTask) MarkAsFailed(errorMsg string) {
|
|||
|
|
st.Status = TaskStatusFailed
|
|||
|
|
now := time.Now()
|
|||
|
|
st.CompletedAt = &now
|
|||
|
|
st.Error = errorMsg
|
|||
|
|
st.RetryCount++
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MarkAsCancelled 标记任务为已取消
|
|||
|
|
func (st *ScheduledTask) MarkAsCancelled() {
|
|||
|
|
st.Status = TaskStatusCancelled
|
|||
|
|
now := time.Now()
|
|||
|
|
st.CompletedAt = &now
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsActive 判断任务是否处于活动状态
|
|||
|
|
func (st *ScheduledTask) IsActive() bool {
|
|||
|
|
return st.Status == TaskStatusPending || st.Status == TaskStatusRunning
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsCancelled 判断任务是否已取消
|
|||
|
|
func (st *ScheduledTask) IsCancelled() bool {
|
|||
|
|
return st.Status == TaskStatusCancelled
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsCompleted 判断任务是否已完成
|
|||
|
|
func (st *ScheduledTask) IsCompleted() bool {
|
|||
|
|
return st.Status == TaskStatusCompleted
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsFailed 判断任务是否执行失败
|
|||
|
|
func (st *ScheduledTask) IsFailed() bool {
|
|||
|
|
return st.Status == TaskStatusFailed
|
|||
|
|
}
|