astra
/
go-mastodon
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Implement gzipped responses

pull/173/head
Alexander Ivanov 2023-01-01 21:46:06 +08:00 committed by GitHub
parent 9faaa4f0dc
commit 6c76b24594
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mastodon.go
View File

@ -3,6 +3,7 @@ package mastodon
import ( import (
"context" "context"
"compress/gzip"
"encoding/json" "encoding/json"
"errors" "errors"
"fmt" "fmt"
@ -76,6 +77,7 @@ func (c *Client) doAPI(ctx context.Context, method string, uri string, params in
} }
} }
req = req.WithContext(ctx) req = req.WithContext(ctx)
req.Header.Set("Accept-Encoding", "gzip")
req.Header.Set("Authorization", "Bearer "+c.Config.AccessToken) req.Header.Set("Authorization", "Bearer "+c.Config.AccessToken)
if params != nil { if params != nil {
req.Header.Set("Content-Type", ct) req.Header.Set("Content-Type", ct)
@ -125,7 +127,16 @@ func (c *Client) doAPI(ctx context.Context, method string, uri string, params in
*pg = *pg2 *pg = *pg2
} }
} }
return json.NewDecoder(resp.Body).Decode(&res)
var reader io.ReadCloser
switch resp.Header.Get("Content-Encoding") {
case "gzip":
reader, err = gzip.NewReader(resp.Body)
defer reader.Close()
default:
reader = resp.Body
}
return json.NewDecoder(reader).Decode(&res)
} }
// NewClient returns a new mastodon API client. // NewClient returns a new mastodon API client.