jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-06-10 03:57:39 +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 { func NewEngine(db db.Database, language string) *Engine {
return &Engine{ return &Engine{
db: db, db: db,
cron: cron.New(), cron: cron.New(),
tasks: utils.Map[int, *Task]{}, tasks: utils.Map[int, *Task]{},
schedulers: utils.Map[string, scheduler]{}, schedulers: utils.Map[string, scheduler]{},
language: language, language: language,
} }
} }
@@ -36,7 +36,7 @@ type Engine struct {
tasks utils.Map[int, *Task] tasks utils.Map[int, *Task]
language string language string
schedulers utils.Map[string, scheduler] schedulers utils.Map[string, scheduler]
buildin *buildin.Downloader buildin *buildin.Downloader
} }
func (c *Engine) registerCronJob(name string, cron string, f func() error) { func (c *Engine) registerCronJob(name string, cron string, f func() error) {
@@ -52,12 +52,11 @@ func (c *Engine) Init() {
go c.checkW500PosterOnStartup() go c.checkW500PosterOnStartup()
} }
func (c *Engine) GetTask(id int) (*Task, bool) { func (c *Engine) GetTask(id int) (*Task, bool) {
return c.tasks.Load(id) 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()) cl, err := buildin.NewDownloader(c.db.GetDownloadDir())
if err != nil { if err != nil {
log.Warnf("buildin downloader error: %v", err) log.Warnf("buildin downloader error: %v", err)
@@ -66,7 +65,7 @@ func (c *Engine) reloadUsingBuildinDownloader(h *ent.History) error{
if err != nil { if err != nil {
return errors.Wrap(err, "download torrent") return errors.Wrap(err, "download torrent")
} }
c.tasks.Store(h.ID, &Task{Torrent: t}) c.tasks.Store(h.ID, &Task{Torrent: t})
return nil return nil
} }
@@ -145,13 +144,13 @@ func (c *Engine) reloadTasks() {
} }
func (c *Engine) buildInDownloader() (pkg.Downloader, error) { func (c *Engine) buildInDownloader() (pkg.Downloader, error) {
if c.buildin!= nil { if c.buildin != nil {
return c.buildin, nil return c.buildin, nil
} }
dir := c.db.GetDownloadDir() dir := c.db.GetDownloadDir()
d, err := buildin.NewDownloader(dir) d, err := buildin.NewDownloader(dir)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "buildin downloader") return nil, errors.Wrap(err, "buildin downloader")
} }
c.buildin = d c.buildin = d
return d, nil return d, nil