/
traces.go
59 lines (49 loc) · 1.48 KB
/
traces.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
package otel
import (
"context"
"go.opentelemetry.io/otel"
"go.opentelemetry.io/otel/exporters/otlp/otlptrace"
"go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc"
"go.opentelemetry.io/otel/propagation"
"go.opentelemetry.io/otel/sdk/resource"
"go.opentelemetry.io/otel/sdk/trace"
semconv "go.opentelemetry.io/otel/semconv/v1.12.0"
)
type traces struct {
exporter *otlptrace.Exporter
provider *trace.TracerProvider
}
func newTraces(ctx context.Context, serviceName, url string) (traces, error) {
// Create exporter
exporter, err := otlptracegrpc.New(ctx, otlptracegrpc.WithEndpoint(url), otlptracegrpc.WithInsecure())
if err != nil {
return traces{}, err
}
// Create resource
resource := resource.NewWithAttributes(
semconv.SchemaURL,
semconv.ServiceNameKey.String(serviceName),
)
// Create provider
provider := trace.NewTracerProvider(
trace.WithBatcher(exporter),
trace.WithResource(resource),
// Set the sampling rate based on the parent span to 60%
trace.WithSampler(trace.ParentBased(trace.TraceIDRatioBased(0.6))),
)
// Set opentelemetry traces provider globally
// otel.SetTracerProvider(provider)
otel.SetTextMapPropagator(
propagation.NewCompositeTextMapPropagator(
propagation.TraceContext{}, // W3C Trace Context format; https://www.w3.org/TR/trace-context/
),
)
return traces{
exporter: exporter,
provider: provider,
}, nil
}
func (t traces) close(ctx context.Context) {
_ = t.provider.Shutdown(ctx)
_ = t.exporter.Shutdown(ctx)
}