micro/selector/default.go

83 lines
1.3 KiB
Go
Raw Normal View History

2016-05-04 00:06:19 +03:00
package selector
import (
"github.com/micro/go-micro/registry"
)
type defaultSelector struct {
2016-06-19 16:41:33 +03:00
so Options
2016-05-04 00:06:19 +03:00
}
func (r *defaultSelector) Init(opts ...Option) error {
for _, o := range opts {
o(&r.so)
}
return nil
}
func (r *defaultSelector) Options() Options {
return r.so
}
func (r *defaultSelector) Select(service string, opts ...SelectOption) (Next, error) {
sopts := SelectOptions{
Strategy: r.so.Strategy,
}
for _, opt := range opts {
opt(&sopts)
}
// get the service
services, err := r.so.Registry.GetService(service)
if err != nil {
return nil, err
}
// apply the filters
for _, filter := range sopts.Filters {
services = filter(services)
}
// if there's nothing left, return
if len(services) == 0 {
return nil, ErrNoneAvailable
2016-05-04 00:06:19 +03:00
}
return sopts.Strategy(services), nil
}
func (r *defaultSelector) Mark(service string, node *registry.Node, err error) {
2016-06-19 16:41:33 +03:00
return
2016-05-04 00:06:19 +03:00
}
func (r *defaultSelector) Reset(service string) {
2016-06-19 16:41:33 +03:00
return
2016-05-04 00:06:19 +03:00
}
func (r *defaultSelector) Close() error {
return nil
}
func (r *defaultSelector) String() string {
return "default"
}
func newDefaultSelector(opts ...Option) Selector {
sopts := Options{
Strategy: Random,
}
for _, opt := range opts {
opt(&sopts)
}
if sopts.Registry == nil {
sopts.Registry = registry.DefaultRegistry
}
2016-05-07 05:08:43 +03:00
return &defaultSelector{
2016-06-19 16:41:33 +03:00
so: sopts,
2016-05-07 01:12:37 +03:00
}
2016-05-04 00:06:19 +03:00
}