-
Notifications
You must be signed in to change notification settings - Fork 2
/
telemetry.go
108 lines (88 loc) · 3.11 KB
/
telemetry.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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
package telemetry
import (
"context"
"fmt"
"log/slog"
"net/http"
"strings"
"github.com/felixge/httpsnoop"
"go.opentelemetry.io/contrib/propagators/aws/xray"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/attribute"
"go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc"
"go.opentelemetry.io/otel/sdk/resource"
sdktrace "go.opentelemetry.io/otel/sdk/trace"
semconv "go.opentelemetry.io/otel/semconv/v1.12.0"
"go.opentelemetry.io/otel/trace"
"google.golang.org/grpc"
)
func Setup(ctx context.Context, resource *resource.Resource) (func(context.Context) error, error) {
traceExporter, err := otlptracegrpc.New(ctx,
otlptracegrpc.WithInsecure(),
otlptracegrpc.WithEndpoint("0.0.0.0:4317"),
otlptracegrpc.WithDialOption(grpc.WithBlock()))
if err != nil {
return nil, fmt.Errorf("failed to create new OTLP trace exporter: %w", err)
}
idg := xray.NewIDGenerator()
tp := sdktrace.NewTracerProvider(
sdktrace.WithResource(resource),
sdktrace.WithSampler(sdktrace.AlwaysSample()),
sdktrace.WithBatcher(traceExporter),
sdktrace.WithIDGenerator(idg),
)
otel.SetTracerProvider(tp)
otel.SetTextMapPropagator(xray.Propagator{})
return traceExporter.Shutdown, nil
}
func WrapHandler(handler http.Handler) http.HandlerFunc {
tracer := otel.GetTracerProvider().Tracer("mlpab")
return func(w http.ResponseWriter, r *http.Request) {
route := r.URL.Path
isWelsh := false
if strings.HasPrefix(r.URL.Path, "/cy/") {
route = route[3:]
isWelsh = true
}
target := r.URL.Path
if len(r.URL.RawQuery) > 0 {
target += "?" + r.URL.RawQuery
}
ctx, span := tracer.Start(r.Context(), route,
trace.WithSpanKind(trace.SpanKindServer),
trace.WithAttributes(attribute.Bool("mlpab.welsh", isWelsh)),
trace.WithAttributes(semconv.HTTPTargetKey.String(target)),
trace.WithAttributes(semconv.NetAttributesFromHTTPRequest("tcp", r)...),
trace.WithAttributes(semconv.EndUserAttributesFromHTTPRequest(r)...),
trace.WithAttributes(semconv.HTTPServerAttributesFromHTTPRequest("mlpab", route, r)...),
)
defer span.End()
m := httpsnoop.CaptureMetrics(handler, w, r.WithContext(ctx))
span.SetAttributes(semconv.HTTPAttributesFromHTTPStatusCode(m.Code)...)
span.SetStatus(semconv.SpanStatusFromHTTPStatusCodeAndSpanKind(m.Code, trace.SpanKindServer))
span.SetAttributes(semconv.HTTPResponseContentLengthKey.Int64(m.Written))
}
}
type SlogHandler struct {
handler slog.Handler
}
func NewSlogHandler(h slog.Handler) slog.Handler {
return &SlogHandler{handler: h}
}
func (h *SlogHandler) Enabled(ctx context.Context, level slog.Level) bool {
return h.handler.Enabled(ctx, level)
}
func (h *SlogHandler) Handle(ctx context.Context, record slog.Record) error {
spanCtx := trace.SpanContextFromContext(ctx)
if spanCtx.HasTraceID() {
traceID := spanCtx.TraceID()
record.AddAttrs(slog.String("trace_id", traceID.String()))
}
return h.handler.Handle(ctx, record)
}
func (h *SlogHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
return NewSlogHandler(h.handler.WithAttrs(attrs))
}
func (h *SlogHandler) WithGroup(name string) slog.Handler {
return NewSlogHandler(h.handler.WithGroup(name))
}