mirror of
https://github.com/cloudreve/cloudreve.git
synced 2025-10-31 08:39:10 +08:00
107 lines
2.9 KiB
Go
107 lines
2.9 KiB
Go
package model
|
||
|
||
import (
|
||
"encoding/json"
|
||
"github.com/HFO4/cloudreve/pkg/util"
|
||
"github.com/jinzhu/gorm"
|
||
"github.com/zyxar/argo/rpc"
|
||
)
|
||
|
||
// Download 离线下载队列模型
|
||
type Download struct {
|
||
gorm.Model
|
||
Status int // 任务状态
|
||
Type int // 任务类型
|
||
Source string `gorm:"type:text"` // 文件下载地址
|
||
TotalSize uint64 // 文件大小
|
||
DownloadedSize uint64 // 文件大小
|
||
GID string `gorm:"size:32,index:gid"` // 任务ID
|
||
Speed int // 下载速度
|
||
Parent string `gorm:"type:text"` // 存储目录
|
||
Attrs string `gorm:"type:text"` // 任务状态属性
|
||
Error string `gorm:"type:text"` // 错误描述
|
||
Dst string `gorm:"type:text"` // 用户文件系统存储父目录路径
|
||
UserID uint // 发起者UID
|
||
TaskID uint // 对应的转存任务ID
|
||
|
||
// 关联模型
|
||
User *User `gorm:"PRELOAD:false,association_autoupdate:false"`
|
||
|
||
// 数据库忽略字段
|
||
StatusInfo rpc.StatusInfo `gorm:"-"`
|
||
Task *Task `gorm:"-"`
|
||
}
|
||
|
||
// AfterFind 找到下载任务后的钩子，处理Status结构
|
||
func (task *Download) AfterFind() (err error) {
|
||
// 解析状态
|
||
if task.Attrs != "" {
|
||
err = json.Unmarshal([]byte(task.Attrs), &task.StatusInfo)
|
||
}
|
||
|
||
if task.TaskID != 0 {
|
||
task.Task, _ = GetTasksByID(task.TaskID)
|
||
}
|
||
|
||
return err
|
||
}
|
||
|
||
// BeforeSave Save下载任务前的钩子
|
||
func (task *Download) BeforeSave() (err error) {
|
||
return task.AfterFind()
|
||
}
|
||
|
||
// Create 创建离线下载记录
|
||
func (task *Download) Create() (uint, error) {
|
||
if err := DB.Create(task).Error; err != nil {
|
||
util.Log().Warning("无法插入离线下载记录, %s", err)
|
||
return 0, err
|
||
}
|
||
return task.ID, nil
|
||
}
|
||
|
||
// Save 更新
|
||
func (task *Download) Save() error {
|
||
if err := DB.Save(task).Error; err != nil {
|
||
util.Log().Warning("无法更新离线下载记录, %s", err)
|
||
return err
|
||
}
|
||
return nil
|
||
}
|
||
|
||
// GetDownloadsByStatus 根据状态检索下载
|
||
func GetDownloadsByStatus(status ...int) []Download {
|
||
var tasks []Download
|
||
DB.Where("status in (?)", status).Find(&tasks)
|
||
return tasks
|
||
}
|
||
|
||
// GetDownloadsByStatusAndUser 根据状态检索和用户ID下载
|
||
// page 为 0 表示列出所有，非零时分页
|
||
func GetDownloadsByStatusAndUser(page, uid uint, status ...int) []Download {
|
||
var tasks []Download
|
||
dbChain := DB
|
||
if page > 0 {
|
||
dbChain = dbChain.Limit(10).Offset((page - 1) * 10).Order("updated_at DESC")
|
||
}
|
||
dbChain.Where("user_id = ? and status in (?)", uid, status).Find(&tasks)
|
||
return tasks
|
||
}
|
||
|
||
// GetDownloadByGid 根据GID和用户ID查找下载
|
||
func GetDownloadByGid(gid string, uid uint) (*Download, error) {
|
||
download := &Download{}
|
||
result := DB.Where("user_id = ? and g_id = ?", uid, gid).First(download)
|
||
return download, result.Error
|
||
}
|
||
|
||
// GetOwner 获取下载任务所属用户
|
||
func (task *Download) GetOwner() *User {
|
||
if task.User == nil {
|
||
if user, err := GetUserByID(task.UserID); err == nil {
|
||
return &user
|
||
}
|
||
}
|
||
return task.User
|
||
}
|