idanoo/autobrr
1
0
Fork 0
mirror of https://github.com/idanoo/autobrr synced 2025-07-23 16:59:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: improve qbit re-announce (#140)

Browse source
This commit is contained in:
Ludvig Lundgren 2022-02-16 00:02:20 +01:00 • committed by GitHub
parent 78f83c674b
commit 43c42a7ee8
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

36
internal/action/qbittorrent.go
View file

@ -47,9 +47,9 @@ func (s *service) qbittorrent(qbt *qbittorrent.Client, action domain.Action, has
}
if !action.Paused && hash != "" {
err = checkTrackerStatus(*qbt, hash)
err = checkTrackerStatus(qbt, hash)
if err != nil {
log.Error().Stack().Err(err).Msgf("could not get tracker status for torrent: %v", hash)
log.Error().Stack().Err(err).Msgf("could not reannounce torrent: %v", hash)
return err
}
}
@ -130,17 +130,15 @@ func (s *service) qbittorrentCheckRulesCanDownload(action domain.Action) (bool,
return true, qbt, nil
}
func checkTrackerStatus(qb qbittorrent.Client, hash string) error {
func checkTrackerStatus(qb *qbittorrent.Client, hash string) error {
announceOK := false
attempts := 0
// initial sleep to give tracker a head start
time.Sleep(4 * time.Second)
time.Sleep(6 * time.Second)
for attempts < ReannounceMaxAttempts {
if attempts > 0 {
log.Debug().Msgf("qBittorrent - run re-announce %v attempt: %v", hash, attempts)
}
log.Debug().Msgf("qBittorrent - run re-announce %v attempt: %v", hash, attempts)
trackers, err := qb.GetTorrentTrackers(hash)
if err != nil {
@ -148,12 +146,12 @@ func checkTrackerStatus(qb qbittorrent.Client, hash string) error {
return err
}
log.Trace().Msgf("qBittorrent - run re-announce %v attempt: %v trackers (%+v)", hash, attempts, trackers)
// check if status not working or something else
working := findTrackerStatus(trackers, qbittorrent.TrackerStatusOK)
working := findTrackerStatus(trackers)
if working {
if attempts > 0 {
log.Debug().Msgf("qBittorrent - re-announce for %v OK", hash)
}
log.Debug().Msgf("qBittorrent - re-announce for %v OK", hash)
announceOK = true
@ -168,12 +166,10 @@ func checkTrackerStatus(qb qbittorrent.Client, hash string) error {
return err
}
attempts++
// add delay for next run
time.Sleep(ReannounceInterval * time.Millisecond)
continue
attempts++
}
// add extra delay before delete
@ -200,14 +196,16 @@ func checkTrackerStatus(qb qbittorrent.Client, hash string) error {
// 2 Tracker has been contacted and is working
// 3 Tracker is updating
// 4 Tracker has been contacted, but it is not working (or doesn't send proper replies)
func findTrackerStatus(slice []qbittorrent.TorrentTracker, status qbittorrent.TrackerStatus) bool {
func findTrackerStatus(slice []qbittorrent.TorrentTracker) bool {
for _, item := range slice {
// if updating skip and give some more time
if item.Status == qbittorrent.TrackerStatusUpdating {
return false
} else if item.Status == status {
if item.Status == qbittorrent.TrackerStatusDisabled {
continue
}
if item.Status == qbittorrent.TrackerStatusOK {
return true
}
}
return false
}