chore(binding): support Content-Encoding

Change-Id: I8ff72eb62108bcd5a7418e434d87cbe03a8fbf37

Author: andeya

binding/body.go

@@ -2,6 +2,9 @@ package binding
 
 import (
 	"bytes"
+	"compress/flate"
+	"compress/gzip"
+	"compress/zlib"
 	jsonpkg "encoding/json"
 	"errors"
 	"io"
@@ -60,12 +63,30 @@ func GetBody(r *http.Request) (*Body, error) {
 		body.Reset()
 		return body, nil
 	default:
-		var buf bytes.Buffer
-		_, err := io.Copy(&buf, r.Body)
-		r.Body.Close()
+		var err error
+		var closeBody = r.Body.Close
+		switch r.Header.Get("Content-Encoding") {
+		case "gzip":
+			var gzipReader *gzip.Reader
+			gzipReader, err = gzip.NewReader(r.Body)
+			if err == nil {
+				r.Body = gzipReader
+			}
+		case "deflate":
+			r.Body = flate.NewReader(r.Body)
+		case "zlib":
+			var readCloser io.ReadCloser
+			readCloser, err = zlib.NewReader(r.Body)
+			if err == nil {
+				r.Body = readCloser
+			}
+		}
 		if err != nil {
 			return nil, err
 		}
+		var buf bytes.Buffer
+		_, _ = io.Copy(&buf, r.Body)
+		_ = closeBody()
 		_body := &Body{
 			Buffer:    &buf,
 			bodyBytes: buf.Bytes(),

binding/body_test.go

@@ -2,6 +2,7 @@ package binding
 
 import (
 	"bytes"
+	"compress/gzip"
 	"io/ioutil"
 	"net/http"
 	"testing"
@@ -10,11 +11,25 @@ import (
 )
 
 func TestBody(t *testing.T) {
+	const USE_GZIP = true
 	var buf bytes.Buffer
-	buf.WriteString("abc")
+	if USE_GZIP {
+		w := gzip.NewWriter(&buf)
+		_, err := w.Write([]byte("abc"))
+		assert.NoError(t, err)
+		err = w.Flush()
+		assert.NoError(t, err)
+	} else {
+		buf.WriteString("abc")
+	}
 	req := &http.Request{
 		Body: ioutil.NopCloser(&buf),
 	}
+	if USE_GZIP {
+		req.Header = map[string][]string{
+			"Content-Encoding": []string{"gzip"},
+		}
+	}
 	body, err := GetBody(req)
 	assert.NoError(t, err)
 	b, err := ioutil.ReadAll(body)