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

feat: qbit add retry and extend timeout (#155)

Browse source
This commit is contained in:
Ludvig Lundgren 2022-02-28 18:38:18 +01:00 • committed by GitHub
parent c8356d667a
commit 1b0d52da85
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 62 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

67
pkg/qbittorrent/client.go
View file

@ -16,6 +16,15 @@ import (
"golang.org/x/net/publicsuffix"
)
var (
backoffSchedule = []time.Duration{
5 * time.Second,
10 * time.Second,
20 * time.Second,
}
timeout = 20 * time.Second
)
type Client struct {
Name string
settings Settings
@ -39,7 +48,7 @@ func NewClient(s Settings) *Client {
log.Error().Err(err).Msg("new client cookie error")
}
httpClient := &http.Client{
Timeout: time.Second * 10,
Timeout: timeout,
Jar: jar,
}
@ -59,13 +68,29 @@ func NewClient(s Settings) *Client {
func (c *Client) get(endpoint string, opts map[string]string) (*http.Response, error) {
reqUrl := fmt.Sprintf("%v://%v:%v/api/v2/%v", c.settings.protocol, c.settings.Hostname, c.settings.Port, endpoint)
var err error
var resp *http.Response
req, err := http.NewRequest("GET", reqUrl, nil)
if err != nil {
log.Error().Err(err).Msgf("GET: error %v", reqUrl)
return nil, err
}
resp, err := c.http.Do(req)
// try request and if fail run 3 retries
for i, backoff := range backoffSchedule {
resp, err = c.http.Do(req)
// request ok, lets break out of the loop
if err == nil {
break
}
log.Debug().Msgf("qbit GET failed: retrying attempt %d - %v", i, reqUrl)
time.Sleep(backoff)
}
if err != nil {
log.Error().Err(err).Msgf("GET: do %v", reqUrl)
return nil, err
@ -83,6 +108,9 @@ func (c *Client) post(endpoint string, opts map[string]string) (*http.Response,
}
}
var err error
var resp *http.Response
reqUrl := fmt.Sprintf("%v://%v:%v/api/v2/%v", c.settings.protocol, c.settings.Hostname, c.settings.Port, endpoint)
req, err := http.NewRequest("POST", reqUrl, strings.NewReader(form.Encode()))
if err != nil {
@ -93,7 +121,20 @@ func (c *Client) post(endpoint string, opts map[string]string) (*http.Response,
// add the content-type so qbittorrent knows what to expect
req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
resp, err := c.http.Do(req)
// try request and if fail run 3 retries
for i, backoff := range backoffSchedule {
resp, err = c.http.Do(req)
// request ok, lets break out of the loop
if err == nil {
break
}
log.Debug().Msgf("qbit POST failed: retrying attempt %d - %v", i, reqUrl)
time.Sleep(backoff)
}
if err != nil {
log.Error().Err(err).Msgf("POST: do %v", reqUrl)
return nil, err
@ -103,6 +144,9 @@ func (c *Client) post(endpoint string, opts map[string]string) (*http.Response,
}
func (c *Client) postFile(endpoint string, fileName string, opts map[string]string) (*http.Response, error) {
var err error
var resp *http.Response
file, err := os.Open(fileName)
if err != nil {
log.Error().Err(err).Msgf("POST file: opening file %v", fileName)
@ -161,13 +205,26 @@ func (c *Client) postFile(endpoint string, fileName string, opts map[string]stri
// Set correct content type
req.Header.Set("Content-Type", multiPartWriter.FormDataContentType())
res, err := c.http.Do(req)
// try request and if fail run 3 retries
for i, backoff := range backoffSchedule {
resp, err = c.http.Do(req)
// request ok, lets break out of the loop
if err == nil {
break
}
log.Debug().Msgf("qbit POST file failed: retrying attempt %d - %v", i, reqUrl)
time.Sleep(backoff)
}
if err != nil {
log.Error().Err(err).Msgf("POST file: could not perform request %v", fileName)
return nil, err
}
return res, nil
return resp, nil
}
func (c *Client) setCookies(cookies []*http.Cookie) {