oiupoyt/discordo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

refactor(http): use gzhttp for gzip and zstd

Browse source
This commit is contained in:
ayn2op 2025-10-21 15:02:20 -04:00
parent 81e4924c22
commit d336b078fc
1 changed files with 6 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
internal/http/transport.go
View file

@ -5,8 +5,7 @@ import (
"net/http"
"github.com/andybalholm/brotli"
"github.com/klauspost/compress/gzip"
"github.com/klauspost/compress/zstd"
"github.com/klauspost/compress/gzhttp"
)
type Transport struct {
@ -15,7 +14,7 @@ type Transport struct {
func NewTransport() *Transport {
return &Transport{
base: http.DefaultTransport,
base: gzhttp.Transport(http.DefaultTransport, gzhttp.TransportAlwaysDecompress(true)),
}
}
@ -25,27 +24,12 @@ func (t *Transport) RoundTrip(req *http.Request) (*http.Response, error) {
return nil, err
}
switch resp.Header.Get("Content-Encoding") {
case "gzip":
resp.Body, err = gzip.NewReader(resp.Body)
if err != nil {
resp.Body.Close()
return nil, err
}
resp.Header.Del("Content-Encoding")
case "zstd":
decoder, err := zstd.NewReader(resp.Body)
if err != nil {
resp.Body.Close()
return nil, err
}
resp.Body = io.NopCloser(decoder)
resp.Header.Del("Content-Encoding")
case "br":
if resp.Header.Get("Content-Encoding") == "br" {
resp.Body = io.NopCloser(brotli.NewReader(resp.Body))
resp.Header.Del("Content-Encoding")
resp.Header.Del("Content-Length")
resp.ContentLength = -1
resp.Uncompressed = true
}
return resp, nil