zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update timeouthandler.go

Browse Source
This commit is contained in:
yangjinheng
2023-02-27 14:06:10 +08:00
committed by kevin
parent e964e530e1
commit 703fb8d970
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rest/handler/timeouthandler.go
View File

@@ -125,6 +125,14 @@ type timeoutWriter struct {
var _ http.Pusher = (*timeoutWriter)(nil) var _ http.Pusher = (*timeoutWriter)(nil)
func (tw *timeoutWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
if hijacked, ok := tw.w.(http.Hijacker); ok {
return hijacked.Hijack()
}
return nil, nil, errors.New("server doesn't support hijacking")
}
// Header returns the underline temporary http.Header. // Header returns the underline temporary http.Header.
func (tw *timeoutWriter) Header() http.Header { return tw.h } func (tw *timeoutWriter) Header() http.Header { return tw.h }