spanprofiler: exit early on unsampled traces

Save the work to examine the span and decorate it, if the whole trace is not sampled hence the data is going nowhere. Signed-off-by: Bryan Boreham <[email protected]>
grafana · May 30, 2024 · cc972d1 · cc972d1
1 parent 27d7d41
commit cc972d1
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 4 deletions.
diff --git a/spanprofiler/spanprofiler_test.go b/spanprofiler/spanprofiler_test.go
@@ -9,7 +9,7 @@ import (
 )
 
 func TestSpanProfiler_pprof_labels_propagation(t *testing.T) {
-	tt := initTestTracer(t)
+	tt := initTestTracer(t, 1)
 	defer func() { require.NoError(t, tt.Close()) }()
 
 	t.Run("pprof labels are not propagated to child spans", func(t *testing.T) {

diff --git a/spanprofiler/tracer.go b/spanprofiler/tracer.go
@@ -41,6 +41,9 @@ func (t *tracer) StartSpan(operationName string, opts ...opentracing.StartSpanOp
 	if !ok {
 		return span
 	}
+	if !spanCtx.IsSampled() {
+		return span
+	}
 	// pprof labels are attached only once, at the span root level.
 	if !isRootSpan(opts...) {
 		return span

diff --git a/spanprofiler/tracer_test.go b/spanprofiler/tracer_test.go
@@ -7,13 +7,14 @@ import (
 	"testing"
 
 	"github.com/opentracing/opentracing-go"
+	"github.com/opentracing/opentracing-go/ext"
 	"github.com/stretchr/testify/require"
 	"github.com/uber/jaeger-client-go"
 	jaegercfg "github.com/uber/jaeger-client-go/config"
 )
 
 func TestTracer_pprof_labels_propagation(t *testing.T) {
-	tt := initTestTracer(t)
+	tt := initTestTracer(t, 1)
 	defer func() { require.NoError(t, tt.Close()) }()
 
 	t.Run("root span name and ID are propagated as pprof labels", func(t *testing.T) {
@@ -78,15 +79,15 @@ func TestTracer_pprof_labels_propagation(t *testing.T) {
 	})
 }
 
-func initTestTracer(t *testing.T) io.Closer {
+func initTestTracer(t testing.TB, sampleRate float64) io.Closer {
 	t.Helper()
 	// We can't use mock tracer as we actually rely on the
 	// Jaeger tracer implementation details.
 	c := &jaegercfg.Configuration{
 		ServiceName: "test",
 		Sampler: &jaegercfg.SamplerConfig{
 			Type:  jaeger.SamplerTypeConst,
-			Param: 1,
+			Param: sampleRate,
 		},
 		Reporter: &jaegercfg.ReporterConfig{
 			LocalAgentHostPort: "127.0.0.100:16686",
@@ -126,3 +127,19 @@ func spanTags(span opentracing.Span) opentracing.Tags {
 	}
 	return s.Tags()
 }
+
+func BenchmarkTracer(b *testing.B) {
+	testTag := opentracing.Tag{Key: string(ext.Component), Value: "test"}
+	run := func(samplingRate float64) func(b *testing.B) {
+		return func(b *testing.B) {
+			tt := initTestTracer(b, samplingRate)
+			defer func() { require.NoError(b, tt.Close()) }()
+
+			for i := 0; i < b.N; i++ {
+				opentracing.StartSpan("foo", ext.RPCServerOption(nil), testTag)
+			}
+		}
+	}
+	b.Run("unsampled", run(0))
+	b.Run("sampled", run(1))
+}