52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
package service
|
|
|
|
import (
|
|
model2 "github.com/IceWhaleTech/CasaOS/service/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type DownloadService interface {
|
|
AddDownloadTask(m model2.PersionDownloadDBModel) //添加下载任务
|
|
EditDownloadState(m model2.PersionDownloadDBModel) //只修改状态
|
|
SaveDownload(m model2.PersionDownloadDBModel)
|
|
DelDownload(uuid string)
|
|
GetDownloadById(uuid string) model2.PersionDownloadDBModel
|
|
GetDownloadListByState(state string) []model2.PersionDownloadDBModel
|
|
}
|
|
type downloadService struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func (d *downloadService) AddDownloadTask(m model2.PersionDownloadDBModel) {
|
|
d.db.Create(&m)
|
|
}
|
|
func (d *downloadService) EditDownloadState(m model2.PersionDownloadDBModel) {
|
|
d.db.Model(&m).Where("uuid = ?", m.UUID).Update("state", m.State)
|
|
}
|
|
|
|
func (d *downloadService) DelDownload(uuid string) {
|
|
var m model2.PersionDownloadDBModel
|
|
d.db.Where("uuid = ?", uuid).Delete(&m)
|
|
}
|
|
func (d *downloadService) GetDownloadById(uuid string) model2.PersionDownloadDBModel {
|
|
var m model2.PersionDownloadDBModel
|
|
d.db.Model(m).Where("uuid = ?", uuid).First(&m)
|
|
return m
|
|
}
|
|
func (d *downloadService) GetDownloadListByState(state string) (list []model2.PersionDownloadDBModel) {
|
|
if len(state) == 0 {
|
|
d.db.Find(&list)
|
|
|
|
} else {
|
|
d.db.Where("state = ?", state).Find(&list)
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (d *downloadService) SaveDownload(m model2.PersionDownloadDBModel) {
|
|
d.db.Save(&m)
|
|
}
|
|
func NewDownloadService(db *gorm.DB) DownloadService {
|
|
return &downloadService{db: db}
|
|
}
|