mirror of
https://github.com/golang/go
synced 2024-09-30 16:28:32 -06:00
c81623a0cb
Also moves core.Key to label.Key, but leaves the implementations behind for now. After using for a while, the word Tag conveys slightly the wrong concept, tagging implies the entire set of information, label maps better to a single named piece of information. A label is just a named key/value pair, it is not really tied to the event package, separating it makes it much easier to understand the public symbols of the event and core packages, and allows us to also move the key implementations somewhere else, which otherwise dominate the API. Change-Id: I46275d531cec91e28af6ab1e74a2713505d52533 Reviewed-on: https://go-review.googlesource.com/c/tools/+/229239 Run-TryBot: Ian Cottrell <iancottrell@google.com> Reviewed-by: Robert Findley <rfindley@google.com> TryBot-Result: Gobot Gobot <gobot@golang.org>
154 lines
3.5 KiB
Go
154 lines
3.5 KiB
Go
package event_test
|
||
|
||
import (
|
||
"context"
|
||
"io/ioutil"
|
||
"log"
|
||
"testing"
|
||
|
||
"golang.org/x/tools/internal/event"
|
||
"golang.org/x/tools/internal/event/core"
|
||
"golang.org/x/tools/internal/event/export"
|
||
"golang.org/x/tools/internal/event/label"
|
||
)
|
||
|
||
type Hooks struct {
|
||
A func(ctx context.Context, a int) (context.Context, func())
|
||
B func(ctx context.Context, b string) (context.Context, func())
|
||
}
|
||
|
||
var (
|
||
aValue = core.NewIntKey("a", "")
|
||
bValue = core.NewStringKey("b", "")
|
||
aCount = core.NewInt64Key("aCount", "Count of time A is called.")
|
||
aStat = core.NewIntKey("aValue", "A value.")
|
||
bCount = core.NewInt64Key("B", "Count of time B is called.")
|
||
bLength = core.NewIntKey("BLen", "B length.")
|
||
|
||
Baseline = Hooks{
|
||
A: func(ctx context.Context, a int) (context.Context, func()) {
|
||
return ctx, func() {}
|
||
},
|
||
B: func(ctx context.Context, b string) (context.Context, func()) {
|
||
return ctx, func() {}
|
||
},
|
||
}
|
||
|
||
StdLog = Hooks{
|
||
A: func(ctx context.Context, a int) (context.Context, func()) {
|
||
log.Printf("A where a=%d", a)
|
||
return ctx, func() {}
|
||
},
|
||
B: func(ctx context.Context, b string) (context.Context, func()) {
|
||
log.Printf("B where b=%q", b)
|
||
return ctx, func() {}
|
||
},
|
||
}
|
||
|
||
Log = Hooks{
|
||
A: func(ctx context.Context, a int) (context.Context, func()) {
|
||
core.Log1(ctx, "A", aValue.Of(a))
|
||
return ctx, func() {}
|
||
},
|
||
B: func(ctx context.Context, b string) (context.Context, func()) {
|
||
core.Log1(ctx, "B", bValue.Of(b))
|
||
return ctx, func() {}
|
||
},
|
||
}
|
||
|
||
Trace = Hooks{
|
||
A: func(ctx context.Context, a int) (context.Context, func()) {
|
||
return core.Start1(ctx, "A", aValue.Of(a))
|
||
},
|
||
B: func(ctx context.Context, b string) (context.Context, func()) {
|
||
return core.Start1(ctx, "B", bValue.Of(b))
|
||
},
|
||
}
|
||
|
||
Stats = Hooks{
|
||
A: func(ctx context.Context, a int) (context.Context, func()) {
|
||
core.Metric1(ctx, aStat.Of(a))
|
||
core.Metric1(ctx, aCount.Of(1))
|
||
return ctx, func() {}
|
||
},
|
||
B: func(ctx context.Context, b string) (context.Context, func()) {
|
||
core.Metric1(ctx, bLength.Of(len(b)))
|
||
core.Metric1(ctx, bCount.Of(1))
|
||
return ctx, func() {}
|
||
},
|
||
}
|
||
|
||
initialList = []int{0, 1, 22, 333, 4444, 55555, 666666, 7777777}
|
||
stringList = []string{
|
||
"A value",
|
||
"Some other value",
|
||
"A nice longer value but not too long",
|
||
"V",
|
||
"",
|
||
"ı",
|
||
"prime count of values",
|
||
}
|
||
)
|
||
|
||
type namedBenchmark struct {
|
||
name string
|
||
test func(*testing.B)
|
||
}
|
||
|
||
func Benchmark(b *testing.B) {
|
||
b.Run("Baseline", Baseline.runBenchmark)
|
||
b.Run("StdLog", StdLog.runBenchmark)
|
||
benchmarks := []namedBenchmark{
|
||
{"Log", Log.runBenchmark},
|
||
{"Trace", Trace.runBenchmark},
|
||
{"Stats", Stats.runBenchmark},
|
||
}
|
||
|
||
event.SetExporter(nil)
|
||
for _, t := range benchmarks {
|
||
b.Run(t.name+"NoExporter", t.test)
|
||
}
|
||
|
||
event.SetExporter(noopExporter)
|
||
for _, t := range benchmarks {
|
||
b.Run(t.name+"Noop", t.test)
|
||
}
|
||
|
||
event.SetExporter(export.Spans(export.LogWriter(ioutil.Discard, false)))
|
||
for _, t := range benchmarks {
|
||
b.Run(t.name, t.test)
|
||
}
|
||
}
|
||
|
||
func A(ctx context.Context, hooks Hooks, a int) int {
|
||
ctx, done := hooks.A(ctx, a)
|
||
defer done()
|
||
return B(ctx, hooks, a, stringList[a%len(stringList)])
|
||
}
|
||
|
||
func B(ctx context.Context, hooks Hooks, a int, b string) int {
|
||
_, done := hooks.B(ctx, b)
|
||
defer done()
|
||
return a + len(b)
|
||
}
|
||
|
||
func (hooks Hooks) runBenchmark(b *testing.B) {
|
||
ctx := context.Background()
|
||
b.ReportAllocs()
|
||
b.ResetTimer()
|
||
var acc int
|
||
for i := 0; i < b.N; i++ {
|
||
for _, value := range initialList {
|
||
acc += A(ctx, hooks, value)
|
||
}
|
||
}
|
||
}
|
||
|
||
func init() {
|
||
log.SetOutput(ioutil.Discard)
|
||
}
|
||
|
||
func noopExporter(ctx context.Context, ev core.Event, lm label.Map) context.Context {
|
||
return ctx
|
||
}
|