From c3cfe9499b3a85c837cf598ceb3f5dafd8035ccb Mon Sep 17 00:00:00 2001
From: shaoshuai <shaoshuai@qiyi.com>
Date: Tue, 28 May 2024 15:30:44 +0800
Subject: [PATCH] fix status error

---
 writer.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/writer.go b/writer.go
index d048da2..7cc2f4f 100644
--- 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
+}