mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-21 07:56:26 +01:00
45 lines
771 B
Go
45 lines
771 B
Go
package promql
|
|
|
|
import (
|
|
"math/rand"
|
|
"sync"
|
|
"testing"
|
|
)
|
|
|
|
func BenchmarkRollupAvg(b *testing.B) {
|
|
rfa := &rollupFuncArg{
|
|
values: benchValues,
|
|
}
|
|
|
|
b.ReportAllocs()
|
|
b.SetBytes(int64(len(benchValues)))
|
|
b.RunParallel(func(pb *testing.PB) {
|
|
var vSum float64
|
|
for pb.Next() {
|
|
vSum += rollupAvg(rfa)
|
|
}
|
|
SinkLock.Lock()
|
|
Sink += vSum
|
|
SinkLock.Unlock()
|
|
})
|
|
}
|
|
|
|
var (
|
|
// Sink is a global sink for benchmarks.
|
|
// It guarantees the compiler doesn't remove the code in benchmarks,
|
|
// which writes data to the Sink.
|
|
Sink float64
|
|
|
|
// SinkLock locks Sink.
|
|
SinkLock sync.Mutex
|
|
)
|
|
|
|
var benchValues = func() []float64 {
|
|
r := rand.New(rand.NewSource(1))
|
|
values := make([]float64, 1000)
|
|
for i := range values {
|
|
values[i] = r.Float64() * 100
|
|
}
|
|
return values
|
|
}()
|