hpz937/zurg
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Couple of fixes

Browse Source
This commit is contained in:
Ben Sarmiento
2023-11-10 21:25:53 +01:00
parent b97f859a32
commit ab31114f2e
7 changed files with 131 additions and 152 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
pkg/http/client.go
View File

@@ -29,10 +29,10 @@ func (r *HTTPClient) Do(req *http.Request) (*http.Response, error) {
return resp, err
}
func NewHTTPClient(token string, maxRetries int, timeout time.Duration) *HTTPClient {
func NewHTTPClient(token string, maxRetries int) *HTTPClient {
return &HTTPClient{
BearerToken: token,
Client: &http.Client{Timeout: timeout},
Client: &http.Client{},
MaxRetries: maxRetries,
Backoff: func(attempt int) time.Duration {
return time.Duration(attempt) * time.Second

5
pkg/realdebrid/api.go
View File

@@ -9,7 +9,6 @@ import (
"net/url"
"strconv"
"strings"
"time"
zurghttp "github.com/debridmediamanager.com/zurg/pkg/http"
"go.uber.org/zap"
@@ -22,9 +21,7 @@ type RealDebrid struct {
func NewRealDebrid(accessToken string, log *zap.SugaredLogger) *RealDebrid {
maxRetries := 10
timeout := 10 * time.Second
client := zurghttp.NewHTTPClient(accessToken, maxRetries, timeout)
log.Debugf("Created an HTTP client with %d max retries and %s timeout", maxRetries, timeout)
client := zurghttp.NewHTTPClient(accessToken, maxRetries)
return &RealDebrid{
log: log,
client: client,