add logging and don't get nodes where they exist in router (#1898)

* add logging and don't get nodes where they exist in router

* add more logging
This commit is contained in:
Asim Aslam 2020-08-06 11:32:06 +01:00 committed by GitHub
parent 8674dc8e62
commit 2b79910ad9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 37 additions and 6 deletions

View File

@ -36,7 +36,7 @@ func newEtcdWatcher(r *etcdRegistry, timeout time.Duration, opts ...registry.Wat
watchPath := prefix watchPath := prefix
if wo.Domain == registry.WildcardDomain { if wo.Domain == registry.WildcardDomain {
if len(wo.Service) > 0 { if len(wo.Service) > 0 {
return nil, errors.New("Cannot watch a service accross domains") return nil, errors.New("Cannot watch a service across domains")
} }
watchPath = prefix watchPath = prefix
} else if len(wo.Service) > 0 { } else if len(wo.Service) > 0 {

View File

@ -24,6 +24,8 @@ type Options struct {
Advertise Strategy Advertise Strategy
// Context for additional options // Context for additional options
Context context.Context Context context.Context
// Precache routes
Precache bool
} }
// Id sets Router Id // Id sets Router Id
@ -68,6 +70,13 @@ func Advertise(a Strategy) Option {
} }
} }
// Precache the routes
func Precache() Option {
return func(o *Options) {
o.Precache = true
}
}
// DefaultOptions returns router default options // DefaultOptions returns router default options
func DefaultOptions() Options { func DefaultOptions() Options {
return Options{ return Options{

View File

@ -157,21 +157,37 @@ func (r *rtr) manageRegistryRoutes(reg registry.Registry, action string) error {
for _, service := range services { for _, service := range services {
// get the services domain from metadata. Fallback to wildcard. // get the services domain from metadata. Fallback to wildcard.
var domain string var domain string
if service.Metadata != nil && len(service.Metadata["domain"]) > 0 { if service.Metadata != nil && len(service.Metadata["domain"]) > 0 {
domain = service.Metadata["domain"] domain = service.Metadata["domain"]
} else { } else {
domain = registry.WildcardDomain domain = registry.WildcardDomain
} }
// we already have nodes
if len(service.Nodes) > 0 {
logger.Tracef("Creating route %v domain: %v", service, domain)
if err := r.manageRoutes(service, action, domain); err != nil {
logger.Tracef("Failed to manage route for %v domain: %v", service, domain)
}
continue
}
// otherwise get all the service info
// get the service to retrieve all its info // get the service to retrieve all its info
srvs, err := reg.GetService(service.Name, registry.GetDomain(domain)) srvs, err := reg.GetService(service.Name, registry.GetDomain(domain))
if err != nil { if err != nil {
logger.Tracef("Failed to get service %s domain: %s", service.Name, domain)
continue continue
} }
// manage the routes for all returned services // manage the routes for all returned services
for _, srv := range srvs { for _, srv := range srvs {
logger.Tracef("Creating route %v domain: %v", srv, domain)
if err := r.manageRoutes(srv, action, domain); err != nil { if err := r.manageRoutes(srv, action, domain); err != nil {
return err logger.Tracef("Failed to manage route for %v domain: %v", srv, domain)
continue
} }
} }
} }
@ -183,8 +199,10 @@ func (r *rtr) manageRegistryRoutes(reg registry.Registry, action string) error {
func (r *rtr) fetchRoutes(service string) error { func (r *rtr) fetchRoutes(service string) error {
services, err := r.options.Registry.GetService(service, registry.GetDomain(registry.WildcardDomain)) services, err := r.options.Registry.GetService(service, registry.GetDomain(registry.WildcardDomain))
if err == registry.ErrNotFound { if err == registry.ErrNotFound {
logger.Tracef("Failed to find route for %s", service)
return nil return nil
} else if err != nil { } else if err != nil {
logger.Tracef("Failed to find route for %s: %v", service, err)
return fmt.Errorf("failed getting services: %v", err) return fmt.Errorf("failed getting services: %v", err)
} }
@ -202,8 +220,10 @@ func (r *rtr) fetchRoutes(service string) error {
domain = registry.WildcardDomain domain = registry.WildcardDomain
} }
logger.Tracef("Creating route %v domain: %v", srv, domain)
if err := r.manageRoutes(srv, "create", domain); err != nil { if err := r.manageRoutes(srv, "create", domain); err != nil {
return err logger.Tracef("Failed to create route for %v domain %v: %v", err)
continue
} }
} }
@ -473,9 +493,11 @@ func (r *rtr) start() error {
return nil return nil
} }
// add all local service routes into the routing table if r.options.Precache {
if err := r.manageRegistryRoutes(r.options.Registry, "create"); err != nil { // add all local service routes into the routing table
return fmt.Errorf("failed adding registry routes: %s", err) if err := r.manageRegistryRoutes(r.options.Registry, "create"); err != nil {
return fmt.Errorf("failed adding registry routes: %s", err)
}
} }
// add default gateway into routing table // add default gateway into routing table