mirror of
https://github.com/raz-varren/sacrificial-socket.git
synced 2024-08-05 10:58:36 +08:00
29 lines
531 B
Go
29 lines
531 B
Go
package ss
|
|
|
|
import (
|
|
"math/rand"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
//RNG is a random number generator that is safe for concurrent use by multiple go routines
|
|
type RNG struct {
|
|
r *rand.Rand
|
|
mu *sync.Mutex
|
|
}
|
|
|
|
//Read reads len(b) random bytes into b and always returns a nil error
|
|
func (r *RNG) Read(b []byte) (int, error) {
|
|
r.mu.Lock()
|
|
defer r.mu.Unlock()
|
|
return r.r.Read(b)
|
|
}
|
|
|
|
//NewRNG creates a new random number generator
|
|
func NewRNG() *RNG {
|
|
return &RNG{
|
|
r: rand.New(rand.NewSource(time.Now().UnixNano())),
|
|
mu: &sync.Mutex{},
|
|
}
|
|
}
|