From 40875517142895327975bb9f27b257296da3b9a0 Mon Sep 17 00:00:00 2001 From: Artem Andreenko Date: Thu, 3 Mar 2022 00:01:57 -0800 Subject: [PATCH] fix statusCode store --- response_writer.go | 1 + 1 file changed, 1 insertion(+) diff --git a/response_writer.go b/response_writer.go index c9ff052..7787ab9 100644 --- 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) }