1
0
mirror of https://github.com/open-telemetry/opentelemetry-go.git synced 2024-11-24 08:22:25 +02:00

Simplify trace context benchmark test (#2109)

Remove unnecessary use of the http Request type and the oteltest
package.
This commit is contained in:
Tyler Yahn 2021-07-21 07:25:18 -07:00 committed by GitHub
parent 63dfe64aae
commit 56d420112f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -19,7 +19,6 @@ import (
"net/http"
"testing"
"go.opentelemetry.io/otel/oteltest"
"go.opentelemetry.io/otel/propagation"
"go.opentelemetry.io/otel/trace"
)
@ -28,20 +27,16 @@ func BenchmarkInject(b *testing.B) {
var t propagation.TraceContext
injectSubBenchmarks(b, func(ctx context.Context, b *testing.B) {
req, _ := http.NewRequest("GET", "http://example.com", nil)
h := http.Header{}
b.ResetTimer()
for i := 0; i < b.N; i++ {
t.Inject(ctx, propagation.HeaderCarrier(req.Header))
t.Inject(ctx, propagation.HeaderCarrier(h))
}
})
}
func injectSubBenchmarks(b *testing.B, fn func(context.Context, *testing.B)) {
b.Run("SampledSpanContext", func(b *testing.B) {
spanID, _ := trace.SpanIDFromHex("00f067aa0ba902b7")
traceID, _ := trace.TraceIDFromHex("4bf92f3577b34da6a3ce929d0e0e4736")
mockTracer := oteltest.DefaultTracer()
b.ReportAllocs()
sc := trace.NewSpanContext(trace.SpanContextConfig{
TraceID: traceID,
@ -49,7 +44,6 @@ func injectSubBenchmarks(b *testing.B, fn func(context.Context, *testing.B)) {
TraceFlags: trace.FlagsSampled,
})
ctx := trace.ContextWithRemoteSpanContext(context.Background(), sc)
ctx, _ = mockTracer.Start(ctx, "inject")
fn(ctx, b)
})