artin/exportarr
1
0
Fork 0
mirror of https://github.com/onedr0p/exportarr.git synced 2026-02-06 10:57:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(add): Request gzip compression in API calls (#99)

Browse Source
This commit is contained in:
Russell Troxel 2023-03-03 20:00:12 -08:00 committed by GitHub
parent 71db273257
commit a23d3f1c09
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
internal/client/client.go
View File

@ -1,11 +1,13 @@
package client
import (
"compress/gzip"
"crypto/tls"
"encoding/base64"
"encoding/json"
"errors"
"fmt"
"io"
"net/http"
"github.com/onedr0p/exportarr/internal/model"
@ -65,17 +67,18 @@ func (c *Client) DoRequest(endpoint string, target interface{}) error {
http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: c.config.Bool("disable-ssl-verify")}
req, err := http.NewRequest("GET", url, nil)
if err != nil {
log.Fatalf("An error has occurred when creating HTTP request %v", err)
return err
}
if c.config.String("basic-auth-username") != "" && c.config.String("basic-auth-password") != "" {
req.Header.Add("Authorization", fmt.Sprintf("Basic %s",
base64.StdEncoding.EncodeToString([]byte(c.config.String("basic-auth-username")+":"+c.config.String("basic-auth-password"))),
))
}
req.Header.Add("X-Api-Key", apiKey)
if err != nil {
log.Fatalf("An error has occurred when creating HTTP request %v", err)
return err
}
req.Header.Add("Accept-Encoding", "gzip")
resp, err := c.httpClient.Do(req)
if err != nil {
@ -90,7 +93,19 @@ func (c *Client) DoRequest(endpoint string, target interface{}) error {
log.Fatal(errMsg)
return errors.New(errMsg)
}
defer resp.Body.Close()
return json.NewDecoder(resp.Body).Decode(target)
var bodyReader io.ReadCloser
switch resp.Header.Get("Content-Encoding") {
case "gzip":
bodyReader, err = gzip.NewReader(resp.Body)
if err != nil {
log.Fatalf("An error has occurred reading gzipped statistics %v", err)
return err
}
defer bodyReader.Close()
default:
bodyReader = resp.Body
}
return json.NewDecoder(bodyReader).Decode(target)
}