fix status error

vearne · May 28, 2024 · c3cfe94 · c3cfe94
1 parent a82a1f8
commit c3cfe94
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/writer.go b/writer.go
@@ -56,3 +56,10 @@ func (tw *TimeoutWriter) Header() http.Header {
 func (tw *TimeoutWriter) Size() int {
 	return tw.size
 }
+
+func (tw *TimeoutWriter) Status() int {
+	if tw.code == 0 || !tw.wroteHeader.Load() {
+		return tw.ResponseWriter.Status()
+	}
+	return tw.code
+}