Fix events not flushing with telemetry handler

AlexGustafsson · Jan 14, 2025 · 8c39822 · 8c39822
1 parent 8b4fcc0
commit 8c39822
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/internal/api/server.go b/internal/api/server.go
@@ -184,9 +184,8 @@ func NewServer(api *store.Store, hub *events.Hub[store.Event], processQueue chan
 		case <-ctx.Done():
 			w.WriteHeader(http.StatusRequestTimeout)
 		case processQueue <- reference:
+			w.WriteHeader(http.StatusAccepted)
 		}
-
-		w.WriteHeader(http.StatusAccepted)
 	})
 
 	s.mux.HandleFunc("GET /api/v1/feed.rss", func(w http.ResponseWriter, r *http.Request) {

diff --git a/internal/httputil/statusrecorder.go b/internal/httputil/statusrecorder.go
@@ -5,13 +5,21 @@ import (
 )
 
 var _ http.ResponseWriter = (*StatusRecorder)(nil)
+var _ http.Flusher = (*StatusRecorder)(nil)
 
 type StatusRecorder struct {
 	Writer http.ResponseWriter
 
 	statusCode int
 }
 
+// Flush implements http.Flusher.
+func (s *StatusRecorder) Flush() {
+	if flusher, ok := s.Writer.(http.Flusher); ok {
+		flusher.Flush()
+	}
+}
+
 // Header implements http.ResponseWriter.
 func (s *StatusRecorder) Header() http.Header {
 	return s.Writer.Header()