62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
|
/*
|
||
|
|
* @Author : huangzj
|
||
|
|
* @Time : 2020/6/29 10:48
|
||
|
|
* @Description：这边的随机获取没有加锁，对于一些需要加锁的场景可以在外面再包装一层
|
||
|
|
*/
|
||
|
|
|
||
|
|
package rand
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"math/rand"
|
||
|
|
)
|
||
|
|
|
||
|
|
var R *rand.Rand
|
||
|
|
|
||
|
|
//获取min和max之间的一个随机数，返回值范围：[min,max]
|
||
|
|
func GetRandInt(min int, max int) int {
|
||
|
|
if min > max {
|
||
|
|
panic(fmt.Sprint("GetRandInt传入参数无效，min = ", min, ",max = ", max))
|
||
|
|
}
|
||
|
|
//Intn返回的结果是[0,n),所以这边会包含min和max
|
||
|
|
value := R.Intn(max-min+1) + min
|
||
|
|
return value
|
||
|
|
}
|
||
|
|
|
||
|
|
// 随机算出count个数不重复
|
||
|
|
func GetRandIntNoRepeat(count, min, max int) []int {
|
||
|
|
if count > max-min+1 {
|
||
|
|
panic("数据错误")
|
||
|
|
}
|
||
|
|
|
||
|
|
randAttr := make([]int, 0)
|
||
|
|
|
||
|
|
// 用于记录以及选中的数
|
||
|
|
selected := make(map[int]interface{}, 0)
|
||
|
|
|
||
|
|
for i := 1; i <= count; i++ {
|
||
|
|
randIndex := GetRandInt(min, max)
|
||
|
|
if _, ok := selected[randIndex]; ok {
|
||
|
|
i -= 1
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
selected[randIndex] = nil
|
||
|
|
randAttr = append(randAttr, randIndex)
|
||
|
|
}
|
||
|
|
|
||
|
|
return randAttr
|
||
|
|
}
|
||
|
|
|
||
|
|
//从一个数组中获取到X个不重复随机数
|
||
|
|
func GetDiffNum(data []int, count int) []int {
|
||
|
|
if count > len(data) {
|
||
|
|
panic("获取随机数数量有误")
|
||
|
|
}
|
||
|
|
diffList := make([]int, 0)
|
||
|
|
diffListIndex := GetRandIntNoRepeat(count, 0, len(data)-1)
|
||
|
|
for _, i := range diffListIndex {
|
||
|
|
diffList = append(diffList, data[i])
|
||
|
|
}
|
||
|
|
return diffList
|
||
|
|
}
|