2018-12-29 16:18:05 +00:00
|
|
|
// Package selector is a way to pick a list of service nodes
|
2015-12-09 19:23:16 +00:00
|
|
|
package selector
|
|
|
|
|
|
|
|
import (
|
|
|
|
"errors"
|
2019-12-03 22:59:44 +03:00
|
|
|
|
2015-12-09 19:23:16 +00:00
|
|
|
"github.com/micro/go-micro/registry"
|
|
|
|
)
|
|
|
|
|
|
|
|
// Selector builds on the registry as a mechanism to pick nodes
|
|
|
|
// and mark their status. This allows host pools and other things
|
|
|
|
// to be built using various algorithms.
|
|
|
|
type Selector interface {
|
2016-01-04 23:07:56 +00:00
|
|
|
Init(opts ...Option) error
|
|
|
|
Options() Options
|
2015-12-09 19:23:16 +00:00
|
|
|
// Select returns a function which should return the next node
|
|
|
|
Select(service string, opts ...SelectOption) (Next, error)
|
|
|
|
// Mark sets the success/error against a node
|
|
|
|
Mark(service string, node *registry.Node, err error)
|
|
|
|
// Reset returns state back to zero for a service
|
|
|
|
Reset(service string)
|
|
|
|
// Close renders the selector unusable
|
|
|
|
Close() error
|
2015-12-19 21:56:14 +00:00
|
|
|
// Name of the selector
|
|
|
|
String() string
|
2015-12-09 19:23:16 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// Next is a function that returns the next node
|
2016-05-03 22:06:19 +01:00
|
|
|
// based on the selector's strategy
|
2015-12-09 19:23:16 +00:00
|
|
|
type Next func() (*registry.Node, error)
|
|
|
|
|
2016-04-23 20:15:01 +01:00
|
|
|
// Filter is used to filter a service during the selection process
|
|
|
|
type Filter func([]*registry.Service) []*registry.Service
|
2015-12-09 19:23:16 +00:00
|
|
|
|
2016-05-03 22:06:19 +01:00
|
|
|
// Strategy is a selection strategy e.g random, round robin
|
|
|
|
type Strategy func([]*registry.Service) Next
|
|
|
|
|
2015-12-09 19:23:16 +00:00
|
|
|
var (
|
2018-12-29 15:44:51 +00:00
|
|
|
DefaultSelector = NewSelector()
|
2015-12-09 19:23:16 +00:00
|
|
|
|
|
|
|
ErrNotFound = errors.New("not found")
|
|
|
|
ErrNoneAvailable = errors.New("none available")
|
|
|
|
)
|