From ca71f4613c5ee12a328609f4f4433e31b648799a Mon Sep 17 00:00:00 2001
From: Mikhail Mazurskiy <mmazurskiy@gitlab.com>
Date: Mon, 4 Dec 2023 15:41:03 +1100
Subject: [PATCH] Add unwrapping to sendDataResponseWriter
---
workhorse/internal/senddata/senddata.go | 10 +++++++---
1 file changed, 7 insertions(+), 3 deletions(-)
diff --git a/workhorse/internal/senddata/senddata.go b/workhorse/internal/senddata/senddata.go
index 764aec1e55830..f0d1da021799d 100644
--- a/workhorse/internal/senddata/senddata.go
+++ b/workhorse/internal/senddata/senddata.go
@@ -44,7 +44,7 @@ func SendData(h http.Handler, injecters ...Injecter) http.Handler {
req: r,
injecters: injecters,
}
- defer s.Flush()
+ defer s.flush()
h.ServeHTTP(&s, r)
}))
}
@@ -101,7 +101,11 @@ func (s *sendDataResponseWriter) tryInject() bool {
return false
}
-func (s *sendDataResponseWriter) Flush() {
+func (s *sendDataResponseWriter) flush() {
s.WriteHeader(http.StatusOK)
- http.NewResponseController(s.rw).Flush()
+}
+
+// Unwrap lets http.ResponseController get the underlying http.ResponseWriter.
+func (s *sendDataResponseWriter) Unwrap() http.ResponseWriter {
+ return s.rw
}
--
GitLab