38 lines
814 B
Go
38 lines
814 B
Go
package roundrobin
|
|
|
|
import (
|
|
"go.unistack.org/micro/v4/selector"
|
|
"go.unistack.org/micro/v4/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"
|
|
}
|