68 lines
1.6 KiB
Go
68 lines
1.6 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"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// AsyncTask 异步任务实体
|
|||
|
|
type AsyncTask struct {
|
|||
|
|
ID string `gorm:"type:char(36);primaryKey"`
|
|||
|
|
Type string `gorm:"not null;index"`
|
|||
|
|
Payload string `gorm:"type:text"`
|
|||
|
|
Status TaskStatus `gorm:"not null;index"`
|
|||
|
|
ScheduledAt *time.Time `gorm:"index"`
|
|||
|
|
StartedAt *time.Time
|
|||
|
|
CompletedAt *time.Time
|
|||
|
|
ErrorMsg string
|
|||
|
|
RetryCount int `gorm:"default:0"`
|
|||
|
|
MaxRetries int `gorm:"default:5"`
|
|||
|
|
CreatedAt time.Time
|
|||
|
|
UpdatedAt time.Time
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TableName 指定表名
|
|||
|
|
func (AsyncTask) TableName() string {
|
|||
|
|
return "async_tasks"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// BeforeCreate GORM钩子，在创建前生成UUID
|
|||
|
|
func (t *AsyncTask) BeforeCreate(tx *gorm.DB) error {
|
|||
|
|
if t.ID == "" {
|
|||
|
|
t.ID = uuid.New().String()
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsCompleted 检查任务是否已完成
|
|||
|
|
func (t *AsyncTask) IsCompleted() bool {
|
|||
|
|
return t.Status == TaskStatusCompleted
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsFailed 检查任务是否失败
|
|||
|
|
func (t *AsyncTask) IsFailed() bool {
|
|||
|
|
return t.Status == TaskStatusFailed
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// IsCancelled 检查任务是否已取消
|
|||
|
|
func (t *AsyncTask) IsCancelled() bool {
|
|||
|
|
return t.Status == TaskStatusCancelled
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// CanRetry 检查任务是否可以重试
|
|||
|
|
func (t *AsyncTask) CanRetry() bool {
|
|||
|
|
return t.Status == TaskStatusFailed && t.RetryCount < t.MaxRetries
|
|||
|
|
}
|