Skip to content

Commit

Permalink
fix: copy headers to response
Browse files Browse the repository at this point in the history
  • Loading branch information
franklinkim committed Nov 26, 2024
1 parent 0853b51 commit 67a1267
Showing 1 changed file with 8 additions and 4 deletions.
12 changes: 8 additions & 4 deletions pkg/collect/collect.go
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
Expand Down Expand Up @@ -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 {
Expand Down

0 comments on commit 67a1267

Please sign in to comment.