74 lines
1.9 KiB
Go
74 lines
1.9 KiB
Go
package model
|
||
|
||
import (
|
||
"github.com/cloudreve/Cloudreve/v3/pkg/util"
|
||
"github.com/jinzhu/gorm"
|
||
)
|
||
|
||
// Task 任务模型
|
||
type Task struct {
|
||
gorm.Model
|
||
Status int // 任务状态
|
||
Type int // 任务类型
|
||
UserID uint // 发起者UID，0表示为系统发起
|
||
Progress int // 进度
|
||
Error string `gorm:"type:text"` // 错误信息
|
||
Props string `gorm:"type:text"` // 任务属性
|
||
}
|
||
|
||
// Create 创建任务记录
|
||
func (task *Task) Create() (uint, error) {
|
||
if err := DB.Create(task).Error; err != nil {
|
||
util.Log().Warning("Failed to insert task record: %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 in (?)", status).Find(&tasks)
|
||
return tasks
|
||
}
|
||
|
||
// GetTasksByID 根据ID检索任务
|
||
func GetTasksByID(id interface{}) (*Task, error) {
|
||
task := &Task{}
|
||
result := DB.Where("id = ?", id).First(task)
|
||
return task, result.Error
|
||
}
|
||
|
||
// ListTasks 列出用户所属的任务
|
||
func ListTasks(uid uint, page, pageSize int, order string) ([]Task, int) {
|
||
var (
|
||
tasks []Task
|
||
total int
|
||
)
|
||
dbChain := DB
|
||
dbChain = dbChain.Where("user_id = ?", uid)
|
||
|
||
// 计算总数用于分页
|
||
dbChain.Model(&Task{}).Count(&total)
|
||
|
||
// 查询记录
|
||
dbChain.Limit(pageSize).Offset((page - 1) * pageSize).Order(order).Find(&tasks)
|
||
|
||
return tasks, total
|
||
}
|