feat(tracing): add sample rate config

joaofnds · Sep 16, 2023 · c1219c3 · c1219c3
1 parent 705b47c
commit c1219c3
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/adapter/tracing/config.go b/adapter/tracing/config.go
@@ -3,7 +3,8 @@ package tracing
 import "time"
 
 type Config struct {
-	Addr    string        `mapstructure:"addr"`
-	Secure  bool          `mapstructure:"secure"`
-	Timeout time.Duration `mapstructure:"timeout"`
+	Addr       string        `mapstructure:"addr"`
+	Secure     bool          `mapstructure:"secure"`
+	Timeout    time.Duration `mapstructure:"timeout"`
+	SampleRate float64       `mapstructure:"sample_rate"`
 }
diff --git a/adapter/tracing/tracing.go b/adapter/tracing/tracing.go
@@ -32,9 +32,9 @@ var Module = fx.Options(
 		return otlptracegrpc.NewUnstarted(opts...)
 	}),
 
-	fx.Provide(func(exporter *otlptrace.Exporter) *sdktrace.TracerProvider {
+	fx.Provide(func(config Config, exporter *otlptrace.Exporter) *sdktrace.TracerProvider {
 		provider := sdktrace.NewTracerProvider(
-			sdktrace.WithSampler(sdktrace.AlwaysSample()),
+			sdktrace.WithSampler(sdktrace.TraceIDRatioBased(config.SampleRate)),
 			sdktrace.WithBatcher(exporter),
 			sdktrace.WithResource(resource.NewWithAttributes(
 				semconv.SchemaURL,

diff --git a/config/config.yaml b/config/config.yaml
@@ -10,6 +10,7 @@ tracing:
   addr: localhost:4317
   secure: false
   timeout: 1s
+  sample_rate: 1.0
 feature_flags:
   polling_interval: 15s
   flags: