feat: support reload qbit tasks

pkg/doc.go

@@ -10,6 +10,7 @@ type Torrent interface {
 	Exists() bool
 	SeedRatio() (float64, error)
 	GetHash() string
+	Reload() error
 }
 
 type Downloader interface {
@@ -17,5 +18,3 @@ type Downloader interface {
 	Download(link, dir string) (Torrent, error)
 }
 
-type Storage interface {
-}

pkg/qbittorrent/qbittorrent.go

@@ -174,3 +174,15 @@ func (t *Torrent) SeedRatio() (float64, error) {
 	}
 	return qb.Ratio, nil
 }
+
+func (t *Torrent) Reload() error {
+	c, err := NewClient(t.URL, t.User, t.Password)
+	if err != nil {
+		return err
+	}
+	t.c = c.c
+	if !t.Exists() {
+		return errors.Errorf("torrent not exists: %v", t.Hash)
+	}
+	return nil
+}

pkg/transmission/transmission.go

@@ -105,12 +105,15 @@ type Torrent struct {
 	Config
 }
 
-func (t *Torrent) reloadClient() error {
+func (t *Torrent) Reload() error {
 	c, err := NewClient(t.Config)
 	if err != nil {
 		return err
 	}
 	t.c = c.c
+	if !t.Exists() {
+		return errors.Errorf("torrent not exists: %v", t.Hash)
+	}
 	return nil
 }
 
@@ -211,18 +214,3 @@ func (t *Torrent) Save() string {
 func (t *Torrent) GetHash() string {
 	return t.Hash
 }
-
-
-func ReloadTorrent(s string) (*Torrent, error) {
-	var torrent = Torrent{}
-	err := json.Unmarshal([]byte(s), &torrent)
-	if err != nil {
-		return nil, err
-	}
-
-	err = torrent.reloadClient()
-	if err != nil {
-		return nil, errors.Wrap(err, "reload client")
-	}
-	return &torrent, nil
-}

server/core/client.go

@@ -1,6 +1,7 @@
 package core
 
 import (
+	"encoding/json"
 	"polaris/db"
 	"polaris/ent"
 	"polaris/ent/downloadclients"
@@ -51,19 +52,44 @@ func (c *Client) Init() {
 func (c *Client) reloadTasks() {
 	allTasks := c.db.GetRunningHistories()
 	for _, t := range allTasks {
-		torrent, err := transmission.ReloadTorrent(t.Saved)
+		var torrent pkg.Torrent
-		if err != nil {
+		if tt, err := c.reloadTransmiision(t.Saved); err == nil {
-			log.Errorf("relaod task %s failed: %v", t.SourceTitle, err)
+			torrent = tt
+			log.Infof("load transmission task: %v", t.Saved)
+		} else if tt, err := c.reloadQbit(t.Saved); err == nil {
+			torrent = tt
+			log.Infof("load qbit task: %v", t.Saved)
+		} else {
+			log.Warnf("load task fail: %v", t.Saved)
 			continue
 		}
-		if !torrent.Exists() { //只要种子还存在于客户端中，就重新加载，有可能是还在做种中
-			continue
-		}
-		log.Infof("reloading task: %d %s", t.ID, t.SourceTitle)
 		c.tasks[t.ID] = &Task{Torrent: torrent}
 	}
 }
 
+func (c *Client) reloadTransmiision(s string) (pkg.Torrent, error) {
+	var t transmission.Torrent
+	if err := json.Unmarshal([]byte(s), &t); err != nil {
+		return nil, err
+	}
+	if err := t.Reload(); err != nil {
+		return nil, err
+	}
+	return &t, nil
+}
+
+func (c *Client) reloadQbit(s string) (pkg.Torrent, error) {
+	var t qbittorrent.Torrent
+	if err := json.Unmarshal([]byte(s), &t); err != nil {
+		return nil, err
+	}
+	if err := t.Reload(); err != nil {
+		return nil, err
+	}
+	return &t, nil
+}
+
+
 func (c *Client) GetDownloadClient() (pkg.Downloader, *ent.DownloadClients, error) {
 	downloaders := c.db.GetAllDonloadClients()
 	for _, d := range downloaders {