micro/client/selector/default.go

110 lines
1.9 KiB
Go
Raw Normal View History

2016-05-04 00:06:19 +03:00
package selector
import (
"time"
2016-05-04 00:06:19 +03:00
"github.com/micro/go-micro/registry"
2019-05-31 18:00:44 +03:00
"github.com/micro/go-micro/registry/cache"
2016-05-04 00:06:19 +03:00
)
type registrySelector struct {
2019-02-13 12:47:31 +03:00
so Options
2019-05-31 18:00:44 +03:00
rc cache.Cache
}
2019-05-31 18:00:44 +03:00
func (c *registrySelector) newCache() cache.Cache {
ropts := []cache.Option{}
2019-02-13 12:47:31 +03:00
if c.so.Context != nil {
if t, ok := c.so.Context.Value("selector_ttl").(time.Duration); ok {
2019-05-31 18:00:44 +03:00
ropts = append(ropts, cache.WithTTL(t))
}
}
2019-05-31 18:00:44 +03:00
return cache.New(c.so.Registry, ropts...)
}
func (c *registrySelector) Init(opts ...Option) error {
2016-05-04 00:06:19 +03:00
for _, o := range opts {
o(&c.so)
2016-05-04 00:06:19 +03:00
}
2019-02-13 12:47:31 +03:00
c.rc.Stop()
2019-05-31 18:00:44 +03:00
c.rc = c.newCache()
2016-05-04 00:06:19 +03:00
return nil
}
func (c *registrySelector) Options() Options {
return c.so
2016-05-04 00:06:19 +03:00
}
func (c *registrySelector) Select(service string, opts ...SelectOption) (Next, error) {
2016-05-04 00:06:19 +03:00
sopts := SelectOptions{
Strategy: c.so.Strategy,
2016-05-04 00:06:19 +03:00
}
for _, opt := range opts {
opt(&sopts)
}
// get the service
// try the cache first
// if that fails go directly to the registry
2019-02-13 12:47:31 +03:00
services, err := c.rc.GetService(service)
2016-05-04 00:06:19 +03:00
if err != nil {
if err == registry.ErrNotFound {
return nil, ErrNotFound
}
2016-05-04 00:06:19 +03:00
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 (c *registrySelector) Mark(service string, node *registry.Node, err error) {
2016-05-04 00:06:19 +03:00
}
func (c *registrySelector) Reset(service string) {
2016-05-04 00:06:19 +03:00
}
// Close stops the watcher and destroys the cache
func (c *registrySelector) Close() error {
2019-02-13 12:47:31 +03:00
c.rc.Stop()
2016-05-04 00:06:19 +03:00
return nil
}
func (c *registrySelector) String() string {
return "registry"
2016-05-04 00:06:19 +03:00
}
func NewSelector(opts ...Option) Selector {
2016-05-04 00:06:19 +03:00
sopts := Options{
Strategy: Random,
}
for _, opt := range opts {
opt(&sopts)
}
if sopts.Registry == nil {
sopts.Registry = registry.DefaultRegistry
}
2019-02-13 12:47:31 +03:00
s := &registrySelector{
so: sopts,
}
2019-05-31 18:00:44 +03:00
s.rc = s.newCache()
2019-02-13 12:47:31 +03:00
return s
2016-05-04 00:06:19 +03:00
}