diff options
author | SjonHortensius <SjonHortensius@users.noreply.github.com> | 2019-08-22 21:18:39 +0800 |
---|---|---|
committer | Felix Lange <fjl@twurst.com> | 2019-08-22 21:18:39 +0800 |
commit | 961aa0533f75027bc5bc80f994adb80d14635553 (patch) | |
tree | 8f11e1db25a73c7c56b9eb2bd859a469ef4538b6 /rpc | |
parent | 54b271a86dd748f3b0bcebeaf678dc34e0d6177a (diff) | |
download | go-tangerine-961aa0533f75027bc5bc80f994adb80d14635553.tar go-tangerine-961aa0533f75027bc5bc80f994adb80d14635553.tar.gz go-tangerine-961aa0533f75027bc5bc80f994adb80d14635553.tar.bz2 go-tangerine-961aa0533f75027bc5bc80f994adb80d14635553.tar.lz go-tangerine-961aa0533f75027bc5bc80f994adb80d14635553.tar.xz go-tangerine-961aa0533f75027bc5bc80f994adb80d14635553.tar.zst go-tangerine-961aa0533f75027bc5bc80f994adb80d14635553.zip |
rpc: enable compression on HTTP transport (#19997)
This change adds support for gzip encoding on HTTP responses.
Gzip encoding is used when the client sets the 'accept-encoding: gzip' header.
Original change by @brianosaurus, with fixes from @SjonHortensius.
Diffstat (limited to 'rpc')
-rw-r--r-- | rpc/gzip.go | 66 | ||||
-rw-r--r-- | rpc/http.go | 1 |
2 files changed, 67 insertions, 0 deletions
diff --git a/rpc/gzip.go b/rpc/gzip.go new file mode 100644 index 000000000..a14fd09d5 --- /dev/null +++ b/rpc/gzip.go @@ -0,0 +1,66 @@ +// Copyright 2019 The go-ethereum Authors +// This file is part of the go-ethereum library. +// +// The go-ethereum library is free software: you can redistribute it and/or modify +// it under the terms of the GNU Lesser General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// The go-ethereum library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public License +// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>. + +package rpc + +import ( + "compress/gzip" + "io" + "io/ioutil" + "net/http" + "strings" + "sync" +) + +var gzPool = sync.Pool{ + New: func() interface{} { + w := gzip.NewWriter(ioutil.Discard) + return w + }, +} + +type gzipResponseWriter struct { + io.Writer + http.ResponseWriter +} + +func (w *gzipResponseWriter) WriteHeader(status int) { + w.Header().Del("Content-Length") + w.ResponseWriter.WriteHeader(status) +} + +func (w *gzipResponseWriter) Write(b []byte) (int, error) { + return w.Writer.Write(b) +} + +func newGzipHandler(next http.Handler) http.Handler { + return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { + if !strings.Contains(r.Header.Get("Accept-Encoding"), "gzip") { + next.ServeHTTP(w, r) + return + } + + w.Header().Set("Content-Encoding", "gzip") + + gz := gzPool.Get().(*gzip.Writer) + defer gzPool.Put(gz) + + gz.Reset(w) + defer gz.Close() + + next.ServeHTTP(&gzipResponseWriter{ResponseWriter: w, Writer: gz}, r) + }) +} diff --git a/rpc/http.go b/rpc/http.go index e8f2cfda7..2c0cb5edb 100644 --- a/rpc/http.go +++ b/rpc/http.go @@ -216,6 +216,7 @@ func NewHTTPServer(cors []string, vhosts []string, timeouts HTTPTimeouts, srv ht // Wrap the CORS-handler within a host-handler handler := newCorsHandler(srv, cors) handler = newVHostHandler(vhosts, handler) + handler = newGzipHandler(handler) // Make sure timeout values are meaningful if timeouts.ReadTimeout < time.Second { |