53 lines
1.1 KiB
Go
53 lines
1.1 KiB
Go
package influxdb
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"github.com/influxdata/influxdb/v2/kit/tracing"
|
|
)
|
|
|
|
// Shared transports for all clients to prevent leaking connections
|
|
var (
|
|
skipVerifyTransport = &http.Transport{
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
}
|
|
defaultTransport = &http.Transport{}
|
|
)
|
|
|
|
func newURL(addr, path string) (*url.URL, error) {
|
|
u, err := url.Parse(addr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
u.Path = path
|
|
return u, nil
|
|
}
|
|
|
|
func newTraceClient(scheme string, insecure bool) *traceClient {
|
|
hc := &traceClient{
|
|
Client: http.Client{
|
|
Transport: defaultTransport,
|
|
},
|
|
}
|
|
if scheme == "https" && insecure {
|
|
hc.Transport = skipVerifyTransport
|
|
}
|
|
|
|
return hc
|
|
}
|
|
|
|
// traceClient always injects any opentracing trace into the client requests.
|
|
type traceClient struct {
|
|
http.Client
|
|
}
|
|
|
|
// Do injects the trace and then performs the request.
|
|
func (c *traceClient) Do(r *http.Request) (*http.Response, error) {
|
|
span, _ := tracing.StartSpanFromContext(r.Context())
|
|
defer span.Finish()
|
|
tracing.InjectToHTTPRequest(span, r)
|
|
return c.Client.Do(r)
|
|
}
|