Merge pull request #43 from cpunion/hijacker

http: nocache, tracer implements http.Hijacker
qiniu · Aug 27, 2023 · fa669d7 · fa669d7
2 parents c0ab963 + 56544ff
commit fa669d7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/http/nocache/nocache.go b/http/nocache/nocache.go
@@ -1,6 +1,8 @@
 package nocache
 
 import (
+	"bufio"
+	"net"
 	"net/http"
 )
 
@@ -19,3 +21,7 @@ func New(h http.Handler) http.Handler {
 		h.ServeHTTP(&respWriter{w}, r)
 	})
 }
+
+func (p *respWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	return p.ResponseWriter.(http.Hijacker).Hijack()
+}
diff --git a/http/tracer/tracer.go b/http/tracer/tracer.go
@@ -1,7 +1,9 @@
 package tracer
 
 import (
+	"bufio"
 	"log"
+	"net"
 	"net/http"
 	"time"
 )
@@ -27,6 +29,10 @@ func (p *teeResponseWriter) WriteHeader(statusCode int) {
 	p.b.WriteHeader(statusCode)
 }
 
+func (p *teeResponseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
+	return p.a.(http.Hijacker).Hijack()
+}
+
 func Tee(a, b http.ResponseWriter) http.ResponseWriter {
 	return &teeResponseWriter{a, b}
 }