You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
54 lines
1.5 KiB
54 lines
1.5 KiB
// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
|
|
//
|
|
// This Source Code Form is subject to the terms of the MIT License.
|
|
// If a copy of the MIT was not distributed with this file,
|
|
// You can obtain one at https://github.com/gogf/gf.
|
|
|
|
package grand
|
|
|
|
import (
|
|
"crypto/rand"
|
|
|
|
"github.com/gogf/gf/v2/errors/gcode"
|
|
"github.com/gogf/gf/v2/errors/gerror"
|
|
)
|
|
|
|
const (
|
|
// Buffer size for uint32 random number.
|
|
bufferChanSize = 10000
|
|
)
|
|
|
|
var (
|
|
// bufferChan is the buffer for random bytes,
|
|
// every item storing 4 bytes.
|
|
bufferChan = make(chan []byte, bufferChanSize)
|
|
)
|
|
|
|
func init() {
|
|
go asyncProducingRandomBufferBytesLoop()
|
|
}
|
|
|
|
// asyncProducingRandomBufferBytes is a named goroutine, which uses an asynchronous goroutine
|
|
// to produce the random bytes, and a buffer chan to store the random bytes.
|
|
// So it has high performance to generate random numbers.
|
|
func asyncProducingRandomBufferBytesLoop() {
|
|
var step int
|
|
for {
|
|
buffer := make([]byte, 1024)
|
|
if n, err := rand.Read(buffer); err != nil {
|
|
panic(gerror.WrapCode(gcode.CodeInternalError, err, `error reading random buffer from system`))
|
|
} else {
|
|
// The random buffer from system is very expensive,
|
|
// so fully reuse the random buffer by changing
|
|
// the step with a different number can
|
|
// improve the performance a lot.
|
|
// for _, step = range []int{4, 5, 6, 7} {
|
|
for _, step = range []int{4} {
|
|
for i := 0; i <= n-4; i += step {
|
|
bufferChan <- buffer[i : i+4]
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|