サンプリング

provider := trace.NewTracerProvider(
    trace.WithSampler(trace.AlwaysSample()),
)

package main

import (
    "context"

    "go.opentelemetry.io/otel/attribute"
    sdktrace "go.opentelemetry.io/otel/sdk/trace"
    "go.opentelemetry.io/otel/trace"
)

// AttributeBasedSampler は、属性値に基づいてスパンをサンプリングします。
// "high.priority" 属性が true に設定されたスパンは必ずサンプリングされます。
type AttributeBasedSampler struct {
    fallback sdktrace.Sampler
}

// NewAttributeBasedSampler は、新しい AttributeBasedSampler を作成します。
func NewAttributeBasedSampler(fallback sdktrace.Sampler) *AttributeBasedSampler {
    return &AttributeBasedSampler{fallback: fallback}
}

func (s *AttributeBasedSampler) ShouldSample(p sdktrace.SamplingParameters) sdktrace.SamplingResult {
    // tracestate を取得するために、都度親スパンのコンテキストを取り出します。
    psc := trace.SpanContextFromContext(p.ParentContext)

    // 属性が high priority かどうかを確認します。
    for _, attr := range p.Attributes {
        if attr.Key == "high.priority" && attr.Value.AsBool() {
            return sdktrace.SamplingResult{
                Decision:   sdktrace.RecordAndSample,
                Attributes: p.Attributes,
                // 重要: 親の tracestate を保持します。
                Tracestate: psc.TraceState(),
            }
        }
    }

    // 他のスパンはデフォルトのサンプラーにフォールバックします。
    result := s.fallback.ShouldSample(p)

    // フォールバック時も、tracestate が保持されるようにしてください。
    // ビルトインサンプラーはすでにこれを処理していますが、確認するのは良い習慣です。
    return sdktrace.SamplingResult{
        Decision:   result.Decision,
        Attributes: result.Attributes,
        Tracestate: psc.TraceState(),
    }
}

func (s *AttributeBasedSampler) Description() string {
    return "AttributeBasedSampler"
}

sampler := NewAttributeBasedSampler(sdktrace.TraceIDRatioBased(0.1))

provider := sdktrace.NewTracerProvider(
    sdktrace.WithSampler(sampler),
)

provider := sdktrace.NewTracerProvider(
    sdktrace.WithSampler(
        sdktrace.ParentBased(
            NewAttributeBasedSampler(sdktrace.TraceIDRatioBased(0.1)),
        ),
    ),
)