micro/selector/random/random.go

package random

import (
	"go.unistack.org/micro/v3/selector"
	"go.unistack.org/micro/v3/util/rand"
)

type random struct{}

func (r *random) Select(routes []string, opts ...selector.SelectOption) (selector.Next, error) {
	// we can't select from an empty pool of routes
	if len(routes) == 0 {
		return nil, selector.ErrNoneAvailable
	}

	// return the next func
	return func() string {
		// if there is only one route provided we'll select it
		if len(routes) == 1 {
			return routes[0]
		}
		var rng rand.Rand
		// select a random route from the slice
		return routes[rng.Intn(len(routes))]
	}, nil
}

func (r *random) Record(_ string, _ error) error {
	return nil
}

func (r *random) Reset() error {
	return nil
}

func (r *random) String() string {
	return "random"
}

// NewSelector returns a random selector
func NewSelector(opts ...selector.Option) selector.Selector {
	return &random{}
}