transfer.sh/vendor/go.opencensus.io/stats/view/benchmark_test.go
2019-03-17 20:19:56 +01:00

92 lines
2.2 KiB
Go

// Copyright 2018, OpenCensus 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 view
import (
"context"
"fmt"
"testing"
"time"
"go.opencensus.io/stats"
"go.opencensus.io/tag"
)
var (
m = stats.Float64("m", "", "")
k1, _ = tag.NewKey("k1")
k2, _ = tag.NewKey("k2")
k3, _ = tag.NewKey("k3")
k4, _ = tag.NewKey("k4")
k5, _ = tag.NewKey("k5")
k6, _ = tag.NewKey("k6")
k7, _ = tag.NewKey("k7")
k8, _ = tag.NewKey("k8")
view = &View{
Measure: m,
Aggregation: Distribution(1, 2, 3, 4, 5, 6, 7, 8, 9, 10),
TagKeys: []tag.Key{k1, k2},
}
)
// BenchmarkRecordReqCommand benchmarks calling the internal recording machinery
// directly.
func BenchmarkRecordReqCommand(b *testing.B) {
w := newWorker()
register := &registerViewReq{views: []*View{view}, err: make(chan error, 1)}
register.handleCommand(w)
if err := <-register.err; err != nil {
b.Fatal(err)
}
const tagCount = 10
ctxs := make([]context.Context, 0, tagCount)
for i := 0; i < tagCount; i++ {
ctx, _ := tag.New(context.Background(),
tag.Upsert(k1, fmt.Sprintf("v%d", i)),
tag.Upsert(k2, fmt.Sprintf("v%d", i)),
tag.Upsert(k3, fmt.Sprintf("v%d", i)),
tag.Upsert(k4, fmt.Sprintf("v%d", i)),
tag.Upsert(k5, fmt.Sprintf("v%d", i)),
tag.Upsert(k6, fmt.Sprintf("v%d", i)),
tag.Upsert(k7, fmt.Sprintf("v%d", i)),
tag.Upsert(k8, fmt.Sprintf("v%d", i)),
)
ctxs = append(ctxs, ctx)
}
b.ReportAllocs()
b.ResetTimer()
for i := 0; i < b.N; i++ {
record := &recordReq{
ms: []stats.Measurement{
m.M(1),
m.M(1),
m.M(1),
m.M(1),
m.M(1),
m.M(1),
m.M(1),
m.M(1),
},
tm: tag.FromContext(ctxs[i%len(ctxs)]),
t: time.Now(),
}
record.handleCommand(w)
}
}