feat: 支持通过 hash 包随机的读取 map 数据

This commit is contained in:
kercylan98 2023-07-06 16:17:09 +08:00
parent 18b9598f5a
commit 9a3548652a
1 changed files with 40 additions and 0 deletions

View File

@ -25,3 +25,43 @@ func ToJson[K comparable, V any](m map[K]V) string {
}
return "{}"
}
// RandomGet 随机获取一个元素
func RandomGet[K comparable, V any](m map[K]V) (v V) {
for _, v := range m {
return v
}
return
}
// RandomGetKey 随机获取一个 key
func RandomGetKey[K comparable, V any](m map[K]V) (k K) {
for k = range m {
return k
}
return
}
// RandomGetN 随机获取 n 个元素
// - 获取到的元素不会是重复的,当 map 的长度不足 n 时,返回的元素等同于 hash.ToSlice
func RandomGetN[K comparable, V any](m map[K]V, n int) (vs []V) {
for _, v := range m {
vs = append(vs, v)
if len(vs) >= n {
return
}
}
return
}
// RandomGetKeyN 随机获取 n 个 key
// - 获取到的元素不会是重复的,当 map 的长度不足 n 时,返回的元素等同于 hash.KeyToSlice
func RandomGetKeyN[K comparable, V any](m map[K]V, n int) (ks []K) {
for k := range m {
ks = append(ks, k)
if len(ks) >= n {
return
}
}
return
}