jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-04-21 19:27:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: format

Browse Source
This commit is contained in:
Simon Ding
2025-04-23 10:26:32 +08:00
parent 9e41a1513c
commit aecbc5f657
1 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
engine/client.go
View File

@@ -18,11 +18,11 @@ import (
func NewEngine(db db.Database, language string) *Engine {
return &Engine{
db: db,
cron: cron.New(),
tasks: utils.Map[int, *Task]{},
db: db,
cron: cron.New(),
tasks: utils.Map[int, *Task]{},
schedulers: utils.Map[string, scheduler]{},
language: language,
language: language,
}
}
@@ -36,7 +36,7 @@ type Engine struct {
tasks utils.Map[int, *Task]
language string
schedulers utils.Map[string, scheduler]
buildin *buildin.Downloader
buildin *buildin.Downloader
}
func (c *Engine) registerCronJob(name string, cron string, f func() error) {
@@ -52,12 +52,11 @@ func (c *Engine) Init() {
go c.checkW500PosterOnStartup()
}
func (c *Engine) GetTask(id int) (*Task, bool) {
return c.tasks.Load(id)
}
func (c *Engine) reloadUsingBuildinDownloader(h *ent.History) error{
func (c *Engine) reloadUsingBuildinDownloader(h *ent.History) error {
cl, err := buildin.NewDownloader(c.db.GetDownloadDir())
if err != nil {
log.Warnf("buildin downloader error: %v", err)
@@ -66,7 +65,7 @@ func (c *Engine) reloadUsingBuildinDownloader(h *ent.History) error{
if err != nil {
return errors.Wrap(err, "download torrent")
}
c.tasks.Store(h.ID, &Task{Torrent: t})
c.tasks.Store(h.ID, &Task{Torrent: t})
return nil
}
@@ -145,13 +144,13 @@ func (c *Engine) reloadTasks() {
}
func (c *Engine) buildInDownloader() (pkg.Downloader, error) {
if c.buildin!= nil {
return c.buildin, nil
if c.buildin != nil {
return c.buildin, nil
}
dir := c.db.GetDownloadDir()
d, err := buildin.NewDownloader(dir)
if err != nil {
return nil, errors.Wrap(err, "buildin downloader")
return nil, errors.Wrap(err, "buildin downloader")
}
c.buildin = d
return d, nil