fix: datarace

formancehq · Oct 23, 2024 · 1086ddf · 1086ddf
1 parent c5db4ba
commit 1086ddf
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/internal/api/common/middlewares_timeout.go b/internal/api/common/middlewares_timeout.go
@@ -16,6 +16,13 @@ type responseWriter struct {
 	headerWritten      bool
 }
 
+func (r *responseWriter) Header() http.Header {
+	r.Lock()
+	defer r.Unlock()
+
+	return r.ResponseWriter.Header()
+}
+
 func (r *responseWriter) Write(bytes []byte) (int, error) {
 	r.Lock()
 	defer r.Unlock()
@@ -83,7 +90,7 @@ func Timeout(configuration TimeoutConfiguration) func(h http.Handler) http.Handl
 			}()
 			defer close(done)
 
-			h.ServeHTTP(w, r.WithContext(newRequestContext))
+			h.ServeHTTP(rw, r.WithContext(newRequestContext))
 		})
 	}
 }