382 lines
9.1 KiB
Go
382 lines
9.1 KiB
Go
package mathUtil
|
||
|
||
import (
|
||
"errors"
|
||
"math/rand"
|
||
"sync"
|
||
"time"
|
||
)
|
||
|
||
//SafeRand 安全的随机
|
||
type SafeRand struct {
|
||
*rand.Rand
|
||
mu sync.Mutex
|
||
}
|
||
|
||
func GetSafeRand() *SafeRand {
|
||
return &SafeRand{
|
||
Rand: rand.New(rand.NewSource(time.Now().UnixNano())),
|
||
}
|
||
}
|
||
|
||
func GetSafeRandInSeed(seed int64) *SafeRand {
|
||
return &SafeRand{
|
||
Rand: rand.New(rand.NewSource(seed)),
|
||
}
|
||
}
|
||
|
||
func (this *SafeRand) Int() int {
|
||
this.mu.Lock()
|
||
defer this.mu.Unlock()
|
||
|
||
return this.Rand.Int()
|
||
}
|
||
|
||
func (this *SafeRand) Intn(n int) int {
|
||
this.mu.Lock()
|
||
defer this.mu.Unlock()
|
||
|
||
return this.Rand.Intn(n)
|
||
}
|
||
|
||
func (this *SafeRand) Int31() int32 {
|
||
this.mu.Lock()
|
||
defer this.mu.Unlock()
|
||
|
||
return this.Rand.Int31()
|
||
}
|
||
|
||
func (this *SafeRand) Int31n(n int32) int32 {
|
||
this.mu.Lock()
|
||
defer this.mu.Unlock()
|
||
|
||
return this.Rand.Int31n(n)
|
||
}
|
||
|
||
func (this *SafeRand) Int63() int64 {
|
||
this.mu.Lock()
|
||
defer this.mu.Unlock()
|
||
|
||
return this.Rand.Int63()
|
||
}
|
||
|
||
func (this *SafeRand) Int63n(n int64) int64 {
|
||
this.mu.Lock()
|
||
defer this.mu.Unlock()
|
||
|
||
return this.Rand.Int63n(n)
|
||
}
|
||
|
||
func (this *SafeRand) Float64() float64 {
|
||
this.mu.Lock()
|
||
defer this.mu.Unlock()
|
||
|
||
return this.Rand.Float64()
|
||
}
|
||
|
||
func (this *SafeRand) Float32() float32 {
|
||
this.mu.Lock()
|
||
defer this.mu.Unlock()
|
||
|
||
return this.Rand.Float32()
|
||
}
|
||
|
||
// 获取指定区间的随机数[lower, upper)
|
||
// lower:区间下限
|
||
// upper:区间上限
|
||
// 返回值：随机数
|
||
func (this *SafeRand) GetRandRangeInt(lower, upper int) int {
|
||
return lower + this.Intn(upper-lower)
|
||
}
|
||
|
||
// 获取指定区间的随机数[lower, upper)
|
||
// lower:区间下限
|
||
// upper:区间上限
|
||
// 返回值：随机数
|
||
func (this *SafeRand) GetRandRangeInt32(lower, upper int32) int32 {
|
||
return lower + this.Int31n(upper-lower)
|
||
}
|
||
|
||
// 获取指定区间的随机数[lower, upper)
|
||
// lower:区间下限
|
||
// upper:区间上限
|
||
// 返回值：随机数
|
||
func (this *SafeRand) GetRandRangeInt64(lower, upper int64) int64 {
|
||
return lower + this.Int63n(upper-lower)
|
||
}
|
||
|
||
// 获取随机数[0, n)
|
||
// randObj:随机对象
|
||
// n:范围上限
|
||
// 返回值：随机数
|
||
func (this *SafeRand) GetRandInt(n int) int {
|
||
return this.Intn(n)
|
||
}
|
||
|
||
// 获取随机数[0, n)
|
||
// n:范围上限
|
||
// 返回值：随机数
|
||
func (this *SafeRand) GetRandInt32(n int32) int32 {
|
||
return this.Int31n(n)
|
||
}
|
||
|
||
// 获取随机数[0, n)
|
||
// randObj:随机对象
|
||
// n:范围上限
|
||
// 返回值：随机数
|
||
func (this *SafeRand) GetRandInt64(n int64) int64 {
|
||
return this.Int63n(n)
|
||
}
|
||
|
||
// 获取随机数[0, 1)
|
||
// randObj:随机对象
|
||
// 返回值：随机数
|
||
func (this *SafeRand) GetRandFloat32() float32 {
|
||
return this.Float32()
|
||
}
|
||
|
||
// 获取随机数[0, 1)
|
||
// 返回值：随机数
|
||
func (this *SafeRand) GetRandFloat64() float64 {
|
||
return this.Float64()
|
||
}
|
||
|
||
// 获取随机数列表（1~10000，超过10000会抛出异常）
|
||
// minValue:获取随机数的区间下限值
|
||
// maxValue:获取随机数的区间上限值
|
||
// count:随机数量
|
||
// ifAllowDuplicate:是否允许重复
|
||
// 返回值
|
||
// 随机数列表
|
||
func (this *SafeRand) GetRandNumList(minValue, maxValue, count int, ifAllowDuplicate bool) ([]int, error) {
|
||
if minValue > maxValue {
|
||
return nil, errors.New("minValue can't be bigger than maxValue.")
|
||
}
|
||
|
||
if !ifAllowDuplicate && (maxValue-minValue+1) < count {
|
||
return nil, errors.New("随机的数量超过区间的元素数量")
|
||
}
|
||
|
||
if (maxValue - minValue + 1) > 10000 {
|
||
return nil, errors.New("随机数的区间不能大于10000")
|
||
}
|
||
|
||
// 定义原始数据
|
||
sourceCount := maxValue - minValue + 1
|
||
source := make([]int, sourceCount, sourceCount)
|
||
for index := 0; index < sourceCount; index++ {
|
||
source[index] = minValue + index
|
||
}
|
||
|
||
// 定义返回值
|
||
resultList := make([]int, 0, count)
|
||
|
||
// 获取随机的索引列表
|
||
randIndextList := this.getRandIndexList(len(source), count, ifAllowDuplicate)
|
||
for _, index := range randIndextList {
|
||
// 判断是否已经取到足够数量的数据？
|
||
if count <= 0 {
|
||
break
|
||
}
|
||
|
||
resultList = append(resultList, source[index])
|
||
count -= 1
|
||
}
|
||
|
||
return resultList, nil
|
||
}
|
||
|
||
// 获取随机的int列表
|
||
// source:源列表
|
||
// count:随机数量
|
||
// ifAllowDuplicate:是否允许重复
|
||
// 返回值
|
||
// 随机数列表
|
||
func (this *SafeRand) GetRandIntList(source []int, count int, ifAllowDuplicate bool) ([]int, error) {
|
||
// 在不允许重复的情况下，需要产生的随机数不能超过范围限制
|
||
if ifAllowDuplicate == false && len(source) < count {
|
||
return nil, errors.New("随机的数量超过列表的元素数量")
|
||
}
|
||
|
||
// 定义返回值
|
||
resultList := make([]int, 0, count)
|
||
|
||
// 获取随机的索引列表
|
||
randIndextList := this.getRandIndexList(len(source), count, ifAllowDuplicate)
|
||
for _, index := range randIndextList {
|
||
// 判断是否已经取到足够数量的数据？
|
||
if count <= 0 {
|
||
break
|
||
}
|
||
|
||
resultList = append(resultList, source[index])
|
||
count -= 1
|
||
}
|
||
|
||
return resultList, nil
|
||
}
|
||
|
||
// 获取随机的int32列表
|
||
// source:源列表
|
||
// count:随机数量
|
||
// ifAllowDuplicate:是否允许重复
|
||
// 返回值
|
||
// 随机数列表
|
||
func (this *SafeRand) GetRandInt32List(source []int32, count int, ifAllowDuplicate bool) ([]int32, error) {
|
||
// 在不允许重复的情况下，需要产生的随机数不能超过范围限制
|
||
if ifAllowDuplicate == false && len(source) < count {
|
||
return nil, errors.New("随机的数量超过列表的元素数量")
|
||
}
|
||
|
||
// 定义返回值
|
||
resultList := make([]int32, 0, count)
|
||
|
||
// 获取随机的索引列表
|
||
randIndextList := this.getRandIndexList(len(source), count, ifAllowDuplicate)
|
||
for _, index := range randIndextList {
|
||
// 判断是否已经取到足够数量的数据？
|
||
if count <= 0 {
|
||
break
|
||
}
|
||
|
||
resultList = append(resultList, source[index])
|
||
count -= 1
|
||
}
|
||
|
||
return resultList, nil
|
||
}
|
||
|
||
// 获取随机的int64列表
|
||
// source:源列表
|
||
// count:随机数量
|
||
// ifAllowDuplicate:是否允许重复
|
||
// 返回值
|
||
// 随机数列表
|
||
func (this *SafeRand) GetRandInt64List(source []int64, count int, ifAllowDuplicate bool) ([]int64, error) {
|
||
// 在不允许重复的情况下，需要产生的随机数不能超过范围限制
|
||
if ifAllowDuplicate == false && len(source) < count {
|
||
return nil, errors.New("随机的数量超过列表的元素数量")
|
||
}
|
||
|
||
// 定义返回值
|
||
resultList := make([]int64, 0, count)
|
||
|
||
// 获取随机的索引列表
|
||
randIndextList := this.getRandIndexList(len(source), count, ifAllowDuplicate)
|
||
for _, index := range randIndextList {
|
||
// 判断是否已经取到足够数量的数据？
|
||
if count <= 0 {
|
||
break
|
||
}
|
||
|
||
resultList = append(resultList, source[index])
|
||
count -= 1
|
||
}
|
||
|
||
return resultList, nil
|
||
}
|
||
|
||
// 获取随机的interface{}列表
|
||
// source:源列表
|
||
// count:随机数量
|
||
// ifAllowDuplicate:是否允许重复
|
||
// 返回值
|
||
// 随机数列表
|
||
func (this *SafeRand) GetRandInterfaceList(source []interface{}, count int, ifAllowDuplicate bool) ([]interface{}, error) {
|
||
// 在不允许重复的情况下，需要产生的随机数不能超过范围限制
|
||
if ifAllowDuplicate == false && len(source) < count {
|
||
return nil, errors.New("随机的数量超过列表的元素数量")
|
||
}
|
||
|
||
// 定义返回值
|
||
resultList := make([]interface{}, 0, count)
|
||
|
||
// 获取随机的索引列表
|
||
randIndextList := this.getRandIndexList(len(source), count, ifAllowDuplicate)
|
||
for _, index := range randIndextList {
|
||
// 判断是否已经取到足够数量的数据？
|
||
if count <= 0 {
|
||
break
|
||
}
|
||
|
||
resultList = append(resultList, source[index])
|
||
count -= 1
|
||
}
|
||
|
||
return resultList, nil
|
||
}
|
||
|
||
// 获取随机的索引列表
|
||
// count:数量
|
||
// ifAllowDuplicate:是否允许重复
|
||
// 返回值
|
||
// 随机索引值列表
|
||
func (this *SafeRand) getRandIndexList(maxNum, count int, ifAllowDuplicate bool) []int {
|
||
// 定义返回值
|
||
randIndextList := make([]int, 0, count)
|
||
|
||
// 使用源列表的数据量来初始化一个仅存放索引值的数组
|
||
indexList := make([]int, maxNum, maxNum)
|
||
for index := 0; index < maxNum; index++ {
|
||
indexList[index] = index
|
||
}
|
||
|
||
// 遍历列表并获取随机对象(通过不断缩小随机的范围来实现)
|
||
maxIndex := len(indexList) - 1
|
||
for {
|
||
if len(randIndextList) < count {
|
||
// 获取随机索引(由于Next方法不取上限值，所以需要maxIndex+1)
|
||
randIndex := this.Intn(maxIndex + 1)
|
||
|
||
// 将数据添加到列表，并增加findCount
|
||
randIndextList = append(randIndextList, indexList[randIndex])
|
||
|
||
// 如果不允许重复，则需要特殊处理
|
||
if !ifAllowDuplicate {
|
||
// 将该位置的数据和最大位置的数据进行交换
|
||
indexList[randIndex], indexList[maxIndex] = indexList[maxIndex], indexList[randIndex]
|
||
|
||
// 将随机的范围缩小
|
||
maxIndex -= 1
|
||
}
|
||
} else {
|
||
break
|
||
}
|
||
}
|
||
|
||
return randIndextList
|
||
}
|
||
|
||
// 获取带权重的随机数据
|
||
// source:源数据
|
||
// 返回值
|
||
// 数据项
|
||
// 错误对象
|
||
func (this *SafeRand) GetRandWeight(source []IWeight) (result IWeight, err error) {
|
||
if source == nil || len(source) == 0 {
|
||
err = errors.New("待随机的列表为空")
|
||
return
|
||
}
|
||
|
||
// 计算出总的数据量，并随机一个[0, total)的值
|
||
total := 0
|
||
for _, item := range source {
|
||
total += item.GetWeight()
|
||
}
|
||
|
||
randNum := this.GetRandInt(total)
|
||
|
||
// 根据随机出来的值，判断位于哪个区间
|
||
total = 0
|
||
for _, item := range source {
|
||
total += item.GetWeight()
|
||
if randNum < total {
|
||
result = item
|
||
return
|
||
}
|
||
}
|
||
|
||
err = errors.New("未找到有效的数据")
|
||
return
|
||
}
|