feat: added sampling rate configuration

cmd/generator/main.go

@@ -4,10 +4,10 @@ import (
  "context"
  "flag"
  "fmt"
+
  "github.com/AH-dark/bytestring"
  md5simd "github.com/minio/md5-simd"
  "github.com/uptrace/opentelemetry-go-extra/otelzap"
- "go.opentelemetry.io/otel"
  "go.uber.org/fx"
  "go.uber.org/zap"
 
@@ -17,7 +17,6 @@ import (
 )
 
 var ctx = context.Background()
-var tracer = otel.Tracer("github.com/AH-dark/gravatar-with-qq-avatar/cmd/generator")
 
 var (
  from = int64(10000)
@@ -31,9 +30,6 @@ func init() {
 }
 
 func generate(ctx context.Context, md5QQMappingRepo dal.MD5QQMappingRepo) error {
- ctx, span := tracer.Start(ctx, "cmd.generator.generate")
- defer span.End()
-
  svr := md5simd.NewServer()
  defer svr.Close()
 

common/observability/trace.go

@@ -74,7 +74,7 @@ func NewTraceProvider(
  ctx, span := tracer.Start(ctx, "observability.NewTraceProvider")
  defer span.End()
 
- tp := tracesdk.NewTracerProvider(
+ opts := []tracesdk.TracerProviderOption{
  tracesdk.WithResource(resource),
  tracesdk.WithBatcher(
  exporter,
@@ -83,7 +83,10 @@ func NewTraceProvider(
  tracesdk.WithExportTimeout(vip.GetDuration("observability.trace.export_timeout")),
  tracesdk.WithMaxQueueSize(vip.GetInt("observability.trace.max_queue_size")),
  ),
- )
+ tracesdk.WithSampler(tracesdk.TraceIDRatioBased(vip.GetFloat64("observability.trace.sampling_rate"))),
+ }
+
+ tp := tracesdk.NewTracerProvider(opts...)
 
  lc.Append(fx.Hook{
  OnStop: func(ctx context.Context) error {