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

chore: refactor rest/timeouthandler (#1415)

Browse Source
This commit is contained in:
Kevin Wan
2022-01-05 11:17:10 +08:00
committed by GitHub
parent a91c3907a8
commit ba6a7c9dc8
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
rest/handler/timeouthandler.go
View File

@@ -118,6 +118,9 @@ type timeoutWriter struct {
var _ http.Pusher = (*timeoutWriter)(nil) var _ http.Pusher = (*timeoutWriter)(nil)
// Header returns the underline temporary http.Header.
func (tw *timeoutWriter) Header() http.Header { return tw.h }
// Push implements the Pusher interface. // Push implements the Pusher interface.
func (tw *timeoutWriter) Push(target string, opts *http.PushOptions) error { func (tw *timeoutWriter) Push(target string, opts *http.PushOptions) error {
if pusher, ok := tw.w.(http.Pusher); ok { if pusher, ok := tw.w.(http.Pusher); ok {
@@ -126,8 +129,8 @@ func (tw *timeoutWriter) Push(target string, opts *http.PushOptions) error {
return http.ErrNotSupported return http.ErrNotSupported
} }
func (tw *timeoutWriter) Header() http.Header { return tw.h } // Write writes the data to the connection as part of an HTTP reply.
// Timeout and multiple header written are guarded.
func (tw *timeoutWriter) Write(p []byte) (int, error) { func (tw *timeoutWriter) Write(p []byte) (int, error) {
tw.mu.Lock() tw.mu.Lock()
defer tw.mu.Unlock() defer tw.mu.Unlock()