From 67a12676ebb4de925406f4ed45994c5a6363f61b Mon Sep 17 00:00:00 2001
From: Kevin Franklin Kim <kevin.kim@bestbytes.com>
Date: Tue, 26 Nov 2024 15:49:21 +0100
Subject: [PATCH] fix: copy headers to response

---
 pkg/collect/collect.go | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/pkg/collect/collect.go b/pkg/collect/collect.go
index 5e0cb86..0b18e1f 100644
--- a/pkg/collect/collect.go
+++ b/pkg/collect/collect.go
@@ -142,8 +142,10 @@ func (c *Collect) gtagHandler(l *zap.Logger, w http.ResponseWriter, r *http.Requ
 	defer resp.Body.Close()
 
 	// copy headers
-	for s := range resp.Header {
-		w.Header().Set(s, resp.Header.Get(s))
+	for key, values := range resp.Header {
+		for _, value := range values {
+			w.Header().Add(key, value)
+		}
 	}
 
 	if _, err := io.Copy(w, resp.Body); err != nil {
@@ -176,8 +178,10 @@ func (c *Collect) mpv2Handler(l *zap.Logger, w http.ResponseWriter, r *http.Requ
 	defer resp.Body.Close()
 
 	// copy headers
-	for s := range resp.Header {
-		w.Header().Set(s, resp.Header.Get(s))
+	for key, values := range resp.Header {
+		for _, value := range values {
+			w.Header().Add(key, value)
+		}
 	}
 
 	if _, err := io.Copy(w, resp.Body); err != nil {