Merge pull request #39 from Ruff-nono/master

fix status error
vearne · May 28, 2024 · a3fdea3 · a3fdea3
2 parents a82a1f8 + c3cfe94
commit a3fdea3
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
+}