Feat: task queue / compression task

models/task.go

@@ -0,0 +1,48 @@
+package model
+
+import (
+	"github.com/HFO4/cloudreve/pkg/util"
+	"github.com/jinzhu/gorm"
+)
+
+// Task 任务模型
+type Task struct {
+	gorm.Model
+	Status   int    // 任务状态
+	Type     int    // 任务类型
+	UserID   uint   // 发起者UID，0表示为系统发起
+	Progress int    // 进度
+	Error    string // 错误信息
+	Props    string `gorm:"type:text"` // 任务属性
+}
+
+// Create 创建任务记录
+func (task *Task) Create() (uint, error) {
+	if err := DB.Create(task).Error; err != nil {
+		util.Log().Warning("无法插入任务记录, %s", err)
+		return 0, err
+	}
+	return task.ID, nil
+}
+
+// SetStatus 设定任务状态
+func (task *Task) SetStatus(status int) error {
+	return DB.Model(task).Select("status").Updates(map[string]interface{}{"status": status}).Error
+}
+
+// SetProgress 设定任务进度
+func (task *Task) SetProgress(progress int) error {
+	return DB.Model(task).Select("progress").Updates(map[string]interface{}{"progress": progress}).Error
+}
+
+// SetError 设定错误信息
+func (task *Task) SetError(err string) error {
+	return DB.Model(task).Select("error").Updates(map[string]interface{}{"error": err}).Error
+}
+
+// GetTasksByStatus 根据状态检索任务
+func GetTasksByStatus(status int) []Task {
+	var tasks []Task
+	DB.Where("status = ?", status).Find(&tasks)
+	return tasks
+}