lib/rand: Optimizations (#7964)
rand.secureSource.Uint64 no longer allocates. rand.String uses a strings.Builder. Benchmark results on linux/amd64: name old time/op new time/op delta SecureSource-8 69.1ns ± 3% 51.7ns ± 3% -25.21% (p=0.000 n=20+10) String-8 2.66µs ± 2% 1.95µs ± 1% -26.61% (p=0.000 n=10+10) name old alloc/op new alloc/op delta SecureSource-8 8.00B ± 0% 0.00B -100.00% (p=0.000 n=20+10) String-8 288B ± 0% 32B ± 0% -88.89% (p=0.000 n=10+10) name old allocs/op new allocs/op delta SecureSource-8 1.00 ± 0% 0.00 -100.00% (p=0.000 n=20+10) String-8 33.0 ± 0% 1.0 ± 0% -96.97% (p=0.000 n=10+10)
This commit is contained in:
@@ -21,6 +21,7 @@ import (
|
||||
type secureSource struct {
|
||||
rd io.Reader
|
||||
mut sync.Mutex
|
||||
buf [8]byte
|
||||
}
|
||||
|
||||
func newSecureSource() *secureSource {
|
||||
@@ -47,18 +48,15 @@ func (s *secureSource) Read(p []byte) (int, error) {
|
||||
}
|
||||
|
||||
func (s *secureSource) Uint64() uint64 {
|
||||
var buf [8]byte
|
||||
|
||||
// Read eight bytes of entropy from the buffered, secure random number
|
||||
// generator. The buffered reader isn't concurrency safe, so we lock
|
||||
// around that.
|
||||
s.mut.Lock()
|
||||
_, err := io.ReadFull(s.rd, buf[:])
|
||||
s.mut.Unlock()
|
||||
defer s.mut.Unlock()
|
||||
|
||||
_, err := io.ReadFull(s.rd, s.buf[:])
|
||||
if err != nil {
|
||||
panic("randomness failure: " + err.Error())
|
||||
}
|
||||
|
||||
// Grab those bytes as an uint64
|
||||
return binary.BigEndian.Uint64(buf[:])
|
||||
return binary.LittleEndian.Uint64(s.buf[:])
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user