Merge pull request #34 from foomo/fix/collect-response-header

fix: copy headers to response
foomo · Nov 26, 2024 · bc2e2e0 · bc2e2e0
2 parents ad3a673 + 67a1267
commit bc2e2e0
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/pkg/collect/collect.go b/pkg/collect/collect.go
@@ -142,7 +142,11 @@ func (c *Collect) gtagHandler(l *zap.Logger, w http.ResponseWriter, r *http.Requ
 	defer resp.Body.Close()
 
 	// copy headers
-	r.Header = resp.Header.Clone()
+	for key, values := range resp.Header {
+		for _, value := range values {
+			w.Header().Add(key, value)
+		}
+	}
 
 	if _, err := io.Copy(w, resp.Body); err != nil {
 		return err
@@ -174,7 +178,11 @@ func (c *Collect) mpv2Handler(l *zap.Logger, w http.ResponseWriter, r *http.Requ
 	defer resp.Body.Close()
 
 	// copy headers
-	r.Header = resp.Header.Clone()
+	for key, values := range resp.Header {
+		for _, value := range values {
+			w.Header().Add(key, value)
+		}
+	}
 
 	if _, err := io.Copy(w, resp.Body); err != nil {
 		return err