ntp/common
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

tracing: option to use a GetClientCertificates function

Browse Source
This commit is contained in:
Ask Bjørn Hansen 2023-10-29 18:18:41 -07:00
parent 5b033a1f0b
commit 62e28b71f1
1 changed files with 24 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
tracing/tracing.go
View File

@ -2,6 +2,8 @@ package tracing
import ( import (
"context" "context"
"crypto/tls"
"crypto/x509"
"os" "os"
"go.ntppool.org/common/logger" "go.ntppool.org/common/logger"
@ -26,9 +28,18 @@ func Tracer() trace.Tracer {
return traceProvider.Tracer("ntppool-tracer") return traceProvider.Tracer("ntppool-tracer")
} }
func Start(ctx context.Context, spanName string, opts ...trace.SpanStartOption) (context.Context, trace.Span) {
return Tracer().Start(ctx, spanName, opts...)
}
type GetClientCertificate func(*tls.CertificateRequestInfo) (*tls.Certificate, error)
type TracerConfig struct { type TracerConfig struct {
ServiceName string ServiceName string
Environment string Environment string
CertificateProvider GetClientCertificate
RootCAs *x509.CertPool
} }
var emptyTpShutdownFunc = func(_ context.Context) error { var emptyTpShutdownFunc = func(_ context.Context) error {
@ -45,7 +56,7 @@ func InitTracer(ctx context.Context, cfg *TracerConfig) (TpShutdownFunc, error)
var exporter otelsdktrace.SpanExporter var exporter otelsdktrace.SpanExporter
if otlpEndPoint := os.Getenv("OTEL_EXPORTER_OTLP_ENDPOINT"); len(otlpEndPoint) > 0 { if otlpEndPoint := os.Getenv("OTEL_EXPORTER_OTLP_ENDPOINT"); len(otlpEndPoint) > 0 {
exporter, err = newOLTPExporter(ctx) exporter, err = newOLTPExporter(ctx, cfg)
} }
if err != nil { if err != nil {
@ -69,8 +80,18 @@ func InitTracer(ctx context.Context, cfg *TracerConfig) (TpShutdownFunc, error)
return tp.Shutdown, nil return tp.Shutdown, nil
} }
func newOLTPExporter(ctx context.Context) (otelsdktrace.SpanExporter, error) { func newOLTPExporter(ctx context.Context, cfg *TracerConfig) (otelsdktrace.SpanExporter, error) {
client := otlptracehttp.NewClient()
opts := []otlptracehttp.Option{otlptracehttp.WithCompression(otlptracehttp.GzipCompression)}
if cfg.CertificateProvider != nil {
opts = append(opts, otlptracehttp.WithTLSClientConfig(&tls.Config{
GetClientCertificate: cfg.CertificateProvider,
RootCAs: cfg.RootCAs,
}))
}
client := otlptracehttp.NewClient(opts...)
exporter, err := otlptrace.New(ctx, client) exporter, err := otlptrace.New(ctx, client)
if err != nil { if err != nil {
logger.Setup().Error("creating OTLP trace exporter", "err", err) logger.Setup().Error("creating OTLP trace exporter", "err", err)