stat response writer should support Flusher interface (#16)

lyft · Aug 6, 2017 · 1a658ed · 1a658ed
1 parent 04fc2f0
commit 1a658ed
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/stat_handler.go b/stat_handler.go
@@ -45,3 +45,9 @@ func (h *statResponseWriter) WriteHeader(code int) {
 	h.handler.scope.NewCounter(strconv.Itoa(code)).Inc()
 	h.delegate.WriteHeader(code)
 }
+
+func (h *statResponseWriter) Flush() {
+	if flusher, ok := h.delegate.(http.Flusher); ok {
+		flusher.Flush()
+	}
+}