Aleksey Shakhmatov
a90c4cd7fa
- Logging: structured slog output with method, URL, status, duration - DefaultHeaders/UserAgent: inject headers without overwriting existing - BearerAuth/BasicAuth: per-request token resolution and static credentials - Recovery: catches panics in the RoundTripper chain
30 lines
810 B
Go
30 lines
810 B
Go
package middleware
|
|
|
|
import "net/http"
|
|
|
|
// DefaultHeaders returns a middleware that adds the given headers to every
|
|
// outgoing request. Existing headers on the request are not overwritten.
|
|
func DefaultHeaders(headers http.Header) Middleware {
|
|
return func(next http.RoundTripper) http.RoundTripper {
|
|
return RoundTripperFunc(func(req *http.Request) (*http.Response, error) {
|
|
for key, values := range headers {
|
|
if req.Header.Get(key) != "" {
|
|
continue
|
|
}
|
|
for _, v := range values {
|
|
req.Header.Add(key, v)
|
|
}
|
|
}
|
|
return next.RoundTrip(req)
|
|
})
|
|
}
|
|
}
|
|
|
|
// UserAgent returns a middleware that sets the User-Agent header on every
|
|
// outgoing request, unless one is already present.
|
|
func UserAgent(ua string) Middleware {
|
|
return DefaultHeaders(http.Header{
|
|
"User-Agent": {ua},
|
|
})
|
|
}
|