diff options
Diffstat (limited to 'mastodon/http.go')
-rw-r--r-- | mastodon/http.go | 41 |
1 files changed, 41 insertions, 0 deletions
diff --git a/mastodon/http.go b/mastodon/http.go new file mode 100644 index 0000000..ca4c771 --- /dev/null +++ b/mastodon/http.go @@ -0,0 +1,41 @@ +package mastodon + +import ( + "fmt" + "io" + "net/http" + "time" +) + +type lr struct { + io.ReadCloser + r *http.Request +} + +func (r *lr) Read(p []byte) (n int, err error) { + n, err = r.ReadCloser.Read(p) + // override the generic error returned by the MaxBytesReader + if _, ok := err.(*http.MaxBytesError); ok { + err = fmt.Errorf("%s \"%s\": response body too large", r.r.Method, r.r.URL) + } + return +} + +type transport struct { + t http.RoundTripper +} + +func (t *transport) RoundTrip(r *http.Request) (*http.Response, error) { + resp, err := t.t.RoundTrip(r) + if resp != nil && resp.Body != nil { + resp.Body = &lr{http.MaxBytesReader(nil, resp.Body, 8<<20), r} + } + return resp, err +} + +var httpClient = &http.Client{ + Transport: &transport{ + t: http.DefaultTransport, + }, + Timeout: 30 * time.Second, +} |