jiazhizhong/polaris
1
0
Fork 0
mirror of https://github.com/simon-ding/polaris.git synced 2026-02-27 22:20:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: reduce time consuming

Browse Source
This commit is contained in:
Simon Ding
2024-10-10 10:29:47 +08:00
parent 1728690860
commit a22100b2fc
5 changed files with 56 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/qbittorrent/qbittorrent.go
View File

@@ -56,11 +56,16 @@ func (c *Client) GetAll() ([]pkg.Torrent, error) {
}
func (c *Client) Download(link, dir string) (pkg.Torrent, error) {
hash, err := utils.MagnetHash(link)
magnet, err := utils.Link2Magnet(link)
if err != nil {
return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
}
hash, err := utils.MagnetHash(magnet)
if err != nil {
return nil, errors.Wrap(err, "get hash")
}
err = c.c.DownloadLinks([]string{link}, qbt.DownloadOptions{Savepath: &dir})
err = c.c.DownloadLinks([]string{magnet}, qbt.DownloadOptions{Savepath: &dir})
if err != nil {
return nil, errors.Wrap(err, "qbt download")
}
@@ -69,11 +74,16 @@ func (c *Client) Download(link, dir string) (pkg.Torrent, error) {
}
func NewTorrent(info Info, magnet string) (*Torrent, error) {
func NewTorrent(info Info, link string) (*Torrent, error) {
c, err := NewClient(info.URL, info.User, info.Password)
if err != nil {
return nil, err
}
magnet, err := utils.Link2Magnet(link)
if err != nil {
return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
}
hash, err := utils.MagnetHash(magnet)
if err != nil {
return nil, err

13
pkg/torznab/torznab.go
View File

@@ -9,7 +9,6 @@ import (
"net/url"
"polaris/db"
"polaris/log"
"polaris/pkg/utils"
"slices"
"strconv"
"time"
@@ -81,14 +80,14 @@ func (r *Response) ToResults(indexer *db.TorznabInfo) []Result {
if slices.Contains(item.Category, "3000") { //exclude audio files
continue
}
link, err := utils.Link2Magnet(item.Link) //TODO time consuming operation
if err != nil {
log.Warnf("converting link to magnet error, error: %v, link: %v", err, item.Link)
continue
}
// link, err := utils.Link2Magnet(item.Link) //TODO time consuming operation
// if err != nil {
// log.Warnf("converting link to magnet error, error: %v, link: %v", err, item.Link)
// continue
// }
r := Result{
Name: item.Title,
Link: link,
Link: item.Link,
Size: mustAtoI(item.Size),
Seeders: mustAtoI(item.GetAttr("seeders")),
Peers: mustAtoI(item.GetAttr("peers")),

21
pkg/transmission/transmission.go
View File

@@ -61,13 +61,18 @@ func (c *Client) GetAll() ([]pkg.Torrent, error) {
}
func (c *Client) Download(link, dir string) (pkg.Torrent, error) {
hash, err := utils.MagnetHash(link)
magnet, err := utils.Link2Magnet(link)
if err != nil {
return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
}
hash, err := utils.MagnetHash(magnet)
if err != nil {
return nil, errors.Wrap(err, "get hash")
}
t, err := c.c.TorrentAdd(context.TODO(), transmissionrpc.TorrentAddPayload{
Filename: &link,
Filename: &magnet,
DownloadDir: &dir,
})
log.Debugf("get torrent info: %+v", t)
@@ -79,17 +84,23 @@ func (c *Client) Download(link, dir string) (pkg.Torrent, error) {
}, err
}
func NewTorrent(cfg Config, magnet string) (*Torrent, error) {
func NewTorrent(cfg Config, link string) (*Torrent, error) {
c, err := NewClient(cfg)
if err != nil {
return nil, err
}
magnet, err := utils.Link2Magnet(link)
if err != nil {
return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
}
hash, err := utils.MagnetHash(magnet)
if err != nil {
return nil, err
}
t := &Torrent{
t := &Torrent{
c: c.c,
hash: hash,
//cfg: cfg,
@@ -103,7 +114,7 @@ func NewTorrent(cfg Config, magnet string) (*Torrent, error) {
type Torrent struct {
//t *transmissionrpc.Torrent
c *transmissionrpc.Client
hash string `json:"hash"`
hash string
//cfg Config
}

17
server/core/resources.go
View File

@@ -44,7 +44,12 @@ func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum,
} else { //season package download
ep = &ent.Episode{}
}
torrent, err := trc.Download(r1.Link, downloadDir)
magnet, err := utils.Link2Magnet(r1.Link)
if err != nil {
return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", r1.Link, err)
}
torrent, err := trc.Download(magnet, downloadDir)
if err != nil {
return nil, errors.Wrap(err, "downloading")
}
@@ -60,7 +65,7 @@ func (c *Client) DownloadEpisodeTorrent(r1 torznab.Result, seriesId, seasonNum,
Status: history.StatusRunning,
Size: r1.Size,
//Saved: torrent.Save(),
Link: r1.Link,
Link: magnet,
DownloadClientID: dlc.ID,
IndexerID: r1.IndexerId,
})
@@ -107,8 +112,12 @@ func (c *Client) DownloadMovie(m *ent.Media, link, name string, size int, indexe
if err != nil {
return nil, errors.Wrap(err, "connect transmission")
}
magnet, err := utils.Link2Magnet(link)
if err != nil {
return nil, errors.Errorf("converting link to magnet error, link: %v, error: %v", link, err)
}
torrent, err := trc.Download(link, c.db.GetDownloadDir())
torrent, err := trc.Download(magnet, c.db.GetDownloadDir())
if err != nil {
return nil, errors.Wrap(err, "downloading")
}
@@ -127,7 +136,7 @@ func (c *Client) DownloadMovie(m *ent.Media, link, name string, size int, indexe
Status: history.StatusRunning,
Size: size,
//Saved: torrent.Save(),
Link: link,
Link: magnet,
DownloadClientID: dlc.ID,
IndexerID: indexerID,
})

10
server/core/scheduler.go
View File

@@ -363,7 +363,13 @@ func (c *Client) downloadMovieSingleEpisode(ep *ent.Episode, targetDir string) (
}
r1 := res[0]
log.Infof("begin download torrent resource: %v", r1.Name)
torrent, err := trc.Download(r1.Link, c.db.GetDownloadDir())
magnet, err := utils.Link2Magnet(r1.Link)
if err != nil {
return "", errors.Errorf("converting link to magnet error, link: %v, error: %v", r1.Link, err)
}
torrent, err := trc.Download(magnet, c.db.GetDownloadDir())
if err != nil {
return "", errors.Wrap(err, "downloading")
}
@@ -377,7 +383,7 @@ func (c *Client) downloadMovieSingleEpisode(ep *ent.Episode, targetDir string) (
Status: history.StatusRunning,
Size: r1.Size,
//Saved: torrent.Save(),
Link: r1.Link,
Link: magnet,
DownloadClientID: dlc.ID,
IndexerID: r1.IndexerId,
})