Merge pull request #59 from Sovietaced/accessToken

Add OTEL to http clients by default
Sovietaced · Jul 4, 2024 · 47c06c3 · 47c06c3
2 parents cdad72f + f792955
commit 47c06c3
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/keyfunc/okta/okta.go b/keyfunc/okta/okta.go
@@ -7,6 +7,7 @@ import (
 	"github.com/benbjohnson/clock"
 	"github.com/golang-jwt/jwt/v5"
 	"github.com/sovietaced/okta-jwt-verifier/metadata"
+	"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
 	"io"
 	"log/slog"
 	"net/http"
@@ -72,7 +73,12 @@ func WithBackgroundCtx(ctx context.Context) Option {
 
 func defaultOptions() *Options {
 	opts := &Options{}
-	WithHttpClient(http.DefaultClient)(opts)
+
+	httpClient := http.Client{
+		Transport: otelhttp.NewTransport(http.DefaultTransport),
+	}
+
+	WithHttpClient(&httpClient)(opts)
 	withClock(clock.New())(opts)
 	WithCacheTtl(DefaultCacheTtl)(opts)
 	WithFetchStrategy(Lazy)(opts)

diff --git a/metadata/okta/okta.go b/metadata/okta/okta.go
@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"github.com/benbjohnson/clock"
 	"github.com/sovietaced/okta-jwt-verifier/metadata"
+	"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
 	"log/slog"
 	"net/http"
 	"sync"
@@ -70,7 +71,11 @@ func WithBackgroundCtx(ctx context.Context) Option {
 
 func defaultOptions() *Options {
 	opts := &Options{}
-	WithHttpClient(http.DefaultClient)(opts)
+
+	httpClient := http.Client{
+		Transport: otelhttp.NewTransport(http.DefaultTransport),
+	}
+	WithHttpClient(&httpClient)(opts)
 	withClock(clock.New())(opts)
 	WithCacheTtl(DefaultCacheTtl)(opts)
 	WithFetchStrategy(Lazy)(opts)