go/opentelemetry-go
1
0
Fork 0
mirror of https://github.com/open-telemetry/opentelemetry-go.git synced 2025-09-16 09:26:25 +02:00
Code Issues Releases Activity

Simplify trace context benchmark test (#2109)

Browse Source 
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
1 changed files with 2 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
propagation/trace_context_benchmark_test.go
View File

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