Commit a44d59f1 authored by Abiola Ibrahim's avatar Abiola Ibrahim

Use ioutil.Discard instead for unneeded bytes.

parent 23631cfa
...@@ -3,10 +3,10 @@ ...@@ -3,10 +3,10 @@
package gzip package gzip
import ( import (
"bytes"
"compress/gzip" "compress/gzip"
"fmt" "fmt"
"io" "io"
"io/ioutil"
"net/http" "net/http"
"strings" "strings"
...@@ -49,12 +49,9 @@ outer: ...@@ -49,12 +49,9 @@ outer:
r.Header.Del("Accept-Encoding") r.Header.Del("Accept-Encoding")
// gzipWriter modifies underlying writer at init, // gzipWriter modifies underlying writer at init,
// use a buffer instead to leave ResponseWriter in // use a discard writer instead to leave ResponseWriter in
// original form. // original form.
var buf = &bytes.Buffer{} gzipWriter, err := newWriter(c, ioutil.Discard)
defer buf.Reset()
gzipWriter, err := newWriter(c, buf)
if err != nil { if err != nil {
// should not happen // should not happen
return http.StatusInternalServerError, err return http.StatusInternalServerError, err
...@@ -65,7 +62,7 @@ outer: ...@@ -65,7 +62,7 @@ outer:
var rw http.ResponseWriter var rw http.ResponseWriter
// if no response filter is used // if no response filter is used
if len(c.ResponseFilters) == 0 { if len(c.ResponseFilters) == 0 {
// replace buffer with ResponseWriter // replace discard writer with ResponseWriter
gzipWriter.Reset(w) gzipWriter.Reset(w)
rw = gz rw = gz
} else { } else {
......
...@@ -52,7 +52,7 @@ func (r *ResponseFilterWriter) WriteHeader(code int) { ...@@ -52,7 +52,7 @@ func (r *ResponseFilterWriter) WriteHeader(code int) {
} }
if r.shouldCompress { if r.shouldCompress {
// replace buffer with ResponseWriter // replace discard writer with ResponseWriter
if gzWriter, ok := r.gzipResponseWriter.Writer.(*gzip.Writer); ok { if gzWriter, ok := r.gzipResponseWriter.Writer.(*gzip.Writer); ok {
gzWriter.Reset(r.ResponseWriter) gzWriter.Reset(r.ResponseWriter)
} }
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment