31 lines
588 B
Go
31 lines
588 B
Go
package middleware
|
|
|
|
import (
|
|
"bufio"
|
|
"net"
|
|
"net/http"
|
|
)
|
|
|
|
type HijackWriter struct {
|
|
conn net.Conn
|
|
}
|
|
|
|
func NewHijackWriter(conn net.Conn) http.ResponseWriter {
|
|
return &HijackWriter{conn: conn}
|
|
}
|
|
|
|
func (rw *HijackWriter) Header() http.Header {
|
|
return http.Header{}
|
|
}
|
|
|
|
func (rw *HijackWriter) Write(bytes []byte) (int, error) {
|
|
return rw.conn.Write(bytes)
|
|
}
|
|
|
|
func (rw *HijackWriter) WriteHeader(statusCode int) {
|
|
}
|
|
|
|
func (rw *HijackWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
|
return rw.conn, bufio.NewReadWriter(bufio.NewReader(rw.conn), bufio.NewWriter(rw.conn)), nil
|
|
}
|