31 lines
605 B
Go
31 lines
605 B
Go
package disco
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/hashicorp/go-cleanhttp"
|
|
)
|
|
|
|
const DefaultUserAgent = "terraform-svchost/1.0"
|
|
|
|
func defaultHttpTransport() http.RoundTripper {
|
|
t := cleanhttp.DefaultPooledTransport()
|
|
return &userAgentRoundTripper{
|
|
innerRt: t,
|
|
userAgent: DefaultUserAgent,
|
|
}
|
|
}
|
|
|
|
type userAgentRoundTripper struct {
|
|
innerRt http.RoundTripper
|
|
userAgent string
|
|
}
|
|
|
|
func (rt *userAgentRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
if _, ok := req.Header["User-Agent"]; !ok {
|
|
req.Header.Set("User-Agent", rt.userAgent)
|
|
}
|
|
|
|
return rt.innerRt.RoundTrip(req)
|
|
}
|