|
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192 |
- // 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 := ®isterViewReq{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)
- }
- }
|