Renamed variables, options and functions
This commit is contained in:
		| @@ -12,12 +12,12 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// AdvertiseToNetworkTick defines how often in seconds do we scal the local registry | 	// AdvertiseTick defines how often in seconds do we scal the local registry | ||||||
| 	// to advertise the local services to the network registry | 	// to advertise the local services to the network registry | ||||||
| 	AdvertiseToNetworkTick = 5 * time.Second | 	AdvertiseTick = 5 * time.Second | ||||||
| 	// AdvertiseNetworkTTL defines network registry TTL in seconds | 	// AdvertiseTTL defines network registry TTL in seconds | ||||||
| 	// NOTE: this is a rather arbitrary picked value subject to change | 	// NOTE: this is a rather arbitrary picked value subject to change | ||||||
| 	AdvertiseNetworkTTL = 120 * time.Second | 	AdvertiseTTL = 120 * time.Second | ||||||
| ) | ) | ||||||
|  |  | ||||||
| type router struct { | type router struct { | ||||||
| @@ -71,25 +71,20 @@ func (r *router) Address() string { | |||||||
| 	return r.opts.Address | 	return r.opts.Address | ||||||
| } | } | ||||||
|  |  | ||||||
| // Gossip returns gossip bind address | // Network returns the address router advertises to the network | ||||||
| func (r *router) Gossip() string { |  | ||||||
| 	return r.opts.GossipAddress |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Network returns router's micro network |  | ||||||
| func (r *router) Network() string { | func (r *router) Network() string { | ||||||
| 	return r.opts.NetworkAddress | 	return r.opts.Advertise | ||||||
| } | } | ||||||
|  |  | ||||||
| // Start starts the router | // Start starts the router | ||||||
| func (r *router) Start() error { | func (r *router) Start() error { | ||||||
| 	// add local service routes into the routing table | 	// add local service routes into the routing table | ||||||
| 	if err := r.addServiceRoutes(r.opts.LocalRegistry, r.opts.Address, DefaultLocalMetric); err != nil { | 	if err := r.addServiceRoutes(r.opts.Registry, DefaultLocalMetric); err != nil { | ||||||
| 		return fmt.Errorf("failed adding routes for local services: %v", err) | 		return fmt.Errorf("failed adding routes for local services: %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// add network service routes into the routing table | 	// add network service routes into the routing table | ||||||
| 	if err := r.addServiceRoutes(r.opts.NetworkRegistry, r.opts.NetworkAddress, DefaultNetworkMetric); err != nil { | 	if err := r.addServiceRoutes(r.opts.Network, DefaultNetworkMetric); err != nil { | ||||||
| 		return fmt.Errorf("failed adding routes for network services: %v", err) | 		return fmt.Errorf("failed adding routes for network services: %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -98,12 +93,12 @@ func (r *router) Start() error { | |||||||
| 		return fmt.Errorf("failed to parse router into service node: %v", err) | 		return fmt.Errorf("failed to parse router into service node: %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	localRegWatcher, err := r.opts.LocalRegistry.Watch() | 	localWatcher, err := r.opts.Registry.Watch() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return fmt.Errorf("failed to create local registry watcher: %v", err) | 		return fmt.Errorf("failed to create local registry watcher: %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	networkRegWatcher, err := r.opts.NetworkRegistry.Watch() | 	networkWatcher, err := r.opts.Network.Watch() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return fmt.Errorf("failed to create network registry watcher: %v", err) | 		return fmt.Errorf("failed to create network registry watcher: %v", err) | ||||||
| 	} | 	} | ||||||
| @@ -115,14 +110,14 @@ func (r *router) Start() error { | |||||||
| 	go func() { | 	go func() { | ||||||
| 		defer r.wg.Done() | 		defer r.wg.Done() | ||||||
| 		// watch local registry and register routes in routine table | 		// watch local registry and register routes in routine table | ||||||
| 		errChan <- r.manageServiceRoutes(localRegWatcher, r.opts.Address, DefaultLocalMetric) | 		errChan <- r.manageServiceRoutes(localWatcher, DefaultLocalMetric) | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
| 	r.wg.Add(1) | 	r.wg.Add(1) | ||||||
| 	go func() { | 	go func() { | ||||||
| 		defer r.wg.Done() | 		defer r.wg.Done() | ||||||
| 		// watch network registry and register routes in routine table | 		// watch network registry and register routes in routine table | ||||||
| 		errChan <- r.manageServiceRoutes(networkRegWatcher, r.opts.NetworkAddress, DefaultNetworkMetric) | 		errChan <- r.manageServiceRoutes(networkWatcher, DefaultNetworkMetric) | ||||||
| 	}() | 	}() | ||||||
|  |  | ||||||
| 	r.wg.Add(1) | 	r.wg.Add(1) | ||||||
| @@ -138,19 +133,19 @@ func (r *router) Start() error { | |||||||
| // addServiceRouteslists all available services in given registry and adds them to the routing table. | // addServiceRouteslists all available services in given registry and adds them to the routing table. | ||||||
| // NOTE: this is a one-off operation done when bootstrapping the routing table of the new router. | // NOTE: this is a one-off operation done when bootstrapping the routing table of the new router. | ||||||
| // It returns error if either the services could not be listed or if the routes could not be added to the routing table. | // It returns error if either the services could not be listed or if the routes could not be added to the routing table. | ||||||
| func (r *router) addServiceRoutes(reg registry.Registry, network string, metric int) error { | func (r *router) addServiceRoutes(reg registry.Registry, metric int) error { | ||||||
| 	services, err := reg.ListServices() | 	services, err := reg.ListServices() | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return fmt.Errorf("failed to list services: %v", err) | 		return fmt.Errorf("failed to list services: %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	for _, service := range services { | 	for _, service := range services { | ||||||
| 		route := NewRoute( | 		route := Route{ | ||||||
| 			DestAddr(service.Name), | 			Destination: service.Name, | ||||||
| 			Gateway(r), | 			Router:      r, | ||||||
| 			Network(network), | 			Network:     r.opts.Advertise, | ||||||
| 			Metric(metric), | 			Metric:      metric, | ||||||
| 		) | 		} | ||||||
| 		if err := r.opts.Table.Add(route); err != nil && err != ErrDuplicateRoute { | 		if err := r.opts.Table.Add(route); err != nil && err != ErrDuplicateRoute { | ||||||
| 			return fmt.Errorf("error adding route for service: %s", service.Name) | 			return fmt.Errorf("error adding route for service: %s", service.Name) | ||||||
| 		} | 		} | ||||||
| @@ -160,15 +155,15 @@ func (r *router) addServiceRoutes(reg registry.Registry, network string, metric | |||||||
| } | } | ||||||
|  |  | ||||||
| // parseToNode parses router into registry.Node and returns the result. | // parseToNode parses router into registry.Node and returns the result. | ||||||
| // It returns error if the router network address could not be parsed into service host and port. | // It returns error if the router network address could not be parsed into host and port. | ||||||
| // NOTE: We use ":" as the default delimiter when we split the network address. | // NOTE: We use ":" as the delimiter when we splitting the router network address. | ||||||
| func (r *router) parseToNode() (*registry.Node, error) { | func (r *router) parseToNode() (*registry.Node, error) { | ||||||
| 	// split on ":" as a standard host/port delimiter | 	// split on ":" as a standard host/port delimiter | ||||||
| 	addr := strings.Split(r.opts.NetworkAddress, ":") | 	addr := strings.Split(r.opts.Advertise, ":") | ||||||
| 	// try to parse network port into integer | 	// try to parse network port into integer | ||||||
| 	port, err := strconv.Atoi(addr[1]) | 	port, err := strconv.Atoi(addr[1]) | ||||||
| 	if err != nil { | 	if err != nil { | ||||||
| 		return nil, fmt.Errorf("could not parse router network address from %s: %v", r.opts.NetworkAddress, err) | 		return nil, fmt.Errorf("could not parse router network address: %v", err) | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	node := ®istry.Node{ | 	node := ®istry.Node{ | ||||||
| @@ -184,7 +179,7 @@ func (r *router) parseToNode() (*registry.Node, error) { | |||||||
| // It returns error if either the local services failed to be listed or if it fails to register local service in network registry. | // It returns error if either the local services failed to be listed or if it fails to register local service in network registry. | ||||||
| func (r *router) advertiseToNetwork(node *registry.Node) error { | func (r *router) advertiseToNetwork(node *registry.Node) error { | ||||||
| 	// ticker to periodically scan the local registry | 	// ticker to periodically scan the local registry | ||||||
| 	ticker := time.NewTicker(AdvertiseToNetworkTick) | 	ticker := time.NewTicker(AdvertiseTick) | ||||||
|  |  | ||||||
| 	for { | 	for { | ||||||
| 		select { | 		select { | ||||||
| @@ -192,7 +187,7 @@ func (r *router) advertiseToNetwork(node *registry.Node) error { | |||||||
| 			return nil | 			return nil | ||||||
| 		case <-ticker.C: | 		case <-ticker.C: | ||||||
| 			// list all local services | 			// list all local services | ||||||
| 			services, err := r.opts.LocalRegistry.ListServices() | 			services, err := r.opts.Registry.ListServices() | ||||||
| 			if err != nil { | 			if err != nil { | ||||||
| 				return fmt.Errorf("failed to list local services: %v", err) | 				return fmt.Errorf("failed to list local services: %v", err) | ||||||
| 			} | 			} | ||||||
| @@ -203,7 +198,7 @@ func (r *router) advertiseToNetwork(node *registry.Node) error { | |||||||
| 					Nodes: []*registry.Node{node}, | 					Nodes: []*registry.Node{node}, | ||||||
| 				} | 				} | ||||||
| 				// register the local service in the network registry | 				// register the local service in the network registry | ||||||
| 				if err := r.opts.NetworkRegistry.Register(svc, registry.RegisterTTL(AdvertiseNetworkTTL)); err != nil { | 				if err := r.opts.Network.Register(svc, registry.RegisterTTL(AdvertiseTTL)); err != nil { | ||||||
| 					return fmt.Errorf("failed to register service %s in network registry: %v", svc.Name, err) | 					return fmt.Errorf("failed to register service %s in network registry: %v", svc.Name, err) | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| @@ -213,7 +208,7 @@ func (r *router) advertiseToNetwork(node *registry.Node) error { | |||||||
|  |  | ||||||
| // manageServiceRoutes watches services in given registry and updates the routing table accordingly. | // manageServiceRoutes watches services in given registry and updates the routing table accordingly. | ||||||
| // It returns error if the service registry watcher has stopped or if the routing table failed to be updated. | // It returns error if the service registry watcher has stopped or if the routing table failed to be updated. | ||||||
| func (r *router) manageServiceRoutes(w registry.Watcher, network string, metric int) error { | func (r *router) manageServiceRoutes(w registry.Watcher, metric int) error { | ||||||
| 	// wait in the background for the router to stop | 	// wait in the background for the router to stop | ||||||
| 	// when the router stops, stop the watcher and exit | 	// when the router stops, stop the watcher and exit | ||||||
| 	r.wg.Add(1) | 	r.wg.Add(1) | ||||||
| @@ -236,17 +231,17 @@ func (r *router) manageServiceRoutes(w registry.Watcher, network string, metric | |||||||
| 			break | 			break | ||||||
| 		} | 		} | ||||||
|  |  | ||||||
| 		route := NewRoute( | 		route := Route{ | ||||||
| 			DestAddr(res.Service.Name), | 			Destination: res.Service.Name, | ||||||
| 			Gateway(r), | 			Router:      r, | ||||||
| 			Network(network), | 			Network:     r.opts.Advertise, | ||||||
| 			Metric(metric), | 			Metric:      metric, | ||||||
| 		) | 		} | ||||||
|  |  | ||||||
| 		switch res.Action { | 		switch res.Action { | ||||||
| 		case "create": | 		case "create": | ||||||
| 			if len(res.Service.Nodes) > 0 { | 			if len(res.Service.Nodes) > 0 { | ||||||
| 				/// only return error if the route is not duplicate, but something else has failed | 				// only return error if the route is not duplicate, but something else has failed | ||||||
| 				if err := r.opts.Table.Add(route); err != nil && err != ErrDuplicateRoute { | 				if err := r.opts.Table.Add(route); err != nil && err != ErrDuplicateRoute { | ||||||
| 					return fmt.Errorf("failed to add route for service: %v", res.Service.Name) | 					return fmt.Errorf("failed to add route for service: %v", res.Service.Name) | ||||||
| 				} | 				} | ||||||
| @@ -274,8 +269,7 @@ func (r *router) Stop() error { | |||||||
|  |  | ||||||
| 	// NOTE: we need a more efficient way of doing this e.g. network routes | 	// NOTE: we need a more efficient way of doing this e.g. network routes | ||||||
| 	// should ideally be autodeleted when the router stops gossiping | 	// should ideally be autodeleted when the router stops gossiping | ||||||
| 	// deregister all services advertised by this router from remote registry | 	query := NewQuery(QueryRouter(r), QueryNetwork(r.opts.Advertise)) | ||||||
| 	query := NewQuery(QueryGateway(r), QueryNetwork(r.opts.NetworkAddress)) |  | ||||||
| 	routes, err := r.opts.Table.Lookup(query) | 	routes, err := r.opts.Table.Lookup(query) | ||||||
| 	if err != nil && err != ErrRouteNotFound { | 	if err != nil && err != ErrRouteNotFound { | ||||||
| 		return fmt.Errorf("failed to lookup routes for router %s: %v", r.opts.ID, err) | 		return fmt.Errorf("failed to lookup routes for router %s: %v", r.opts.ID, err) | ||||||
| @@ -289,10 +283,10 @@ func (r *router) Stop() error { | |||||||
|  |  | ||||||
| 	for _, route := range routes { | 	for _, route := range routes { | ||||||
| 		service := ®istry.Service{ | 		service := ®istry.Service{ | ||||||
| 			Name:  route.Options().DestAddr, | 			Name:  route.Destination, | ||||||
| 			Nodes: []*registry.Node{node}, | 			Nodes: []*registry.Node{node}, | ||||||
| 		} | 		} | ||||||
| 		if err := r.opts.NetworkRegistry.Deregister(service); err != nil { | 		if err := r.opts.Network.Deregister(service); err != nil { | ||||||
| 			return fmt.Errorf("failed to deregister service %s from network registry: %v", service.Name, err) | 			return fmt.Errorf("failed to deregister service %s from network registry: %v", service.Name, err) | ||||||
| 		} | 		} | ||||||
| 	} | 	} | ||||||
| @@ -310,7 +304,7 @@ func (r *router) String() string { | |||||||
| 	data := []string{ | 	data := []string{ | ||||||
| 		r.opts.ID, | 		r.opts.ID, | ||||||
| 		r.opts.Address, | 		r.opts.Address, | ||||||
| 		r.opts.NetworkAddress, | 		r.opts.Advertise, | ||||||
| 		fmt.Sprintf("%d", r.opts.Table.Size()), | 		fmt.Sprintf("%d", r.opts.Table.Size()), | ||||||
| 	} | 	} | ||||||
| 	table.Append(data) | 	table.Append(data) | ||||||
|   | |||||||
| @@ -64,7 +64,7 @@ func (t *table) Options() TableOptions { | |||||||
|  |  | ||||||
| // Add adds a route to the routing table | // Add adds a route to the routing table | ||||||
| func (t *table) Add(r Route) error { | func (t *table) Add(r Route) error { | ||||||
| 	destAddr := r.Options().DestAddr | 	destAddr := r.Destination | ||||||
| 	sum := t.hash(r) | 	sum := t.hash(r) | ||||||
|  |  | ||||||
| 	t.Lock() | 	t.Lock() | ||||||
| @@ -86,7 +86,7 @@ func (t *table) Add(r Route) error { | |||||||
| 	} | 	} | ||||||
|  |  | ||||||
| 	// only add the route if the route override is explicitly requested | 	// only add the route if the route override is explicitly requested | ||||||
| 	if _, ok := t.m[destAddr][sum]; ok && r.Options().Policy == OverrideIfExists { | 	if _, ok := t.m[destAddr][sum]; ok && r.Policy == OverrideIfExists { | ||||||
| 		t.m[destAddr][sum] = r | 		t.m[destAddr][sum] = r | ||||||
| 		go t.sendEvent(&Event{Type: UpdateEvent, Route: r}) | 		go t.sendEvent(&Event{Type: UpdateEvent, Route: r}) | ||||||
| 		return nil | 		return nil | ||||||
| @@ -94,7 +94,7 @@ func (t *table) Add(r Route) error { | |||||||
|  |  | ||||||
| 	// if we reached this point without already returning the route already exists | 	// if we reached this point without already returning the route already exists | ||||||
| 	// we return nil only if explicitly requested by the client | 	// we return nil only if explicitly requested by the client | ||||||
| 	if r.Options().Policy == IgnoreIfExists { | 	if r.Policy == IgnoreIfExists { | ||||||
| 		return nil | 		return nil | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -106,7 +106,7 @@ func (t *table) Delete(r Route) error { | |||||||
| 	t.Lock() | 	t.Lock() | ||||||
| 	defer t.Unlock() | 	defer t.Unlock() | ||||||
|  |  | ||||||
| 	destAddr := r.Options().DestAddr | 	destAddr := r.Destination | ||||||
| 	sum := t.hash(r) | 	sum := t.hash(r) | ||||||
|  |  | ||||||
| 	if _, ok := t.m[destAddr]; !ok { | 	if _, ok := t.m[destAddr]; !ok { | ||||||
| @@ -121,7 +121,7 @@ func (t *table) Delete(r Route) error { | |||||||
|  |  | ||||||
| // Update updates routing table with new route | // Update updates routing table with new route | ||||||
| func (t *table) Update(r Route) error { | func (t *table) Update(r Route) error { | ||||||
| 	destAddr := r.Options().DestAddr | 	destAddr := r.Destination | ||||||
| 	sum := t.hash(r) | 	sum := t.hash(r) | ||||||
|  |  | ||||||
| 	t.Lock() | 	t.Lock() | ||||||
| @@ -165,29 +165,33 @@ func (t *table) Lookup(q Query) ([]Route, error) { | |||||||
| 	var results []Route | 	var results []Route | ||||||
|  |  | ||||||
| 	for destAddr, routes := range t.m { | 	for destAddr, routes := range t.m { | ||||||
| 		if q.Options().DestAddr != "*" { | 		if q.Options().Destination != "*" { | ||||||
| 			if q.Options().DestAddr != destAddr { | 			if q.Options().Destination != destAddr { | ||||||
| 				continue | 				continue | ||||||
| 			} | 			} | ||||||
| 			for _, route := range routes { | 			for _, route := range routes { | ||||||
| 				if q.Options().Network == "*" || q.Options().Network == route.Options().Network { | 				if q.Options().Network == "*" || q.Options().Network == route.Network { | ||||||
| 					if q.Options().Gateway.ID() == "*" || q.Options().Gateway.ID() == route.Options().Gateway.ID() { | 					if q.Options().Router.ID() == "*" || q.Options().Router.ID() == route.Router.ID() { | ||||||
|  | 						if route.Metric <= q.Options().Metric { | ||||||
| 							results = append(results, route) | 							results = append(results, route) | ||||||
| 						} | 						} | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
|  | 		} | ||||||
|  |  | ||||||
| 		if q.Options().DestAddr == "*" { | 		if q.Options().Destination == "*" { | ||||||
| 			for _, route := range routes { | 			for _, route := range routes { | ||||||
| 				if q.Options().Network == "*" || q.Options().Network == route.Options().Network { | 				if q.Options().Network == "*" || q.Options().Network == route.Router.Network() { | ||||||
| 					if q.Options().Gateway.ID() == "*" || q.Options().Gateway.ID() == route.Options().Gateway.ID() { | 					if q.Options().Router.ID() == "*" || q.Options().Router.ID() == route.Router.ID() { | ||||||
|  | 						if route.Metric <= q.Options().Metric { | ||||||
| 							results = append(results, route) | 							results = append(results, route) | ||||||
| 						} | 						} | ||||||
| 					} | 					} | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| 		} | 		} | ||||||
|  | 	} | ||||||
|  |  | ||||||
| 	if len(results) == 0 && q.Options().Policy != DiscardNoRoute { | 	if len(results) == 0 && q.Options().Policy != DiscardNoRoute { | ||||||
| 		return nil, ErrRouteNotFound | 		return nil, ErrRouteNotFound | ||||||
| @@ -200,7 +204,7 @@ func (t *table) Lookup(q Query) ([]Route, error) { | |||||||
| func (t *table) Watch(opts ...WatchOption) (Watcher, error) { | func (t *table) Watch(opts ...WatchOption) (Watcher, error) { | ||||||
| 	// by default watch everything | 	// by default watch everything | ||||||
| 	wopts := WatchOptions{ | 	wopts := WatchOptions{ | ||||||
| 		DestAddr: "*", | 		Destination: "*", | ||||||
| 		Network:     "*", | 		Network:     "*", | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
| @@ -252,15 +256,15 @@ func (t *table) String() string { | |||||||
|  |  | ||||||
| 	// create nice table printing structure | 	// create nice table printing structure | ||||||
| 	table := tablewriter.NewWriter(sb) | 	table := tablewriter.NewWriter(sb) | ||||||
| 	table.SetHeader([]string{"Destination", "Gateway", "Network", "Metric"}) | 	table.SetHeader([]string{"Destination", "Router", "Network", "Metric"}) | ||||||
|  |  | ||||||
| 	for _, destRoute := range t.m { | 	for _, destRoute := range t.m { | ||||||
| 		for _, route := range destRoute { | 		for _, route := range destRoute { | ||||||
| 			strRoute := []string{ | 			strRoute := []string{ | ||||||
| 				route.Options().DestAddr, | 				route.Destination, | ||||||
| 				route.Options().Gateway.Address(), | 				route.Router.Address(), | ||||||
| 				route.Options().Network, | 				route.Network, | ||||||
| 				fmt.Sprintf("%d", route.Options().Metric), | 				fmt.Sprintf("%d", route.Metric), | ||||||
| 			} | 			} | ||||||
| 			table.Append(strRoute) | 			table.Append(strRoute) | ||||||
| 		} | 		} | ||||||
| @@ -274,12 +278,12 @@ func (t *table) String() string { | |||||||
|  |  | ||||||
| // hash hashes the route using router gateway and network address | // hash hashes the route using router gateway and network address | ||||||
| func (t *table) hash(r Route) uint64 { | func (t *table) hash(r Route) uint64 { | ||||||
| 	destAddr := r.Options().DestAddr | 	destAddr := r.Destination | ||||||
| 	gwAddr := r.Options().Gateway.Address() | 	routerAddr := r.Router.Address() | ||||||
| 	netAddr := r.Options().Network | 	netAddr := r.Network | ||||||
|  |  | ||||||
| 	t.h.Reset() | 	t.h.Reset() | ||||||
| 	t.h.Write([]byte(destAddr + gwAddr + netAddr)) | 	t.h.Write([]byte(destAddr + routerAddr + netAddr)) | ||||||
|  |  | ||||||
| 	return t.h.Sum64() | 	return t.h.Sum64() | ||||||
| } | } | ||||||
|   | |||||||
| @@ -6,31 +6,26 @@ import ( | |||||||
| ) | ) | ||||||
|  |  | ||||||
| var ( | var ( | ||||||
| 	// DefaultGossipAddress is default gossip bind address | 	// DefaultAddress is default router address | ||||||
| 	DefaultGossipAddress = ":9093" | 	DefaultAddress = ":9093" | ||||||
| 	// DefaultNetworkAddress is default network bind address | 	// DefaultAdvertise is default address advertised to the network | ||||||
| 	DefaultNetworkAddress = ":9094" | 	DefaultAdvertise = ":9094" | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // Options are router options | // Options are router options | ||||||
| type Options struct { | type Options struct { | ||||||
| 	// ID is router id | 	// ID is router id | ||||||
| 	ID string | 	ID string | ||||||
| 	// Address is router micro service address | 	// Address is router address | ||||||
| 	Address string | 	Address string | ||||||
| 	// GossipAddress is router gossip address | 	// Advertise is the address advertised to the network | ||||||
| 	GossipAddress string | 	Advertise string | ||||||
| 	// NetworkAddress is micro network address | 	// Registry is the local registry | ||||||
| 	NetworkAddress string | 	Registry registry.Registry | ||||||
| 	// LocalRegistry is router local registry | 	// Networkis the network registry | ||||||
| 	LocalRegistry registry.Registry | 	Network registry.Registry | ||||||
| 	// NetworkRegistry is router remote registry |  | ||||||
| 	// NOTE: we need some abstraction on top of gossip.Registry |  | ||||||
| 	NetworkRegistry registry.Registry |  | ||||||
| 	// Table is routing table | 	// Table is routing table | ||||||
| 	Table Table | 	Table Table | ||||||
| 	// RIB is Routing Information Base |  | ||||||
| 	RIB RIB |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // ID sets Router ID | // ID sets Router ID | ||||||
| @@ -47,59 +42,43 @@ func Address(a string) Option { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // GossipAddress sets router gossip address | // Advertise sets the address that is advertise to the network | ||||||
| func GossipAddress(a string) Option { | func Advertise(n string) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.GossipAddress = a | 		o.Advertise = n | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // NetworkAddress sets router network address | // RoutingTable sets the routing table | ||||||
| func NetworkAddress(n string) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.NetworkAddress = n |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // RoutingTable allows to specify custom routing table |  | ||||||
| func RoutingTable(t Table) Option { | func RoutingTable(t Table) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.Table = t | 		o.Table = t | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // LocalRegistry allows to specify local registry | // Registry sets the local registry | ||||||
| func LocalRegistry(r registry.Registry) Option { | func Registry(r registry.Registry) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.LocalRegistry = r | 		o.Registry = r | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // NetworkRegistry allows to specify remote registry | // Network sets the network registry | ||||||
| func NetworkRegistry(r registry.Registry) Option { | func Network(r registry.Registry) Option { | ||||||
| 	return func(o *Options) { | 	return func(o *Options) { | ||||||
| 		o.NetworkRegistry = r | 		o.Network = r | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // RouterRIB allows to configure RIB |  | ||||||
| func RouterRIB(r RIB) Option { |  | ||||||
| 	return func(o *Options) { |  | ||||||
| 		o.RIB = r |  | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // DefaultOptions returns router default options | // DefaultOptions returns router default options | ||||||
| func DefaultOptions() Options { | func DefaultOptions() Options { | ||||||
| 	// NOTE: by default both local and network registies use default registry i.e. mdns | 	// NOTE: by default both local and network registies use default registry i.e. mdns | ||||||
| 	// TODO: DefaultRIB needs to be added once it's properly figured out |  | ||||||
| 	return Options{ | 	return Options{ | ||||||
| 		ID:        uuid.New().String(), | 		ID:        uuid.New().String(), | ||||||
| 		Address:         ":8083", | 		Address:   DefaultAddress, | ||||||
| 		GossipAddress:   DefaultGossipAddress, | 		Advertise: DefaultAdvertise, | ||||||
| 		NetworkAddress:  DefaultNetworkAddress, | 		Registry:  registry.DefaultRegistry, | ||||||
| 		LocalRegistry:   registry.DefaultRegistry, | 		Network:   registry.DefaultRegistry, | ||||||
| 		NetworkRegistry: registry.DefaultRegistry, |  | ||||||
| 		Table:     NewTable(), | 		Table:     NewTable(), | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|   | |||||||
| @@ -27,20 +27,22 @@ type QueryOption func(*QueryOptions) | |||||||
|  |  | ||||||
| // QueryOptions are routing table query options | // QueryOptions are routing table query options | ||||||
| type QueryOptions struct { | type QueryOptions struct { | ||||||
| 	// DestAddr is destination address | 	// Destination is destination address | ||||||
| 	DestAddr string | 	Destination string | ||||||
| 	// NetworkAddress is network address | 	// Network is network address | ||||||
| 	Network string | 	Network string | ||||||
| 	// Gateway is gateway address | 	// Router is gateway address | ||||||
| 	Gateway Router | 	Router Router | ||||||
|  | 	// Metric is route metric | ||||||
|  | 	Metric int | ||||||
| 	// Policy is query lookup policy | 	// Policy is query lookup policy | ||||||
| 	Policy LookupPolicy | 	Policy LookupPolicy | ||||||
| } | } | ||||||
|  |  | ||||||
| // QueryDestAddr sets query destination address | // QueryDestination sets query destination address | ||||||
| func QueryDestAddr(a string) QueryOption { | func QueryDestination(a string) QueryOption { | ||||||
| 	return func(o *QueryOptions) { | 	return func(o *QueryOptions) { | ||||||
| 		o.DestAddr = a | 		o.Destination = a | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -51,14 +53,22 @@ func QueryNetwork(a string) QueryOption { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // QueryGateway sets query gateway address | // QueryRouter sets query gateway address | ||||||
| func QueryGateway(r Router) QueryOption { | func QueryRouter(r Router) QueryOption { | ||||||
| 	return func(o *QueryOptions) { | 	return func(o *QueryOptions) { | ||||||
| 		o.Gateway = r | 		o.Router = r | ||||||
|  | 	} | ||||||
|  | } | ||||||
|  |  | ||||||
|  | // QueryMetric sets query metric | ||||||
|  | func QueryMetric(m int) QueryOption { | ||||||
|  | 	return func(o *QueryOptions) { | ||||||
|  | 		o.Metric = m | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // QueryPolicy sets query policy | // QueryPolicy sets query policy | ||||||
|  | // NOTE: this might be renamed to filter or some such | ||||||
| func QueryPolicy(p LookupPolicy) QueryOption { | func QueryPolicy(p LookupPolicy) QueryOption { | ||||||
| 	return func(o *QueryOptions) { | 	return func(o *QueryOptions) { | ||||||
| 		o.Policy = p | 		o.Policy = p | ||||||
| @@ -82,10 +92,12 @@ func NewQuery(opts ...QueryOption) Query { | |||||||
| 	r := newRouter(ID("*")) | 	r := newRouter(ID("*")) | ||||||
|  |  | ||||||
| 	// default options | 	// default options | ||||||
|  | 	// NOTE: by default we use DefaultNetworkMetric | ||||||
| 	qopts := QueryOptions{ | 	qopts := QueryOptions{ | ||||||
| 		DestAddr: "*", | 		Destination: "*", | ||||||
| 		Network:     "*", | 		Network:     "*", | ||||||
| 		Gateway:  r, | 		Router:      r, | ||||||
|  | 		Metric:      DefaultNetworkMetric, | ||||||
| 		Policy:      DiscardNoRoute, | 		Policy:      DiscardNoRoute, | ||||||
| 	} | 	} | ||||||
|  |  | ||||||
|   | |||||||
| @@ -1,30 +0,0 @@ | |||||||
| package router |  | ||||||
|  |  | ||||||
| // RIB is Routing Information Base. |  | ||||||
| // RIB is used to source the base routing table. |  | ||||||
| type RIB interface { |  | ||||||
| 	// Initi initializes RIB |  | ||||||
| 	Init(...RIBOption) error |  | ||||||
| 	// Options returns RIB options |  | ||||||
| 	Options() RIBOptions |  | ||||||
| 	// Routes returns routes |  | ||||||
| 	Routes() []Route |  | ||||||
| 	// String returns debug info |  | ||||||
| 	String() string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // RIBOption sets RIB options |  | ||||||
| type RIBOption func(*RIBOptions) |  | ||||||
|  |  | ||||||
| // RIBOptions are RIB options |  | ||||||
| type RIBOptions struct { |  | ||||||
| 	// Source defines RIB source URL |  | ||||||
| 	Source string |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Source sets RIB source |  | ||||||
| func Source(s string) RIBOption { |  | ||||||
| 	return func(o *RIBOptions) { |  | ||||||
| 		o.Source = s |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
							
								
								
									
										106
									
								
								router/route.go
									
									
									
									
									
								
							
							
						
						
									
										106
									
								
								router/route.go
									
									
									
									
									
								
							| @@ -14,18 +14,18 @@ var ( | |||||||
| 	DefaultNetworkMetric = 10 | 	DefaultNetworkMetric = 10 | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // AddPolicy defines routing table addition policy | // RoutePolicy defines routing table addition policy | ||||||
| type AddPolicy int | type RoutePolicy int | ||||||
|  |  | ||||||
| const ( | const ( | ||||||
| 	// OverrideIfExists overrides route if it already exists | 	// OverrideIfExists overrides route if it already exists | ||||||
| 	OverrideIfExists AddPolicy = iota | 	OverrideIfExists RoutePolicy = iota | ||||||
| 	// IgnoreIfExists does not modify existing route | 	// IgnoreIfExists does not modify existing route | ||||||
| 	IgnoreIfExists | 	IgnoreIfExists | ||||||
| ) | ) | ||||||
|  |  | ||||||
| // String returns human reprensentation of policy | // String returns human reprensentation of policy | ||||||
| func (p AddPolicy) String() string { | func (p RoutePolicy) String() string { | ||||||
| 	switch p { | 	switch p { | ||||||
| 	case OverrideIfExists: | 	case OverrideIfExists: | ||||||
| 		return "OVERRIDE" | 		return "OVERRIDE" | ||||||
| @@ -36,100 +36,34 @@ func (p AddPolicy) String() string { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| // RouteOption is used to set routing table entry options | // Route is network route | ||||||
| type RouteOption func(*RouteOptions) | type Route struct { | ||||||
|  | 	// Destination is destination address | ||||||
| // RouteOptions are route options | 	Destination string | ||||||
| type RouteOptions struct { | 	// Router is the network router | ||||||
| 	// DestAddr is destination address | 	Router Router | ||||||
| 	DestAddr string | 	// Network is micro network address | ||||||
| 	// Gateway is the next route hop |  | ||||||
| 	Gateway Router |  | ||||||
| 	// Network defines micro network |  | ||||||
| 	Network string | 	Network string | ||||||
| 	// Metric is route cost metric | 	// Metric is the route cost metric | ||||||
| 	Metric int | 	Metric int | ||||||
| 	// Policy defines route addition policy | 	// Policy defines route policy | ||||||
| 	Policy AddPolicy | 	Policy RoutePolicy | ||||||
| } |  | ||||||
|  |  | ||||||
| // DestAddr sets destination address |  | ||||||
| func DestAddr(a string) RouteOption { |  | ||||||
| 	return func(o *RouteOptions) { |  | ||||||
| 		o.DestAddr = a |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Gateway sets the route gateway |  | ||||||
| func Gateway(r Router) RouteOption { |  | ||||||
| 	return func(o *RouteOptions) { |  | ||||||
| 		o.Gateway = r |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Network sets micro network |  | ||||||
| func Network(n string) RouteOption { |  | ||||||
| 	return func(o *RouteOptions) { |  | ||||||
| 		o.Network = n |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Metric sets route metric |  | ||||||
| func Metric(m int) RouteOption { |  | ||||||
| 	return func(o *RouteOptions) { |  | ||||||
| 		o.Metric = m |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // RoutePolicy sets add route policy |  | ||||||
| func RoutePolicy(p AddPolicy) RouteOption { |  | ||||||
| 	return func(o *RouteOptions) { |  | ||||||
| 		o.Policy = p |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Route is routing table route |  | ||||||
| type Route interface { |  | ||||||
| 	// Options returns route options |  | ||||||
| 	Options() RouteOptions |  | ||||||
| } |  | ||||||
|  |  | ||||||
| type route struct { |  | ||||||
| 	opts RouteOptions |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // NewRoute returns new routing table route |  | ||||||
| func NewRoute(opts ...RouteOption) Route { |  | ||||||
| 	eopts := RouteOptions{} |  | ||||||
|  |  | ||||||
| 	for _, o := range opts { |  | ||||||
| 		o(&eopts) |  | ||||||
| 	} |  | ||||||
|  |  | ||||||
| 	return &route{ |  | ||||||
| 		opts: eopts, |  | ||||||
| 	} |  | ||||||
| } |  | ||||||
|  |  | ||||||
| // Options returns route options |  | ||||||
| func (r *route) Options() RouteOptions { |  | ||||||
| 	return r.opts |  | ||||||
| } | } | ||||||
|  |  | ||||||
| // String allows to print the route | // String allows to print the route | ||||||
| func (r *route) String() string { | func (r *Route) String() string { | ||||||
| 	// this will help us build routing table string | 	// this will help us build routing table string | ||||||
| 	sb := &strings.Builder{} | 	sb := &strings.Builder{} | ||||||
|  |  | ||||||
| 	// create nice table printing structure | 	// create nice table printing structure | ||||||
| 	table := tablewriter.NewWriter(sb) | 	table := tablewriter.NewWriter(sb) | ||||||
| 	table.SetHeader([]string{"Destination", "Gateway", "Network", "Metric"}) | 	table.SetHeader([]string{"Destination", "Router", "Network", "Metric"}) | ||||||
|  |  | ||||||
| 	strRoute := []string{ | 	strRoute := []string{ | ||||||
| 		r.opts.DestAddr, | 		r.Destination, | ||||||
| 		r.opts.Gateway.Address(), | 		r.Router.Address(), | ||||||
| 		r.opts.Network, | 		r.Network, | ||||||
| 		fmt.Sprintf("%d", r.opts.Metric), | 		fmt.Sprintf("%d", r.Metric), | ||||||
| 	} | 	} | ||||||
| 	table.Append(strRoute) | 	table.Append(strRoute) | ||||||
|  |  | ||||||
|   | |||||||
| @@ -13,8 +13,6 @@ type Router interface { | |||||||
| 	Table() Table | 	Table() Table | ||||||
| 	// Address returns the router adddress | 	// Address returns the router adddress | ||||||
| 	Address() string | 	Address() string | ||||||
| 	// Gossip returns the router gossip address |  | ||||||
| 	Gossip() string |  | ||||||
| 	// Network returns the router network address | 	// Network returns the router network address | ||||||
| 	Network() string | 	Network() string | ||||||
| 	// Start starts the router | 	// Start starts the router | ||||||
|   | |||||||
| @@ -54,6 +54,8 @@ type WatchOption func(*WatchOptions) | |||||||
| type Watcher interface { | type Watcher interface { | ||||||
| 	// Next is a blocking call that returns watch result | 	// Next is a blocking call that returns watch result | ||||||
| 	Next() (*Event, error) | 	Next() (*Event, error) | ||||||
|  | 	// Chan returns event channel | ||||||
|  | 	Chan() (<-chan *Event, error) | ||||||
| 	// Stop stops watcher | 	// Stop stops watcher | ||||||
| 	Stop() | 	Stop() | ||||||
| } | } | ||||||
| @@ -61,16 +63,16 @@ type Watcher interface { | |||||||
| // WatchOptions are table watcher options | // WatchOptions are table watcher options | ||||||
| type WatchOptions struct { | type WatchOptions struct { | ||||||
| 	// Specify destination address to watch | 	// Specify destination address to watch | ||||||
| 	DestAddr string | 	Destination string | ||||||
| 	// Specify network to watch | 	// Specify network to watch | ||||||
| 	Network string | 	Network string | ||||||
| } | } | ||||||
|  |  | ||||||
| // WatchDestAddr sets what destination to watch | // WatchDestination sets what destination to watch | ||||||
| // Destination is usually microservice name | // Destination is usually microservice name | ||||||
| func WatchDestAddr(a string) WatchOption { | func WatchDestination(a string) WatchOption { | ||||||
| 	return func(o *WatchOptions) { | 	return func(o *WatchOptions) { | ||||||
| 		o.DestAddr = a | 		o.Destination = a | ||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
| @@ -94,13 +96,13 @@ func (w *tableWatcher) Next() (*Event, error) { | |||||||
| 	for { | 	for { | ||||||
| 		select { | 		select { | ||||||
| 		case res := <-w.resChan: | 		case res := <-w.resChan: | ||||||
| 			switch w.opts.DestAddr { | 			switch w.opts.Destination { | ||||||
| 			case "*", "": | 			case "*", "": | ||||||
| 				if w.opts.Network == "*" || w.opts.Network == res.Route.Options().Network { | 				if w.opts.Network == "*" || w.opts.Network == res.Route.Network { | ||||||
| 					return res, nil | 					return res, nil | ||||||
| 				} | 				} | ||||||
| 			case res.Route.Options().DestAddr: | 			case res.Route.Destination: | ||||||
| 				if w.opts.Network == "*" || w.opts.Network == res.Route.Options().Network { | 				if w.opts.Network == "*" || w.opts.Network == res.Route.Network { | ||||||
| 					return res, nil | 					return res, nil | ||||||
| 				} | 				} | ||||||
| 			} | 			} | ||||||
| @@ -110,6 +112,11 @@ func (w *tableWatcher) Next() (*Event, error) { | |||||||
| 	} | 	} | ||||||
| } | } | ||||||
|  |  | ||||||
|  | // Chan returns watcher events channel | ||||||
|  | func (w *tableWatcher) Chan() (<-chan *Event, error) { | ||||||
|  | 	return w.resChan, nil | ||||||
|  | } | ||||||
|  |  | ||||||
| // Stop stops routing table watcher | // Stop stops routing table watcher | ||||||
| func (w *tableWatcher) Stop() { | func (w *tableWatcher) Stop() { | ||||||
| 	select { | 	select { | ||||||
| @@ -125,10 +132,10 @@ func (w *tableWatcher) String() string { | |||||||
| 	sb := &strings.Builder{} | 	sb := &strings.Builder{} | ||||||
|  |  | ||||||
| 	table := tablewriter.NewWriter(sb) | 	table := tablewriter.NewWriter(sb) | ||||||
| 	table.SetHeader([]string{"DestAddr", "Network"}) | 	table.SetHeader([]string{"Destination", "Network"}) | ||||||
|  |  | ||||||
| 	data := []string{ | 	data := []string{ | ||||||
| 		w.opts.DestAddr, | 		w.opts.Destination, | ||||||
| 		w.opts.Network, | 		w.opts.Network, | ||||||
| 	} | 	} | ||||||
| 	table.Append(data) | 	table.Append(data) | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user