9456a4e922
Some checks failed
CI / Run lints and checks (push) Failing after 47s
CI / Run tests (push) Failing after 1s
CI / Run race tests (push) Failing after 5s
CI / Build CLI docs (push) Has been skipped
CI / Run buf actions (push) Failing after 5s
CodeQL / Analyze (go) (push) Failing after 3s
CodeQL / Analyze (javascript) (push) Failing after 2s
Docker Image Scanners / scanners (push) Failing after 3s
Format / format (push) Failing after 3s
Synchronize Issue Labels / Synchronize Issue Labels (push) Has been skipped
CI / Generate release (push) Has been skipped
CI / Draft newsletter (push) Has been skipped
CI / Pending approval Slack notification (push) Has been skipped
CI / Send newsletter (push) Has been skipped
89 lines
2.9 KiB
Go
89 lines
2.9 KiB
Go
// 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
|
|
}
|