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

oryx/otelx/jaeger.go

@@ -0,0 +1,88 @@
+// Copyright © 2023 Ory Corp
+// SPDX-License-Identifier: Apache-2.0
+
+package otelx
+
+import (
+	"net"
+
+	"go.opentelemetry.io/contrib/propagators/b3"
+	jaegerPropagator "go.opentelemetry.io/contrib/propagators/jaeger"
+	"go.opentelemetry.io/contrib/samplers/jaegerremote"
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/exporters/jaeger"
+	"go.opentelemetry.io/otel/propagation"
+	"go.opentelemetry.io/otel/sdk/resource"
+	sdktrace "go.opentelemetry.io/otel/sdk/trace"
+	semconv "go.opentelemetry.io/otel/semconv/v1.27.0"
+	"go.opentelemetry.io/otel/trace"
+)
+
+// SetupJaeger configures and returns a Jaeger tracer.
+//
+// The returned tracer will by default attempt to send spans to a local Jaeger agent.
+// Optionally, [otelx.JaegerConfig.LocalAgentAddress] can be set to specify a different target.
+//
+// By default, unless a parent sampler has taken a sampling decision, every span is sampled.
+// [otelx.JaegerSampling.TraceIDRatio] may be used to customize the sampling probability,
+// optionally alongside [otelx.JaegerSampling.ServerURL] to consult a remote server
+// for the sampling strategy to be used.
+func SetupJaeger(t *Tracer, tracerName string, c *Config) (trace.Tracer, error) {
+	host, port, err := net.SplitHostPort(c.Providers.Jaeger.LocalAgentAddress)
+	if err != nil {
+		return nil, err
+	}
+
+	exp, err := jaeger.New(
+		jaeger.WithAgentEndpoint(
+			jaeger.WithAgentHost(host), jaeger.WithAgentPort(port),
+		),
+	)
+	if err != nil {
+		return nil, err
+	}
+
+	tpOpts := []sdktrace.TracerProviderOption{
+		sdktrace.WithBatcher(exp),
+		sdktrace.WithResource(resource.NewWithAttributes(
+			semconv.SchemaURL,
+			semconv.ServiceName(c.ServiceName),
+			semconv.DeploymentEnvironmentName(c.DeploymentEnvironment))),
+	}
+
+	samplingServerURL := c.Providers.Jaeger.Sampling.ServerURL
+	traceIdRatio := c.Providers.Jaeger.Sampling.TraceIDRatio
+
+	sampler := sdktrace.TraceIDRatioBased(traceIdRatio)
+
+	if samplingServerURL != "" {
+		sampler = jaegerremote.New(
+			"jaegerremote",
+			jaegerremote.WithSamplingServerURL(samplingServerURL),
+			jaegerremote.WithInitialSampler(sampler),
+		)
+	}
+
+	// Respect any sampling decision taken by the client.
+	sampler = sdktrace.ParentBased(sampler)
+	tpOpts = append(tpOpts, sdktrace.WithSampler(sampler))
+
+	tp := sdktrace.NewTracerProvider(tpOpts...)
+	otel.SetTracerProvider(tp)
+
+	// At the moment, software across our cloud stack only support Zipkin (B3)
+	// and Jaeger propagation formats. Proposals for standardized formats for
+	// context propagation are in the works (ref: https://www.w3.org/TR/trace-context/
+	// and https://www.w3.org/TR/baggage/).
+	//
+	// Simply add propagation.TraceContext{} and propagation.Baggage{}
+	// here to enable those as well.
+	prop := propagation.NewCompositeTextMapPropagator(
+		propagation.TraceContext{},
+		jaegerPropagator.Jaeger{},
+		b3.New(b3.WithInjectEncoding(b3.B3MultipleHeader|b3.B3SingleHeader)),
+		propagation.Baggage{},
+	)
+	otel.SetTextMapPropagator(prop)
+	return tp.Tracer(tracerName), nil
+}