2019-10-08 19:22:15 +02:00
|
|
|
// Copyright 2019, 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.
|
|
|
|
|
2020-01-28 20:13:46 +02:00
|
|
|
package testtrace_test
|
2019-10-08 19:22:15 +02:00
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"net/http"
|
|
|
|
"testing"
|
|
|
|
|
2019-11-01 20:40:29 +02:00
|
|
|
"go.opentelemetry.io/otel/api/trace"
|
|
|
|
mocktrace "go.opentelemetry.io/otel/internal/trace"
|
2019-10-08 19:22:15 +02:00
|
|
|
)
|
|
|
|
|
|
|
|
func BenchmarkExtractB3(b *testing.B) {
|
|
|
|
testGroup := []struct {
|
|
|
|
singleHeader bool
|
|
|
|
name string
|
|
|
|
tests []extractTest
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
singleHeader: false,
|
|
|
|
name: "multiple headers",
|
|
|
|
tests: extractMultipleHeaders,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
singleHeader: true,
|
|
|
|
name: "single headers",
|
|
|
|
tests: extractSingleHeader,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
singleHeader: false,
|
|
|
|
name: "invalid multiple headers",
|
|
|
|
tests: extractInvalidB3MultipleHeaders,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
singleHeader: true,
|
|
|
|
name: "invalid single headers",
|
|
|
|
tests: extractInvalidB3SingleHeader,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, tg := range testGroup {
|
2020-01-28 20:13:46 +02:00
|
|
|
propagator := trace.B3{SingleHeader: tg.singleHeader}
|
2019-10-08 19:22:15 +02:00
|
|
|
for _, tt := range tg.tests {
|
|
|
|
traceBenchmark(tg.name+"/"+tt.name, b, func(b *testing.B) {
|
|
|
|
ctx := context.Background()
|
|
|
|
req, _ := http.NewRequest("GET", "http://example.com", nil)
|
|
|
|
for h, v := range tt.headers {
|
|
|
|
req.Header.Set(h, v)
|
|
|
|
}
|
2019-10-15 22:12:56 +02:00
|
|
|
b.ReportAllocs()
|
2019-10-08 19:22:15 +02:00
|
|
|
b.ResetTimer()
|
2019-10-15 22:12:56 +02:00
|
|
|
for i := 0; i < b.N; i++ {
|
2020-02-14 09:16:05 +02:00
|
|
|
_ = propagator.Extract(ctx, req.Header)
|
2019-10-15 22:12:56 +02:00
|
|
|
}
|
2019-10-08 19:22:15 +02:00
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func BenchmarkInjectB3(b *testing.B) {
|
|
|
|
var id uint64
|
|
|
|
testGroup := []struct {
|
|
|
|
singleHeader bool
|
|
|
|
name string
|
|
|
|
tests []injectTest
|
|
|
|
}{
|
|
|
|
{
|
|
|
|
singleHeader: false,
|
|
|
|
name: "multiple headers",
|
|
|
|
tests: injectB3MultipleHeader,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
singleHeader: true,
|
|
|
|
name: "single headers",
|
|
|
|
tests: injectB3SingleleHeader,
|
|
|
|
},
|
|
|
|
}
|
|
|
|
|
|
|
|
mockTracer := &mocktrace.MockTracer{
|
|
|
|
Sampled: false,
|
2019-10-16 19:24:38 +02:00
|
|
|
StartSpanID: &id,
|
2019-10-08 19:22:15 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
for _, tg := range testGroup {
|
|
|
|
id = 0
|
2020-01-28 20:13:46 +02:00
|
|
|
propagator := trace.B3{SingleHeader: tg.singleHeader}
|
2019-10-08 19:22:15 +02:00
|
|
|
for _, tt := range tg.tests {
|
|
|
|
traceBenchmark(tg.name+"/"+tt.name, b, func(b *testing.B) {
|
|
|
|
req, _ := http.NewRequest("GET", "http://example.com", nil)
|
|
|
|
ctx := context.Background()
|
|
|
|
if tt.parentSc.IsValid() {
|
2020-02-04 18:55:03 +02:00
|
|
|
ctx = trace.ContextWithRemoteSpanContext(ctx, tt.parentSc)
|
2019-10-08 19:22:15 +02:00
|
|
|
}
|
2020-02-04 18:55:03 +02:00
|
|
|
ctx, _ = mockTracer.Start(ctx, "inject")
|
2019-10-15 22:12:56 +02:00
|
|
|
b.ReportAllocs()
|
2019-10-08 19:22:15 +02:00
|
|
|
b.ResetTimer()
|
2019-10-15 22:12:56 +02:00
|
|
|
for i := 0; i < b.N; i++ {
|
|
|
|
propagator.Inject(ctx, req.Header)
|
|
|
|
}
|
2019-10-08 19:22:15 +02:00
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func traceBenchmark(name string, b *testing.B, fn func(*testing.B)) {
|
|
|
|
b.Run(name, func(b *testing.B) {
|
|
|
|
b.ReportAllocs()
|
|
|
|
fn(b)
|
|
|
|
})
|
|
|
|
b.Run(name, func(b *testing.B) {
|
|
|
|
b.ReportAllocs()
|
|
|
|
fn(b)
|
|
|
|
})
|
|
|
|
}
|