feat: random 包通过权重和概率随机产生一个成员支持返回产生成员的索引或 Key

This commit is contained in:
kercylan98
2023-09-19 12:37:53 +08:00
parent e9bc9fb481
commit 782a1adb37
2 changed files with 26 additions and 5 deletions

View File

@@ -8,6 +8,13 @@ import (
// ProbabilitySlice 按概率随机从切片中产生一个数据并返回命中的对象及是否未命中
// - 当总概率小于 1 将会发生未命中的情况
func ProbabilitySlice[T any](getProbabilityHandle func(data T) float64, data ...T) (hit T, miss bool) {
hit, _, miss = ProbabilitySliceIndex(getProbabilityHandle, data...)
return hit, miss
}
// ProbabilitySliceIndex 按概率随机从切片中产生一个数据并返回命中的对象及对象索引以及是否未命中
// - 当总概率小于 1 将会发生未命中的情况
func ProbabilitySliceIndex[T any](getProbabilityHandle func(data T) float64, data ...T) (hit T, index int, miss bool) {
rd := rand.New(rand.NewSource(time.Now().UnixNano()))
var total float64
@@ -53,10 +60,10 @@ func ProbabilitySlice[T any](getProbabilityHandle func(data T) float64, data ...
}
}
if i >= len(data) {
return hit, true
return hit, -1, true
}
hit = data[i]
return hit, false
return hit, i, false
}
// Probability 输入一个概率,返回是否命中