mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-04 18:51:17 +00:00
26 lines
516 B
Go
26 lines
516 B
Go
package middleware
|
|
|
|
import (
|
|
"net/http"
|
|
)
|
|
|
|
// HeaderAdder adds headers to responses
|
|
type HeaderAdder struct {
|
|
http.Header
|
|
}
|
|
|
|
// Wrap implements Middleware
|
|
func (h HeaderAdder) Wrap(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// Do it in pre-order since headers need to be added before
|
|
// writing the response body
|
|
dst := w.Header()
|
|
for k, vv := range h.Header {
|
|
for _, v := range vv {
|
|
dst.Add(k, v)
|
|
}
|
|
}
|
|
next.ServeHTTP(w, r)
|
|
})
|
|
}
|