From 5f80e0a7dce5baed4a2e1ea377347b515b872adb Mon Sep 17 00:00:00 2001
From: jarhmj <849104648@qq.com>
Date: Thu, 22 Apr 2021 23:30:48 +0800
Subject: [PATCH] when the status code changes and Written, ResponseWriter
 status code should be changed together.Otherwise,logs status code is not same
 as response.

---
 response_writer.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/response_writer.go b/response_writer.go
index 26826689..6e38f0a4 100644
--- a/response_writer.go
+++ b/response_writer.go
@@ -59,10 +59,11 @@ func (w *responseWriter) reset(writer http.ResponseWriter) {
 
 func (w *responseWriter) WriteHeader(code int) {
 	if code > 0 && w.status != code {
+		w.status = code
 		if w.Written() {
 			debugPrint("[WARNING] Headers were already written. Wanted to override status code %d with %d", w.status, code)
+			w.ResponseWriter.WriteHeader(code)
 		}
-		w.status = code
 	}
 }