Add aws componenets benchmark

propagators/aws/xray/propagator_test.go

@@ -15,9 +15,14 @@
 package xray
 
 import (
+	"context"
+	"net/http"
 	"strings"
 	"testing"
 
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/propagation"
+
 	"github.com/stretchr/testify/assert"
 
 	"go.opentelemetry.io/otel/trace"
@@ -96,3 +101,32 @@ func TestAwsXrayExtract(t *testing.T) {
 		assert.Equal(t, trace.NewSpanContext(test.expected), sc, info...)
 	}
 }
+
+func BenchmarkPropagatorExtract(b *testing.B) {
+	propagator := Propagator{}
+
+	ctx := context.Background()
+	req, _ := http.NewRequest("GET", "http://example.com", nil)
+
+	req.Header.Set("Root", "1-8a3c60f7-d188f8fa79d48a391a778fa6")
+	req.Header.Set("Parent", "53995c3f42cd8ad8")
+	req.Header.Set("Sampled", "1")
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		_ = propagator.Extract(ctx, propagation.HeaderCarrier(req.Header))
+	}
+}
+
+func BenchmarkPropagatorInject(b *testing.B) {
+	propagator := Propagator{}
+	tracer := otel.Tracer("test")
+
+	req, _ := http.NewRequest("GET", "http://example.com", nil)
+	ctx, _ := tracer.Start(context.Background(), "Parent operation...")
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		propagator.Inject(ctx, propagation.HeaderCarrier(req.Header))
+	}
+}

propagators/aws/xray/sampled_unsampled_span_test.go

@@ -0,0 +1,220 @@
+// Copyright The OpenTelemetry Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package xray
+
+import (
+	"context"
+	"testing"
+
+	"go.opentelemetry.io/otel"
+	"go.opentelemetry.io/otel/attribute"
+	sdktrace "go.opentelemetry.io/otel/sdk/trace"
+	"go.opentelemetry.io/otel/trace"
+)
+
+var tracer = otel.Tracer("sample-app")
+
+func startAndEndSampledSpan() {
+	var span trace.Span
+	_, span = tracer.Start(
+		context.Background(),
+		"Example Trace",
+	)
+
+	defer span.End()
+}
+
+func startAndEndNestedSampledSpan() {
+	var span trace.Span
+	ctx, span := tracer.Start(context.Background(), "Parent operation...")
+	defer span.End()
+
+	_, span = tracer.Start(ctx, "Sub operation...")
+	defer span.End()
+}
+
+func getCurrentSampledSpan() trace.Span {
+	var span trace.Span
+	ctx, span := tracer.Start(
+		context.Background(),
+		"Example Trace",
+	)
+	defer span.End()
+
+	return trace.SpanFromContext(ctx)
+}
+
+func addAttributesToSampledSpan() {
+	var span trace.Span
+	_, span = tracer.Start(
+		context.Background(),
+		"Example Trace",
+	)
+	defer span.End()
+
+	span.SetAttributes(attribute.Key("example attribute 1").String("value 1"))
+	span.SetAttributes(attribute.Key("example attribute 2").String("value 2"))
+}
+
+func startAndEndUnSampledSpan() {
+	var span trace.Span
+	_, span = tracer.Start(
+		context.Background(),
+		"Example Trace",
+	)
+
+	defer span.End()
+}
+
+func startAndEndNestedUnSampledSpan() {
+	var span trace.Span
+	ctx, span := tracer.Start(context.Background(), "Parent operation...")
+	defer span.End()
+
+	_, span = tracer.Start(ctx, "Sub operation...")
+	defer span.End()
+}
+
+func getCurrentUnSampledSpan() trace.Span {
+	var span trace.Span
+	ctx, span := tracer.Start(
+		context.Background(),
+		"Example Trace",
+	)
+	defer span.End()
+
+	return trace.SpanFromContext(ctx)
+}
+
+func addAttributesToUnSampledSpan() {
+	var span trace.Span
+	_, span = tracer.Start(
+		context.Background(),
+		"Example Trace",
+	)
+	defer span.End()
+
+	span.SetAttributes(attribute.Key("example attribute 1").String("value 1"))
+	span.SetAttributes(attribute.Key("example attribute 2").String("value 2"))
+}
+
+func initialization() {
+	idg := NewIDGenerator()
+
+	tp := sdktrace.NewTracerProvider(
+		sdktrace.WithSampler(sdktrace.AlwaysSample()),
+		sdktrace.WithIDGenerator(idg),
+	)
+
+	otel.SetTracerProvider(tp)
+	otel.SetTextMapPropagator(Propagator{})
+}
+
+func BenchmarkStartAndEndSampledSpan(b *testing.B) {
+	currentTraceProvider := otel.GetTracerProvider()
+	currentPropagator := otel.GetTextMapPropagator()
+
+	initialization()
+	for i := 0; i < b.N; i++ {
+		startAndEndSampledSpan()
+	}
+	otel.SetTracerProvider(currentTraceProvider)
+	otel.SetTextMapPropagator(currentPropagator)
+}
+
+func BenchmarkStartAndEndNestedSampledSpan(b *testing.B) {
+	currentTraceProvider := otel.GetTracerProvider()
+	currentPropagator := otel.GetTextMapPropagator()
+
+	initialization()
+	for i := 0; i < b.N; i++ {
+		startAndEndNestedSampledSpan()
+	}
+	otel.SetTracerProvider(currentTraceProvider)
+	otel.SetTextMapPropagator(currentPropagator)
+}
+
+func BenchmarkGetCurrentSampledSpan(b *testing.B) {
+	currentTraceProvider := otel.GetTracerProvider()
+	currentPropagator := otel.GetTextMapPropagator()
+
+	initialization()
+	for i := 0; i < b.N; i++ {
+		getCurrentSampledSpan()
+	}
+	otel.SetTracerProvider(currentTraceProvider)
+	otel.SetTextMapPropagator(currentPropagator)
+}
+
+func BenchmarkAddAttributesToSampledSpan(b *testing.B) {
+	currentTraceProvider := otel.GetTracerProvider()
+	currentPropagator := otel.GetTextMapPropagator()
+
+	initialization()
+	for i := 0; i < b.N; i++ {
+		addAttributesToSampledSpan()
+	}
+	otel.SetTracerProvider(currentTraceProvider)
+	otel.SetTextMapPropagator(currentPropagator)
+}
+
+func BenchmarkStartAndEndUnSampledSpan(b *testing.B) {
+	currentTraceProvider := otel.GetTracerProvider()
+	currentPropagator := otel.GetTextMapPropagator()
+
+	initialization()
+	for i := 0; i < b.N; i++ {
+		startAndEndUnSampledSpan()
+	}
+	otel.SetTracerProvider(currentTraceProvider)
+	otel.SetTextMapPropagator(currentPropagator)
+}
+
+func BenchmarkStartAndEndNestedUnSampledSpan(b *testing.B) {
+	currentTraceProvider := otel.GetTracerProvider()
+	currentPropagator := otel.GetTextMapPropagator()
+
+	initialization()
+	for i := 0; i < b.N; i++ {
+		startAndEndNestedUnSampledSpan()
+	}
+	otel.SetTracerProvider(currentTraceProvider)
+	otel.SetTextMapPropagator(currentPropagator)
+}
+
+func BenchmarkGetCurrentUnSampledSpan(b *testing.B) {
+	currentTraceProvider := otel.GetTracerProvider()
+	currentPropagator := otel.GetTextMapPropagator()
+
+	initialization()
+	for i := 0; i < b.N; i++ {
+		getCurrentUnSampledSpan()
+	}
+	otel.SetTracerProvider(currentTraceProvider)
+	otel.SetTextMapPropagator(currentPropagator)
+}
+
+func BenchmarkAddAttributesToUnSampledSpan(b *testing.B) {
+	currentTraceProvider := otel.GetTracerProvider()
+	currentPropagator := otel.GetTextMapPropagator()
+
+	initialization()
+	for i := 0; i < b.N; i++ {
+		addAttributesToUnSampledSpan()
+	}
+
+	otel.SetTracerProvider(currentTraceProvider)
+	otel.SetTextMapPropagator(currentPropagator)
+}