mbiryukova/frostfs-http-gw
1
0
Fork 0
forked from TrueCloudLab/frostfs-http-gw
Code Pull requests Activity

Add connection pool implementation (part 3)

Browse source 
Signed-off-by: Pavel Korotkov <pavel@nspcc.ru>
This commit is contained in:
Pavel Korotkov 2021-04-05 20:13:49 +03:00 committed by Pavel Korotkov
parent a44551d42b
commit f7007f2085
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
connections/generator.go → connections/sampler.go
View file

@ -23,15 +23,15 @@ func (wl *workList) pop() int {
}
func NewSampler(probabilities []float64, source rand.Source) *Sampler {
generator := &Sampler{}
sampler := &Sampler{}
var (
small workList
large workList
)
n := len(probabilities)
generator.randomGenerator = rand.New(source)
generator.probabilities = make([]float64, n)
generator.alias = make([]int, n)
sampler.randomGenerator = rand.New(source)
sampler.probabilities = make([]float64, n)
sampler.alias = make([]int, n)
// Compute scaled probabilities.
p := make([]float64, n)
for i := 0; i < n; i++ {
@ -46,8 +46,8 @@ func NewSampler(probabilities []float64, source rand.Source) *Sampler {
}
for len(large) > 0 && len(small) > 0 {
l, g := small.pop(), large.pop()
generator.probabilities[l] = p[l]
generator.alias[l] = g
sampler.probabilities[l] = p[l]
sampler.alias[l] = g
p[g] = (p[g] + p[l]) - 1
if p[g] < 1 {
small.push(g)
@ -57,13 +57,13 @@ func NewSampler(probabilities []float64, source rand.Source) *Sampler {
}
for len(large) > 0 {
g := large.pop()
generator.probabilities[g] = 1
sampler.probabilities[g] = 1
}
for len(small) > 0 {
l := small.pop()
generator.probabilities[l] = 1
sampler.probabilities[l] = 1
}
return generator
return sampler
}
func (g *Sampler) Next() int {