57 lines
1009 B
Go
57 lines
1009 B
Go
package selector
|
|
|
|
import (
|
|
"math/rand"
|
|
|
|
"github.com/micro/go-micro/v3/router"
|
|
)
|
|
|
|
type random struct{}
|
|
|
|
func (r *random) Init(opts ...Option) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *random) Options() Options {
|
|
return Options{}
|
|
}
|
|
|
|
func (r *random) Select(routes []router.Route, opts ...SelectOption) (*router.Route, error) {
|
|
// parse the options
|
|
options := NewSelectOptions(opts...)
|
|
|
|
// apply the filters
|
|
for _, f := range options.Filters {
|
|
routes = f(routes)
|
|
}
|
|
|
|
// we can't select from an empty pool of routes
|
|
if len(routes) == 0 {
|
|
return nil, ErrNoneAvailable
|
|
}
|
|
|
|
// if there is only one route provided we'll select it
|
|
if len(routes) == 1 {
|
|
return &routes[0], nil
|
|
}
|
|
|
|
// select a random route from the slice
|
|
return &routes[rand.Intn(len(routes)-1)], nil
|
|
}
|
|
|
|
func (r *random) Record(route router.Route, err error) error {
|
|
return nil
|
|
}
|
|
|
|
func (r *random) Close() error {
|
|
return nil
|
|
}
|
|
|
|
func (r *random) String() string {
|
|
return "random"
|
|
}
|
|
|
|
func newSelector(...Option) Selector {
|
|
return &random{}
|
|
}
|