@@ -76,19 +76,20 @@ var _ utilnet.TLSClientConfigHolder = &SpdyRoundTripper{}
76
76
var _ httpstream.UpgradeRoundTripper = & SpdyRoundTripper {}
77
77
var _ utilnet.Dialer = & SpdyRoundTripper {}
78
78
79
- // NewRoundTripper creates a new SpdyRoundTripper that will use
80
- // the specified tlsConfig.
81
- func NewRoundTripper (tlsConfig * tls.Config , followRedirects , requireSameHostRedirects bool ) httpstream. UpgradeRoundTripper {
82
- return NewSpdyRoundTripper (tlsConfig , followRedirects , requireSameHostRedirects )
79
+ // NewRoundTripper creates a new SpdyRoundTripper that will use the specified
80
+ // tlsConfig.
81
+ func NewRoundTripper (tlsConfig * tls.Config , followRedirects , requireSameHostRedirects bool ) * SpdyRoundTripper {
82
+ return NewRoundTripperWithProxy (tlsConfig , followRedirects , requireSameHostRedirects , utilnet . NewProxierWithNoProxyCIDR ( http . ProxyFromEnvironment ) )
83
83
}
84
84
85
- // NewSpdyRoundTripper creates a new SpdyRoundTripper that will use
86
- // the specified tlsConfig. This function is mostly meant for unit tests .
87
- func NewSpdyRoundTripper (tlsConfig * tls.Config , followRedirects , requireSameHostRedirects bool ) * SpdyRoundTripper {
85
+ // NewRoundTripperWithProxy creates a new SpdyRoundTripper that will use the
86
+ // specified tlsConfig and proxy func .
87
+ func NewRoundTripperWithProxy (tlsConfig * tls.Config , followRedirects , requireSameHostRedirects bool , proxier func ( * http. Request ) ( * url. URL , error ) ) * SpdyRoundTripper {
88
88
return & SpdyRoundTripper {
89
89
tlsConfig : tlsConfig ,
90
90
followRedirects : followRedirects ,
91
91
requireSameHostRedirects : requireSameHostRedirects ,
92
+ proxier : proxier ,
92
93
}
93
94
}
94
95
@@ -116,11 +117,7 @@ func (s *SpdyRoundTripper) Dial(req *http.Request) (net.Conn, error) {
116
117
// dial dials the host specified by req, using TLS if appropriate, optionally
117
118
// using a proxy server if one is configured via environment variables.
118
119
func (s * SpdyRoundTripper ) dial (req * http.Request ) (net.Conn , error ) {
119
- proxier := s .proxier
120
- if proxier == nil {
121
- proxier = utilnet .NewProxierWithNoProxyCIDR (http .ProxyFromEnvironment )
122
- }
123
- proxyURL , err := proxier (req )
120
+ proxyURL , err := s .proxier (req )
124
121
if err != nil {
125
122
return nil , err
126
123
}
0 commit comments