feat: add trace sampler with configuration

1995parham · Sep 16, 2024 · 1be209c · 1be209c
1 parent 90ec5b2
commit 1be209c
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/internal/infra/config/default.go b/internal/infra/config/default.go
@@ -9,7 +9,7 @@ import (
 
 // Default return default configuration.
 func Default() Config {
-	// nolint: exhaustruct
+	// nolint: exhaustruct, mnd
 	return Config{
 		Logger: logger.Config{
 			Level: "debug",
@@ -24,6 +24,7 @@ func Default() Config {
 			Trace: telemetry.Trace{
 				Enabled:  true,
 				Endpoint: "127.0.0.1:4317",
+				Ratio:    0.1,
 			},
 		},
 		NATS: cmq.Config{

diff --git a/internal/infra/telemetry/config.go b/internal/infra/telemetry/config.go
@@ -13,6 +13,7 @@ type Meter struct {
 }
 
 type Trace struct {
-	Enabled  bool   `json:"enabled,omitempty"  koanf:"enabled"`
-	Endpoint string `json:"endpoint,omitempty" koanf:"endpoint"`
+	Enabled  bool    `json:"enabled,omitempty"  koanf:"enabled"`
+	Endpoint string  `json:"endpoint,omitempty" koanf:"endpoint"`
+	Ratio    float64 `json:"ratio,omitempty"    koanf:"ratio"`
 }
diff --git a/internal/infra/telemetry/provide.go b/internal/infra/telemetry/provide.go
@@ -98,7 +98,11 @@ func Provide(lc fx.Lifecycle, cfg Config) Telemetery {
 	}
 
 	bsp := trace.NewBatchSpanProcessor(exporter)
-	tp := trace.NewTracerProvider(trace.WithSpanProcessor(bsp), trace.WithResource(res))
+	tp := trace.NewTracerProvider(
+		trace.WithSpanProcessor(bsp),
+		trace.WithSampler(trace.ParentBased(trace.TraceIDRatioBased(cfg.Trace.Ratio))),
+		trace.WithResource(res),
+	)
 
 	mp := metric.NewMeterProvider(metric.WithResource(res), metric.WithReader(reader))