let middleware.responseRecorder be a http.Hijacker

Signed-off-by: Tw <tw19881113@gmail.com>
This commit is contained in:
Tw 2015-06-12 11:42:28 +08:00
parent 01aca02edc
commit e79a88856a

View File

@ -1,6 +1,9 @@
package middleware
import (
"bufio"
"errors"
"net"
"net/http"
"time"
)
@ -48,3 +51,12 @@ func (r *responseRecorder) Write(buf []byte) (int, error) {
}
return n, err
}
// Hijacker is a wrapper of http.Hijacker underearth if any,
// otherwise it just returns an error.
func (r *responseRecorder) Hijack() (net.Conn, *bufio.ReadWriter, error) {
if hj, ok := r.ResponseWriter.(http.Hijacker); ok {
return hj.Hijack()
}
return nil, nil, errors.New("I'm not a Hijacker")
}