From 5fbe8ff5c4ddeb78a504c1c573fc684fa9c37109 Mon Sep 17 00:00:00 2001
From: Kevin Wan <wanjunfeng@gmail.com>
Date: Tue, 9 Apr 2024 17:19:47 +0800
Subject: [PATCH] chore: coding style (#4054)

---
 rest/handler/timeouthandler.go | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/rest/handler/timeouthandler.go b/rest/handler/timeouthandler.go
index 9a1b001d..98a8883c 100644
--- a/rest/handler/timeouthandler.go
+++ b/rest/handler/timeouthandler.go
@@ -191,6 +191,15 @@ func (tw *timeoutWriter) Write(p []byte) (int, error) {
 	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) {
 	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) {
 	if code < 100 || code > 599 {
 		panic(fmt.Sprintf("invalid WriteHeader code %v", code))