snapshot of https://github.com/ory/keto.git (master, 2026-09-16; blobs > 131072B omitted)

oryx/otelx/middleware.go

@@ -0,0 +1,39 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package otelx
+
+import (
+	"cmp"
+	"net/http"
+	"strings"
+
+	"go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp"
+
+	"github.com/ory/x/httprouterx"
+)
+
+var withDefaultFilters = otelhttp.WithFilter(func(r *http.Request) bool {
+	return !(strings.HasPrefix(r.URL.Path, "/health") ||
+		strings.HasPrefix(r.URL.Path, "/admin/health") ||
+		strings.HasPrefix(r.URL.Path, "/metrics") ||
+		strings.HasPrefix(r.URL.Path, "/admin/metrics"))
+})
+
+func NewMiddleware(next http.Handler, operation string, opts ...otelhttp.Option) http.Handler {
+	myOpts := []otelhttp.Option{
+		withDefaultFilters,
+		otelhttp.WithSpanNameFormatter(func(operation string, r *http.Request) string {
+			return cmp.Or(r.Pattern, operation, r.Method+" "+r.URL.Path)
+		}),
+	}
+	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		callback := func(r2 *http.Request) {
+			r.Pattern = cmp.Or(r.Pattern, r2.Pattern)
+		}
+		r2 := httprouterx.WithAfterMatchHook(r, callback)
+		next.ServeHTTP(w, r2)
+		r.Pattern = cmp.Or(r2.Pattern, r.Pattern) // best-effort in case callback never is called
+	})
+	return otelhttp.NewHandler(handler, operation, append(myOpts, opts...)...)
+}