mirror of https://github.com/grafana/grafana
Add interface Tracer, add Opentelemetry (#41963)
* Add interface Tracer, add Opentelemetry * Fix lint * Fix failing tests and return error if config not parsed fo opentelemetry * Update defaults.ini Add comment with jaeger url * go mod tidy * Remove comments that are not needed * Move OpentracingSpan to tracing.go * Add opentelemetry to sample.inipull/42612/head
idafurjes
4 years ago
committed by
GitHub
parent
be578e5700
commit
d993b12415
@ -0,0 +1,123 @@ |
|||||||
|
package tracing |
||||||
|
|
||||||
|
import ( |
||||||
|
"context" |
||||||
|
"time" |
||||||
|
|
||||||
|
"github.com/grafana/grafana/pkg/infra/log" |
||||||
|
"github.com/grafana/grafana/pkg/setting" |
||||||
|
"go.opentelemetry.io/otel" |
||||||
|
"go.opentelemetry.io/otel/attribute" |
||||||
|
"go.opentelemetry.io/otel/exporters/jaeger" |
||||||
|
"go.opentelemetry.io/otel/sdk/resource" |
||||||
|
tracesdk "go.opentelemetry.io/otel/sdk/trace" |
||||||
|
semconv "go.opentelemetry.io/otel/semconv/v1.4.0" |
||||||
|
trace "go.opentelemetry.io/otel/trace" |
||||||
|
) |
||||||
|
|
||||||
|
type Tracer interface { |
||||||
|
Start(ctx context.Context, spanName string, opts ...trace.SpanStartOption) (context.Context, Span) |
||||||
|
Run(context.Context) error |
||||||
|
} |
||||||
|
|
||||||
|
type Span interface { |
||||||
|
End() |
||||||
|
SetAttributes(kv ...attribute.KeyValue) |
||||||
|
} |
||||||
|
|
||||||
|
var ( |
||||||
|
GlobalTracer trace.Tracer |
||||||
|
) |
||||||
|
|
||||||
|
type OpentelemetryTracingService struct { |
||||||
|
enabled bool |
||||||
|
address string |
||||||
|
log log.Logger |
||||||
|
|
||||||
|
tracerProvider *tracesdk.TracerProvider |
||||||
|
|
||||||
|
Cfg *setting.Cfg |
||||||
|
} |
||||||
|
|
||||||
|
type OpentelemetrySpan struct { |
||||||
|
span trace.Span |
||||||
|
} |
||||||
|
|
||||||
|
func (ots *OpentelemetryTracingService) parseSettingsOpentelemetry() error { |
||||||
|
section, err := ots.Cfg.Raw.GetSection("tracing.opentelemetry.jaeger") |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
ots.address = section.Key("address").MustString("") |
||||||
|
if ots.address != "" { |
||||||
|
ots.enabled = true |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (ots *OpentelemetryTracingService) initTracerProvider() (*tracesdk.TracerProvider, error) { |
||||||
|
// Create the Jaeger exporter
|
||||||
|
exp, err := jaeger.New(jaeger.WithCollectorEndpoint(jaeger.WithEndpoint(ots.address))) |
||||||
|
if err != nil { |
||||||
|
return nil, err |
||||||
|
} |
||||||
|
|
||||||
|
tp := tracesdk.NewTracerProvider( |
||||||
|
tracesdk.WithBatcher(exp), |
||||||
|
tracesdk.WithResource(resource.NewWithAttributes( |
||||||
|
semconv.SchemaURL, |
||||||
|
semconv.ServiceNameKey.String("grafana"), |
||||||
|
attribute.String("environment", "production"), |
||||||
|
)), |
||||||
|
) |
||||||
|
|
||||||
|
return tp, nil |
||||||
|
} |
||||||
|
|
||||||
|
func (ots *OpentelemetryTracingService) initOpentelemetryTracer() error { |
||||||
|
tp, err := ots.initTracerProvider() |
||||||
|
if err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
// Register our TracerProvider as the global so any imported
|
||||||
|
// instrumentation in the future will default to using it
|
||||||
|
// only if tracing is enabled
|
||||||
|
if ots.enabled { |
||||||
|
otel.SetTracerProvider(tp) |
||||||
|
} |
||||||
|
|
||||||
|
ots.tracerProvider = tp |
||||||
|
GlobalTracer = otel.GetTracerProvider().Tracer("component-main") |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (ots *OpentelemetryTracingService) Run(ctx context.Context) error { |
||||||
|
<-ctx.Done() |
||||||
|
|
||||||
|
ots.log.Info("Closing tracing") |
||||||
|
ctxShutdown, cancel := context.WithTimeout(ctx, time.Second*5) |
||||||
|
defer cancel() |
||||||
|
|
||||||
|
if err := ots.tracerProvider.Shutdown(ctxShutdown); err != nil { |
||||||
|
return err |
||||||
|
} |
||||||
|
|
||||||
|
return nil |
||||||
|
} |
||||||
|
|
||||||
|
func (ots *OpentelemetryTracingService) Start(ctx context.Context, spanName string, opts ...trace.SpanStartOption) (context.Context, Span) { |
||||||
|
ctx, span := GlobalTracer.Start(ctx, spanName) |
||||||
|
oSpan := OpentelemetrySpan{span: span} |
||||||
|
return ctx, oSpan |
||||||
|
} |
||||||
|
|
||||||
|
func (s OpentelemetrySpan) End() { |
||||||
|
s.span.End() |
||||||
|
} |
||||||
|
|
||||||
|
func (s OpentelemetrySpan) SetAttributes(kv ...attribute.KeyValue) { |
||||||
|
s.span.SetAttributes(kv...) |
||||||
|
} |
||||||
Loading…
Reference in new issue