mirror of
https://github.com/labstack/echo.git
synced 2024-12-24 20:14:31 +02:00
Dropped Response#Writer/SetWriter functions
Signed-off-by: Vishal Rana <vr@labstack.com>
This commit is contained in:
parent
d9a6052c67
commit
869cdcd19a
@ -67,23 +67,24 @@ func GzipWithConfig(config GzipConfig) echo.MiddlewareFunc {
|
|||||||
res := c.Response()
|
res := c.Response()
|
||||||
res.Header().Add(echo.HeaderVary, echo.HeaderAcceptEncoding)
|
res.Header().Add(echo.HeaderVary, echo.HeaderAcceptEncoding)
|
||||||
if strings.Contains(c.Request().Header.Get(echo.HeaderAcceptEncoding), gzipScheme) {
|
if strings.Contains(c.Request().Header.Get(echo.HeaderAcceptEncoding), gzipScheme) {
|
||||||
rw := res.Writer()
|
rw := res.Writer
|
||||||
w, err := gzip.NewWriterLevel(rw, config.Level)
|
w, err := gzip.NewWriterLevel(rw, config.Level)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
defer func() {
|
defer func() {
|
||||||
|
|
||||||
if res.Size == 0 {
|
if res.Size == 0 {
|
||||||
// We have to reset response to it's pristine state when
|
// We have to reset response to it's pristine state when
|
||||||
// nothing is written to body or error is returned.
|
// nothing is written to body or error is returned.
|
||||||
// See issue #424, #407.
|
// See issue #424, #407.
|
||||||
res.SetWriter(rw)
|
res.Writer = rw
|
||||||
w.Reset(ioutil.Discard)
|
w.Reset(ioutil.Discard)
|
||||||
}
|
}
|
||||||
w.Close()
|
w.Close()
|
||||||
}()
|
}()
|
||||||
grw := &gzipResponseWriter{Writer: w, ResponseWriter: rw}
|
grw := &gzipResponseWriter{Writer: w, ResponseWriter: rw}
|
||||||
res.SetWriter(grw)
|
res.Writer = grw
|
||||||
}
|
}
|
||||||
return next(c)
|
return next(c)
|
||||||
}
|
}
|
||||||
|
28
response.go
28
response.go
@ -11,7 +11,7 @@ type (
|
|||||||
// by an HTTP handler to construct an HTTP response.
|
// by an HTTP handler to construct an HTTP response.
|
||||||
// See: https://golang.org/pkg/net/http/#ResponseWriter
|
// See: https://golang.org/pkg/net/http/#ResponseWriter
|
||||||
Response struct {
|
Response struct {
|
||||||
writer http.ResponseWriter
|
Writer http.ResponseWriter
|
||||||
Status int
|
Status int
|
||||||
Size int64
|
Size int64
|
||||||
Committed bool
|
Committed bool
|
||||||
@ -21,17 +21,7 @@ type (
|
|||||||
|
|
||||||
// NewResponse creates a new instance of Response.
|
// NewResponse creates a new instance of Response.
|
||||||
func NewResponse(w http.ResponseWriter, e *Echo) (r *Response) {
|
func NewResponse(w http.ResponseWriter, e *Echo) (r *Response) {
|
||||||
return &Response{writer: w, echo: e}
|
return &Response{Writer: w, echo: e}
|
||||||
}
|
|
||||||
|
|
||||||
// SetWriter sets the http.ResponseWriter instance for this Response.
|
|
||||||
func (r *Response) SetWriter(w http.ResponseWriter) {
|
|
||||||
r.writer = w
|
|
||||||
}
|
|
||||||
|
|
||||||
// Writer returns the http.ResponseWriter instance for this Response.
|
|
||||||
func (r *Response) Writer() http.ResponseWriter {
|
|
||||||
return r.writer
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Header returns the header map for the writer that will be sent by
|
// Header returns the header map for the writer that will be sent by
|
||||||
@ -41,7 +31,7 @@ func (r *Response) Writer() http.ResponseWriter {
|
|||||||
// To suppress implicit response headers, set their value to nil.
|
// To suppress implicit response headers, set their value to nil.
|
||||||
// Example: https://golang.org/pkg/net/http/#example_ResponseWriter_trailers
|
// Example: https://golang.org/pkg/net/http/#example_ResponseWriter_trailers
|
||||||
func (r *Response) Header() http.Header {
|
func (r *Response) Header() http.Header {
|
||||||
return r.writer.Header()
|
return r.Writer.Header()
|
||||||
}
|
}
|
||||||
|
|
||||||
// WriteHeader sends an HTTP response header with status code. If WriteHeader is
|
// WriteHeader sends an HTTP response header with status code. If WriteHeader is
|
||||||
@ -54,7 +44,7 @@ func (r *Response) WriteHeader(code int) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
r.Status = code
|
r.Status = code
|
||||||
r.writer.WriteHeader(code)
|
r.Writer.WriteHeader(code)
|
||||||
r.Committed = true
|
r.Committed = true
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -63,7 +53,7 @@ func (r *Response) Write(b []byte) (n int, err error) {
|
|||||||
if !r.Committed {
|
if !r.Committed {
|
||||||
r.WriteHeader(http.StatusOK)
|
r.WriteHeader(http.StatusOK)
|
||||||
}
|
}
|
||||||
n, err = r.writer.Write(b)
|
n, err = r.Writer.Write(b)
|
||||||
r.Size += int64(n)
|
r.Size += int64(n)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@ -72,14 +62,14 @@ func (r *Response) Write(b []byte) (n int, err error) {
|
|||||||
// buffered data to the client.
|
// buffered data to the client.
|
||||||
// See [http.Flusher](https://golang.org/pkg/net/http/#Flusher)
|
// See [http.Flusher](https://golang.org/pkg/net/http/#Flusher)
|
||||||
func (r *Response) Flush() {
|
func (r *Response) Flush() {
|
||||||
r.writer.(http.Flusher).Flush()
|
r.Writer.(http.Flusher).Flush()
|
||||||
}
|
}
|
||||||
|
|
||||||
// Hijack implements the http.Hijacker interface to allow an HTTP handler to
|
// Hijack implements the http.Hijacker interface to allow an HTTP handler to
|
||||||
// take over the connection.
|
// take over the connection.
|
||||||
// See [http.Hijacker](https://golang.org/pkg/net/http/#Hijacker)
|
// See [http.Hijacker](https://golang.org/pkg/net/http/#Hijacker)
|
||||||
func (r *Response) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
func (r *Response) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
||||||
return r.writer.(http.Hijacker).Hijack()
|
return r.Writer.(http.Hijacker).Hijack()
|
||||||
}
|
}
|
||||||
|
|
||||||
// CloseNotify implements the http.CloseNotifier interface to allow detecting
|
// CloseNotify implements the http.CloseNotifier interface to allow detecting
|
||||||
@ -88,11 +78,11 @@ func (r *Response) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
|||||||
// client has disconnected before the response is ready.
|
// client has disconnected before the response is ready.
|
||||||
// See [http.CloseNotifier](https://golang.org/pkg/net/http/#CloseNotifier)
|
// See [http.CloseNotifier](https://golang.org/pkg/net/http/#CloseNotifier)
|
||||||
func (r *Response) CloseNotify() <-chan bool {
|
func (r *Response) CloseNotify() <-chan bool {
|
||||||
return r.writer.(http.CloseNotifier).CloseNotify()
|
return r.Writer.(http.CloseNotifier).CloseNotify()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r *Response) reset(w http.ResponseWriter) {
|
func (r *Response) reset(w http.ResponseWriter) {
|
||||||
r.writer = w
|
r.Writer = w
|
||||||
r.Size = 0
|
r.Size = 0
|
||||||
r.Status = http.StatusOK
|
r.Status = http.StatusOK
|
||||||
r.Committed = false
|
r.Committed = false
|
||||||
|
Loading…
Reference in New Issue
Block a user