fix statusCode store

miolini · Mar 3, 2022 · 4087551 · 4087551
1 parent 09802d7
commit 4087551
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/response_writer.go b/response_writer.go
@@ -39,6 +39,7 @@ func (counter *ResponseWriterCounter) Header() http.Header {
 
 // WriteHeader returns underlying WriteHeader, while setting Runtime header
 func (counter *ResponseWriterCounter) WriteHeader(statusCode int) {
+	counter.statusCode = statusCode
 	counter.Header().Set("X-Runtime", fmt.Sprintf("%.6f", time.Since(counter.started).Seconds()))
 	counter.ResponseWriter.WriteHeader(statusCode)
 }