mirrors/navidrome
1
0
Fork 0
mirror of https://github.com/navidrome/navidrome.git synced 2025-04-03 20:47:35 +03:00
Code Issues Projects Releases Packages Wiki Activity

Simplify random.Int64 usage with generics

Browse source
This commit is contained in:
Deluan 2024-05-11 20:10:46 -04:00
parent 0ae2944073
commit 3463d0c208
4 changed files with 7 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
utils/random/number.go
View file

@ -3,9 +3,11 @@ package random
import (
"crypto/rand"
"math/big"
"golang.org/x/exp/constraints"
)
func Int64(max int64) int64 {
rnd, _ := rand.Int(rand.Reader, big.NewInt(max))
func Int64[T constraints.Integer](max T) int64 {
rnd, _ := rand.Int(rand.Reader, big.NewInt(int64(max)))
return rnd.Int64()
}

2
utils/random/weighted_random_chooser.go
View file

@ -39,7 +39,7 @@ func (w *WeightedChooser) weightedChoice() (int, error) {
if w.totalWeight == 0 {
return 0, errors.New("no choices available")
}
rnd := Int64(int64(w.totalWeight))
rnd := Int64(w.totalWeight)
for i, weight := range w.weights {
rnd -= int64(weight)
if rnd < 0 {