mirror of
https://github.com/labstack/echo.git
synced 2024-12-22 20:06:21 +02:00
2483d2a140
Signed-off-by: Vishal Rana <vr@labstack.com>
85 lines
1.5 KiB
Go
85 lines
1.5 KiB
Go
package echo
|
|
|
|
import (
|
|
"bufio"
|
|
"net"
|
|
"net/http"
|
|
)
|
|
|
|
type (
|
|
Response struct {
|
|
writer http.ResponseWriter
|
|
status int
|
|
size int64
|
|
committed bool
|
|
echo *Echo
|
|
}
|
|
)
|
|
|
|
func NewResponse(w http.ResponseWriter, e *Echo) *Response {
|
|
return &Response{writer: w, echo: e}
|
|
}
|
|
|
|
func (r *Response) SetWriter(w http.ResponseWriter) {
|
|
r.writer = w
|
|
}
|
|
|
|
func (r *Response) Header() http.Header {
|
|
return r.writer.Header()
|
|
}
|
|
|
|
func (r *Response) Writer() http.ResponseWriter {
|
|
return r.writer
|
|
}
|
|
|
|
func (r *Response) WriteHeader(code int) {
|
|
if r.committed {
|
|
r.echo.Logger().Warn("response already committed")
|
|
return
|
|
}
|
|
r.status = code
|
|
r.writer.WriteHeader(code)
|
|
r.committed = true
|
|
}
|
|
|
|
func (r *Response) Write(b []byte) (n int, err error) {
|
|
n, err = r.writer.Write(b)
|
|
r.size += int64(n)
|
|
return n, err
|
|
}
|
|
|
|
// Flush wraps response writer's Flush function.
|
|
func (r *Response) Flush() {
|
|
r.writer.(http.Flusher).Flush()
|
|
}
|
|
|
|
// Hijack wraps response writer's Hijack function.
|
|
func (r *Response) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
|
return r.writer.(http.Hijacker).Hijack()
|
|
}
|
|
|
|
// CloseNotify wraps response writer's CloseNotify function.
|
|
func (r *Response) CloseNotify() <-chan bool {
|
|
return r.writer.(http.CloseNotifier).CloseNotify()
|
|
}
|
|
|
|
func (r *Response) Status() int {
|
|
return r.status
|
|
}
|
|
|
|
func (r *Response) Size() int64 {
|
|
return r.size
|
|
}
|
|
|
|
func (r *Response) Committed() bool {
|
|
return r.committed
|
|
}
|
|
|
|
func (r *Response) reset(w http.ResponseWriter, e *Echo) {
|
|
r.writer = w
|
|
r.size = 0
|
|
r.status = http.StatusOK
|
|
r.committed = false
|
|
r.echo = e
|
|
}
|