/
http.go
39 lines (34 loc) · 970 Bytes
/
http.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
package httpencoding
import (
"bytes"
"io"
"net/http"
)
// AcceptEncodingWriter will handle request Accept-Encoding header, set corresponding Content-Encoding header.
// Caller must call Close() on the returned writer to flush the data.
func AcceptEncodingWriter(w http.ResponseWriter, r *http.Request) io.WriteCloser {
enc, _ := AcceptEncoding("", r.Header.Get("Accept-Encoding"))
if enc != "" {
w.Header().Set("Content-Encoding", enc)
wr, _ := NewWriter(enc, w)
return wr
}
return writeCloser(w)
}
func ContentEncodingReadAll(resp *http.Response) ([]byte, error) {
r, err := ContentEncodingReader(resp)
if err != nil {
return nil, err
}
buf := &bytes.Buffer{}
_, err = io.Copy(buf, r)
return buf.Bytes(), err
}
func ContentEncodingReader(resp *http.Response) (r io.Reader, err error) {
enc := resp.Header.Get("Content-Encoding")
if enc != "" && !resp.Uncompressed {
r, err = NewReader(enc, resp.Body)
return
}
return resp.Body, nil
}