-
Notifications
You must be signed in to change notification settings - Fork 25
/
datadog.go
80 lines (63 loc) · 2 KB
/
datadog.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
// Unless explicitly stated otherwise all files in this repository are licensed
// under the Apache License Version 2.0.
// This product includes software developed at Datadog (https://www.datadoghq.com/).
// Copyright 2024 Datadog, Inc.
package datadog
import (
"strconv"
"github.com/DataDog/chaos-controller/o11y/tracer/types"
"go.opentelemetry.io/otel/trace"
ddotel "gopkg.in/DataDog/dd-trace-go.v1/ddtrace/opentelemetry"
ddtracer "gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
)
// Sink describes a datadog tracer sink
type Sink struct {
provider *ddotel.TracerProvider
}
// New initiated datadog tracer sink
func New() Sink {
provider := ddotel.NewTracerProvider(
ddtracer.WithProfilerCodeHotspots(true),
ddtracer.WithLogStartup(false),
)
return Sink{provider: provider}
}
// GetLoggableTraceContext returns a Datadog-friendly trace context for logs
// It allows to connect logs with corresponding traces and spans.
func (Sink) GetLoggableTraceContext(span trace.Span) []interface{} {
stringLogContext := []string{
"dd.trace_id", convertTraceID(span.SpanContext().TraceID().String()),
"dd.span_id", convertTraceID(span.SpanContext().SpanID().String()),
}
interfaceLogContext := make([]interface{}, len(stringLogContext))
for i, v := range stringLogContext {
interfaceLogContext[i] = v
}
return interfaceLogContext
}
// GetProvider returns this sink's provider for otel
func (d Sink) GetProvider() trace.TracerProvider {
return d.provider
}
// Stop shutdowns the datadog provider for otel
func (d Sink) Stop() error {
return d.provider.Shutdown()
}
// GetSinkName returns the name of the sink (datadog)
func (d Sink) GetSinkName() string {
return string(types.SinkDriverDatadog)
}
// convertTraceID converts a OTLP Trace/Span ID into a DataDog Trace/Span ID
func convertTraceID(id string) string {
if len(id) < 16 {
return ""
}
if len(id) > 16 {
id = id[16:]
}
intValue, err := strconv.ParseUint(id, 16, 64)
if err != nil {
return ""
}
return strconv.FormatUint(intValue, 10)
}