zlei6/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2026-05-13 09:50:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: coding style (#4054)

Browse Source
This commit is contained in:
Kevin Wan
2024-04-09 17:19:47 +08:00
committed by GitHub
parent 6f763f71f9
commit 5fbe8ff5c4
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
rest/handler/timeouthandler.go
View File

@@ -191,6 +191,15 @@ func (tw *timeoutWriter) Write(p []byte) (int, error) {
return tw.wbuf.Write(p) return tw.wbuf.Write(p)
} }
func (tw *timeoutWriter) WriteHeader(code int) {
tw.mu.Lock()
defer tw.mu.Unlock()
if !tw.wroteHeader {
tw.writeHeaderLocked(code)
}
}
func (tw *timeoutWriter) writeHeaderLocked(code int) { func (tw *timeoutWriter) writeHeaderLocked(code int) {
checkWriteHeaderCode(code) checkWriteHeaderCode(code)
@@ -209,15 +218,6 @@ func (tw *timeoutWriter) writeHeaderLocked(code int) {
} }
} }
func (tw *timeoutWriter) WriteHeader(code int) {
tw.mu.Lock()
defer tw.mu.Unlock()
if !tw.wroteHeader {
tw.writeHeaderLocked(code)
}
}
func checkWriteHeaderCode(code int) { func checkWriteHeaderCode(code int) {
if code < 100 || code > 599 { if code < 100 || code > 599 {
panic(fmt.Sprintf("invalid WriteHeader code %v", code)) panic(fmt.Sprintf("invalid WriteHeader code %v", code))