daniel
/
distribution
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #2985 from novas0x2a/default-transport 

make it possible to wrap the client transport in another one
master
Derek McGowan 2019-12-13 11:36:12 -08:00 committed by GitHub
parent 29c300c106 c486db2d71
commit bdf3438b52
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
registry/client/transport/transport.go
View File

@ -6,6 +6,13 @@ import (
"sync" "sync"
) )
func identityTransportWrapper(rt http.RoundTripper) http.RoundTripper {
return rt
}
// DefaultTransportWrapper allows a user to wrap every generated transport
var DefaultTransportWrapper = identityTransportWrapper
// RequestModifier represents an object which will do an inplace // RequestModifier represents an object which will do an inplace
// modification of an HTTP request. // modification of an HTTP request.
type RequestModifier interface { type RequestModifier interface {
@ -31,10 +38,11 @@ func (h headerModifier) ModifyRequest(req *http.Request) error {
// NewTransport creates a new transport which will apply modifiers to // NewTransport creates a new transport which will apply modifiers to
// the request on a RoundTrip call. // the request on a RoundTrip call.
func NewTransport(base http.RoundTripper, modifiers ...RequestModifier) http.RoundTripper { func NewTransport(base http.RoundTripper, modifiers ...RequestModifier) http.RoundTripper {
return &transport{ return DefaultTransportWrapper(
Modifiers: modifiers, &transport{
Base: base, Modifiers: modifiers,
} Base: base,
})
} }
// transport is an http.RoundTripper that makes HTTP requests after // transport is an http.RoundTripper that makes HTTP requests after