ResponseWriterCounter: return statusCode

miolini · Feb 6, 2020 · 09802d7 · 09802d7
1 parent 1628b0e
commit 09802d7
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/response_writer.go b/response_writer.go
@@ -12,8 +12,9 @@ import (
 // ResponseWriterCounter is counter for http.ResponseWriter
 type ResponseWriterCounter struct {
 	http.ResponseWriter
-	count   uint64
-	started time.Time
+	count      uint64
+	started    time.Time
+	statusCode int
 }
 
 // NewResponseWriterCounter function create new ResponseWriterCounter
@@ -56,3 +57,8 @@ func (counter *ResponseWriterCounter) Count() uint64 {
 func (counter *ResponseWriterCounter) Started() time.Time {
 	return counter.started
 }
+
+// StatusCode returns sent status code
+func (counter *ResponseWriterCounter) StatusCode() int {
+	return counter.statusCode
+}