ceremonyclient/pebble/internal/fastrand/fastrand_test.go
Cassandra Heart 2e2a1e4789
v1.2.0 (#31)
2024-01-03 01:31:42 -06:00

87 lines
1.7 KiB
Go

// Copyright 2020 The LevelDB-Go and Pebble Authors. All rights reserved. Use
// of this source code is governed by a BSD-style license that can be found in
// the LICENSE file.
package fastrand
import (
"fmt"
"sync"
"testing"
"time"
"golang.org/x/exp/rand"
)
type defaultRand struct {
mu sync.Mutex
src rand.PCGSource
}
func newDefaultRand() *defaultRand {
r := &defaultRand{}
r.src.Seed(uint64(time.Now().UnixNano()))
return r
}
func (r *defaultRand) Uint32() uint32 {
r.mu.Lock()
i := uint32(r.src.Uint64())
r.mu.Unlock()
return i
}
func BenchmarkFastRand(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
Uint32()
}
})
}
func BenchmarkDefaultRand(b *testing.B) {
r := newDefaultRand()
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
r.Uint32()
}
})
}
// Benchmarks for single-threaded (ST) use of fastrand compared to
// constructing a Rand, which can have heap allocation overhead.
// Global state to disable elision of benchmark code.
var xg uint32
func BenchmarkSTFastRand(b *testing.B) {
var x uint32
for i := 0; i < b.N; i++ {
// Arbitrary constant.
x = Uint32n(2097152)
}
xg = x
}
func BenchmarkSTDefaultRand(b *testing.B) {
for _, newPeriod := range []int{0, 10, 100, 1000} {
name := "no-new"
if newPeriod > 0 {
name = fmt.Sprintf("new-period=%d", newPeriod)
}
b.Run(name, func(b *testing.B) {
r := rand.New(rand.NewSource(uint64(time.Now().UnixNano())))
b.ResetTimer()
var x uint32
for i := 0; i < b.N; i++ {
if newPeriod > 0 && i%newPeriod == 0 {
r = rand.New(rand.NewSource(uint64(time.Now().UnixNano())))
}
// Arbitrary constant.
x = uint32(r.Uint64n(2097152))
}
xg = x
})
}
}