feat: 支持通过 hash 包随机的读取 map 数据
This commit is contained in:
parent
18b9598f5a
commit
9a3548652a
|
@ -25,3 +25,43 @@ func ToJson[K comparable, V any](m map[K]V) string {
|
||||||
}
|
}
|
||||||
return "{}"
|
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
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in New Issue