38 lines
594 B
Go
38 lines
594 B
Go
package headers
|
|
|
|
import "net/http"
|
|
|
|
type Headers struct {
|
|
header http.Header
|
|
rt http.RoundTripper
|
|
}
|
|
|
|
func NewHeaders(c *http.Client) Headers {
|
|
if c.Transport == nil {
|
|
c.Transport = http.DefaultTransport
|
|
}
|
|
|
|
ret := Headers{
|
|
header: http.Header{},
|
|
rt: c.Transport,
|
|
}
|
|
|
|
c.Transport = ret
|
|
|
|
return ret
|
|
}
|
|
|
|
func (h *Headers) Add(key, value string) {
|
|
h.header.Add(key, value)
|
|
}
|
|
|
|
func (h Headers) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
for key, vals := range h.header {
|
|
for _, val := range vals {
|
|
req.Header.Add(key, val)
|
|
}
|
|
}
|
|
|
|
return h.rt.RoundTrip(req)
|
|
}
|