github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-02-02 16:28:39 +08:00
Code Issues 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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats 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()