Evstigneev Denis
38c5fe8b5a
All checks were successful
test / test (push) Successful in 42s
## Pull Request template Please, go through these steps before clicking submit on this PR. 1. Give a descriptive title to your PR. 2. Provide a description of your changes. 3. Make sure you have some relevant tests. 4. Put `closes #XXXX` in your comment to auto-close the issue that your PR fixes (if applicable). **PLEASE REMOVE THIS TEMPLATE BEFORE SUBMITTING** Reviewed-on: #369 Co-authored-by: Evstigneev Denis <danteevstigneev@yandex.ru> Co-committed-by: Evstigneev Denis <danteevstigneev@yandex.ru>
38 lines
814 B
Go
38 lines
814 B
Go
package roundrobin
|
|
|
|
import (
|
|
"go.unistack.org/micro/v3/selector"
|
|
"go.unistack.org/micro/v3/util/rand"
|
|
)
|
|
|
|
// NewSelector returns an initialised round robin selector
|
|
func NewSelector(_ ...selector.Option) selector.Selector {
|
|
return new(roundrobin)
|
|
}
|
|
|
|
type roundrobin struct{}
|
|
|
|
// Select return routes based on algo
|
|
func (r *roundrobin) Select(routes []string, _ ...selector.SelectOption) (selector.Next, error) {
|
|
if len(routes) == 0 {
|
|
return nil, selector.ErrNoneAvailable
|
|
}
|
|
var rng rand.Rand
|
|
i := rng.Intn(len(routes))
|
|
|
|
return func() string {
|
|
route := routes[i%len(routes)]
|
|
// increment
|
|
i++
|
|
return route
|
|
}, nil
|
|
}
|
|
|
|
func (r *roundrobin) Record(_ string, _ error) error { return nil }
|
|
|
|
func (r *roundrobin) Reset() error { return nil }
|
|
|
|
func (r *roundrobin) String() string {
|
|
return "roundrobin"
|
|
}
|