mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-09 15:37:56 -05:00
9bceaa59d2
* Fixed otelhttp client setups. Note: This may not be the best solution as the http client is defined in many places. There should be a canoncial http client with the proper setup. * Changelog fragment * go mod tidy and gazelle
63 lines
1.9 KiB
Go
63 lines
1.9 KiB
Go
// Package tracing sets up jaeger as an opentracing tool
|
|
// for services in Prysm.
|
|
package tracing
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
prysmTrace "github.com/prysmaticlabs/prysm/v5/monitoring/tracing/trace"
|
|
"github.com/sirupsen/logrus"
|
|
"go.opentelemetry.io/otel"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
"go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp"
|
|
"go.opentelemetry.io/otel/propagation"
|
|
"go.opentelemetry.io/otel/sdk/resource"
|
|
"go.opentelemetry.io/otel/sdk/trace"
|
|
semconv "go.opentelemetry.io/otel/semconv/v1.17.0"
|
|
"go.opentelemetry.io/otel/trace/noop"
|
|
)
|
|
|
|
var log = logrus.WithField("prefix", "tracing")
|
|
|
|
// Setup creates and initializes a new Jaegar tracing configuration with opentelemetry.
|
|
func Setup(ctx context.Context, serviceName, processName, endpoint string, sampleFraction float64, enable bool) error {
|
|
if !enable {
|
|
otel.SetTracerProvider(noop.NewTracerProvider())
|
|
return nil
|
|
}
|
|
prysmTrace.TracingEnabled = true
|
|
|
|
if serviceName == "" {
|
|
return errors.New("tracing service name cannot be empty")
|
|
}
|
|
|
|
log.Infof("Starting otel exporter endpoint at address = %s", endpoint)
|
|
exporter, err := otlptracehttp.New(ctx, otlptracehttp.WithEndpointURL(endpoint))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
tp := trace.NewTracerProvider(
|
|
trace.WithSampler(trace.TraceIDRatioBased(sampleFraction)),
|
|
trace.WithBatcher(
|
|
exporter,
|
|
trace.WithMaxExportBatchSize(trace.DefaultMaxExportBatchSize),
|
|
trace.WithBatchTimeout(trace.DefaultScheduleDelay*time.Millisecond),
|
|
trace.WithMaxExportBatchSize(trace.DefaultMaxExportBatchSize),
|
|
),
|
|
trace.WithResource(
|
|
resource.NewWithAttributes(
|
|
semconv.SchemaURL,
|
|
semconv.ServiceNameKey.String(serviceName),
|
|
attribute.String("process_name", processName),
|
|
),
|
|
),
|
|
)
|
|
|
|
otel.SetTracerProvider(tp)
|
|
otel.SetTextMapPropagator(propagation.NewCompositeTextMapPropagator(propagation.TraceContext{}, propagation.Baggage{}))
|
|
|
|
return nil
|
|
}
|