Commit d7051e98 authored by Matt Holt's avatar Matt Holt

Merge pull request #134 from tw4452852/hijack

middleware: let middleware.responseRecorder be a http.Hijacker
parents f36d9bfa e79a8885
package middleware package middleware
import ( import (
"bufio"
"errors"
"net"
"net/http" "net/http"
"time" "time"
) )
...@@ -48,3 +51,12 @@ func (r *responseRecorder) Write(buf []byte) (int, error) { ...@@ -48,3 +51,12 @@ func (r *responseRecorder) Write(buf []byte) (int, error) {
} }
return n, err 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")
}
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