37 lines
783 B
Go
37 lines
783 B
Go
package httputil
|
|
|
|
import (
|
|
"net/http"
|
|
)
|
|
|
|
type TransportWrapper struct {
|
|
Default http.RoundTripper
|
|
Pre func(*http.Request) error
|
|
Post func(*http.Request, *http.Response) error
|
|
}
|
|
|
|
var _ http.RoundTripper = (*TransportWrapper)(nil)
|
|
|
|
func NewTransportWrapper() *TransportWrapper {
|
|
return &TransportWrapper{
|
|
Default: http.DefaultTransport,
|
|
Pre: func(*http.Request) error { return nil },
|
|
Post: func(*http.Request, *http.Response) error { return nil },
|
|
}
|
|
}
|
|
|
|
func (c *TransportWrapper) RoundTrip(req *http.Request) (r *http.Response, err error) {
|
|
if err := c.Pre(req); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
r, err = c.Default.RoundTrip(req)
|
|
|
|
// Call Post regardless of error:
|
|
if postErr := c.Post(req, r); postErr != nil {
|
|
return r, postErr
|
|
}
|
|
|
|
return r, err
|
|
}
|