micro/selector/roundrobin/roundrobin.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"
}