gin/response_writer.go

90 lines
1.7 KiB
Go
Raw Normal View History

2014-08-29 17:49:50 +00:00
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
// Use of this source code is governed by a MIT style
// license that can be found in the LICENSE file.
2014-07-03 22:01:28 +00:00
package gin
import (
2014-08-25 11:58:43 +00:00
"bufio"
"errors"
2014-08-18 03:24:48 +00:00
"log"
2014-08-25 11:58:43 +00:00
"net"
2014-07-03 22:01:28 +00:00
"net/http"
)
type (
ResponseWriter interface {
http.ResponseWriter
http.Hijacker
http.Flusher
http.CloseNotifier
2014-07-03 22:01:28 +00:00
Status() int
Written() bool
2014-08-18 03:24:48 +00:00
WriteHeaderNow()
2014-07-03 22:01:28 +00:00
}
responseWriter struct {
http.ResponseWriter
status int
written bool
}
)
func (w *responseWriter) reset(writer http.ResponseWriter) {
w.ResponseWriter = writer
w.status = 200
2014-07-03 22:01:28 +00:00
w.written = false
}
2014-08-18 03:24:48 +00:00
func (w *responseWriter) WriteHeader(code int) {
2014-08-24 02:35:11 +00:00
if code > 0 {
2014-08-18 03:24:48 +00:00
w.status = code
if w.written {
log.Println("[GIN] WARNING. Headers were already written!")
}
}
2014-07-03 22:01:28 +00:00
}
2014-08-18 03:24:48 +00:00
func (w *responseWriter) WriteHeaderNow() {
if !w.written {
w.written = true
w.ResponseWriter.WriteHeader(w.status)
}
}
func (w *responseWriter) Write(data []byte) (n int, err error) {
w.WriteHeaderNow()
2014-08-18 03:24:48 +00:00
return w.ResponseWriter.Write(data)
2014-07-03 22:01:28 +00:00
}
func (w *responseWriter) Status() int {
return w.status
}
func (w *responseWriter) Written() bool {
return w.written
}
2014-08-25 11:58:43 +00:00
// Implements the http.Hijacker interface
2014-08-25 11:58:43 +00:00
func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
hijacker, ok := w.ResponseWriter.(http.Hijacker)
if !ok {
return nil, nil, errors.New("the ResponseWriter doesn't support the Hijacker interface")
}
return hijacker.Hijack()
}
// Implements the http.CloseNotify interface
func (w *responseWriter) CloseNotify() <-chan bool {
return w.ResponseWriter.(http.CloseNotifier).CloseNotify()
}
// Implements the http.Flush interface
func (w *responseWriter) Flush() {
flusher, ok := w.ResponseWriter.(http.Flusher)
if ok {
flusher.Flush()
}
}