3pp-mirror/pomerium
1
0
Fork 0
mirror of https://github.com/pomerium/pomerium.git synced 2025-06-02 02:42:57 +02:00
Code Issues Projects Releases Packages Wiki Activity

internal/httputil: add HTTPStatsRoundTripper to DefaultClient (#828)

Browse source
This commit is contained in:
Cuong Manh Le 2020-06-09 01:34:32 +07:00 • committed by GitHub
parent 7abe3a3b02
commit 9e711b4612
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 41 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

22
internal/httputil/client.go
View file

@ -12,19 +12,29 @@ import (
"net/url"
"time"
"go.opencensus.io/plugin/ochttp"
"github.com/pomerium/pomerium/internal/telemetry/metrics"
"github.com/pomerium/pomerium/internal/telemetry/requestid"
"github.com/pomerium/pomerium/internal/tripper"
)
// ErrTokenRevoked signifies a token revokation or expiration error
var ErrTokenRevoked = errors.New("token expired or revoked")
type httpClient struct {
*http.Client
requestIDTripper http.RoundTripper
}
func (c *httpClient) Do(req *http.Request) (*http.Response, error) {
tripperChain := tripper.NewChain(metrics.HTTPMetricsRoundTripper("idp_http_client", req.Host))
c.Client.Transport = tripperChain.Then(c.requestIDTripper)
return c.Client.Do(req)
}
// DefaultClient avoids leaks by setting an upper limit for timeouts.
var DefaultClient = &http.Client{
Timeout: 1 * time.Minute,
//todo(bdd): incorporate metrics.HTTPMetricsRoundTripper
Transport: requestid.NewRoundTripper(&ochttp.Transport{}),
var DefaultClient = &httpClient{
&http.Client{Timeout: 1 * time.Minute},
requestid.NewRoundTripper(http.DefaultTransport),
}
// Client provides a simple helper interface to make HTTP requests