Enhance the RouteEvent to notify listeners of alternative viable routes.
Traditionally the route event only notified listeners of the best selected
route for a given prefix, but some listeners are interested in all resolved
routes for the prefix.
CORD-905
Change-Id: Ia3e1e3a8e3e825ba894e6835e0860c3ed698d29b
diff --git a/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/DefaultResolvedRouteStore.java b/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/DefaultResolvedRouteStore.java
index d3f97b9..fa07d14 100644
--- a/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/DefaultResolvedRouteStore.java
+++ b/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/DefaultResolvedRouteStore.java
@@ -16,6 +16,8 @@
package org.onosproject.incubator.net.routing.impl;
+import com.google.common.collect.ImmutableSet;
+import com.google.common.collect.Maps;
import com.googlecode.concurrenttrees.common.KeyValuePair;
import com.googlecode.concurrenttrees.radix.node.concrete.DefaultByteArrayNodeFactory;
import com.googlecode.concurrenttrees.radixinverted.ConcurrentInvertedRadixTree;
@@ -57,8 +59,8 @@
}
@Override
- public RouteEvent updateRoute(ResolvedRoute route) {
- return getDefaultRouteTable(route).update(route);
+ public RouteEvent updateRoute(ResolvedRoute route, Set<ResolvedRoute> alternatives) {
+ return getDefaultRouteTable(route).update(route, alternatives);
}
@Override
@@ -87,6 +89,11 @@
}
@Override
+ public Collection<ResolvedRoute> getAllRoutes(IpPrefix prefix) {
+ return getDefaultRouteTable(prefix.address()).getAllRoutes(prefix);
+ }
+
+ @Override
public Optional<ResolvedRoute> longestPrefixMatch(IpAddress ip) {
return getDefaultRouteTable(ip).longestPrefixMatch(ip);
}
@@ -105,6 +112,7 @@
*/
private class RouteTable {
private final InvertedRadixTree<ResolvedRoute> routeTable;
+ private final Map<IpPrefix, Set<ResolvedRoute>> alternativeRoutes;
/**
* Creates a new route table.
@@ -112,27 +120,58 @@
public RouteTable() {
routeTable = new ConcurrentInvertedRadixTree<>(
new DefaultByteArrayNodeFactory());
+
+ alternativeRoutes = Maps.newHashMap();
}
/**
* Adds or updates the route in the route table.
*
* @param route route to update
+ * @param alternatives alternative routes
*/
- public RouteEvent update(ResolvedRoute route) {
+ public RouteEvent update(ResolvedRoute route, Set<ResolvedRoute> alternatives) {
+ Set<ResolvedRoute> immutableAlternatives = checkAlternatives(route, alternatives);
+
synchronized (this) {
ResolvedRoute oldRoute = routeTable.put(createBinaryString(route.prefix()), route);
+ Set<ResolvedRoute> oldRoutes = alternativeRoutes.put(route.prefix(), immutableAlternatives);
- // No need to proceed if the new route is the same
- if (route.equals(oldRoute)) {
- return null;
+ if (!route.equals(oldRoute)) {
+ if (oldRoute == null) {
+ return new RouteEvent(RouteEvent.Type.ROUTE_ADDED, route,
+ immutableAlternatives);
+ } else {
+ return new RouteEvent(RouteEvent.Type.ROUTE_UPDATED, route,
+ oldRoute, immutableAlternatives);
+ }
}
- if (oldRoute == null) {
- return new RouteEvent(RouteEvent.Type.ROUTE_ADDED, route);
- } else {
- return new RouteEvent(RouteEvent.Type.ROUTE_UPDATED, route, oldRoute);
+ if (!immutableAlternatives.equals(oldRoutes)) {
+ return new RouteEvent(RouteEvent.Type.ALTERNATIVE_ROUTES_CHANGED,
+ route, immutableAlternatives);
}
+
+ return null;
+ }
+ }
+
+ /**
+ * Checks that the best route is present in the alternatives list and
+ * returns an immutable set of alternatives.
+ *
+ * @param route best route
+ * @param alternatives alternatives
+ * @return immutable set of alternative routes
+ */
+ private Set<ResolvedRoute> checkAlternatives(ResolvedRoute route, Set<ResolvedRoute> alternatives) {
+ if (!alternatives.contains(route)) {
+ return ImmutableSet.<ResolvedRoute>builder()
+ .addAll(alternatives)
+ .add(route)
+ .build();
+ } else {
+ return ImmutableSet.copyOf(alternatives);
}
}
@@ -146,6 +185,7 @@
String key = createBinaryString(prefix);
ResolvedRoute route = routeTable.getValueForExactKey(key);
+ alternativeRoutes.remove(prefix);
if (route != null) {
routeTable.remove(key);
@@ -176,6 +216,10 @@
return Optional.ofNullable(routeTable.getValueForExactKey(createBinaryString(prefix)));
}
+ public Collection<ResolvedRoute> getAllRoutes(IpPrefix prefix) {
+ return alternativeRoutes.getOrDefault(prefix, Collections.emptySet());
+ }
+
/**
* Performs a longest prefix match with the given IP in the route table.
*
diff --git a/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/ResolvedRouteStore.java b/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/ResolvedRouteStore.java
index 55ca3f9..2285572 100644
--- a/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/ResolvedRouteStore.java
+++ b/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/ResolvedRouteStore.java
@@ -35,9 +35,10 @@
* Adds or updates the best route for the given prefix.
*
* @param route new best route for this prefix
+ * @param alternatives alternative resolved routes
* @return event describing the change
*/
- RouteEvent updateRoute(ResolvedRoute route);
+ RouteEvent updateRoute(ResolvedRoute route, Set<ResolvedRoute> alternatives);
/**
* Removes the best route for the given prefix.
@@ -71,6 +72,15 @@
Optional<ResolvedRoute> getRoute(IpPrefix prefix);
/**
+ * Returns all resolved routes stored for the given prefix, including the
+ * best selected route.
+ *
+ * @param prefix IP prefix to look up routes for
+ * @return all stored resolved routes for this prefix
+ */
+ Collection<ResolvedRoute> getAllRoutes(IpPrefix prefix);
+
+ /**
* Performs a longest prefix match of the best routes on the given IP address.
*
* @param ip IP address
diff --git a/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/RouteManager.java b/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/RouteManager.java
index 5b637b8..539b2cb 100644
--- a/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/RouteManager.java
+++ b/incubator/net/src/main/java/org/onosproject/incubator/net/routing/impl/RouteManager.java
@@ -23,7 +23,7 @@
import org.apache.felix.scr.annotations.ReferenceCardinality;
import org.apache.felix.scr.annotations.Service;
import org.onlab.packet.IpAddress;
-import org.onosproject.event.ListenerService;
+import org.onlab.packet.IpPrefix;
import org.onosproject.incubator.net.routing.InternalRouteEvent;
import org.onosproject.incubator.net.routing.NextHop;
import org.onosproject.incubator.net.routing.ResolvedRoute;
@@ -68,8 +68,7 @@
*/
@Service
@Component
-public class RouteManager implements ListenerService<RouteEvent, RouteListener>,
- RouteService, RouteAdminService {
+public class RouteManager implements RouteService, RouteAdminService {
private final Logger log = LoggerFactory.getLogger(getClass());
@@ -129,7 +128,8 @@
resolvedRouteStore.getRouteTables().stream()
.map(resolvedRouteStore::getRoutes)
.flatMap(Collection::stream)
- .map(route -> new RouteEvent(RouteEvent.Type.ROUTE_ADDED, route))
+ .map(route -> new RouteEvent(RouteEvent.Type.ROUTE_ADDED, route,
+ resolvedRouteStore.getAllRoutes(route.prefix())))
.forEach(l::post);
listeners.put(listener, l);
@@ -259,25 +259,31 @@
}
private ResolvedRoute decide(ResolvedRoute route1, ResolvedRoute route2) {
- return Comparator.<ResolvedRoute, IpAddress>comparing(route -> route.nextHop())
+ return Comparator.comparing(ResolvedRoute::nextHop)
.compare(route1, route2) <= 0 ? route1 : route2;
}
- private void store(ResolvedRoute route) {
- post(resolvedRouteStore.updateRoute(route));
+ private void store(ResolvedRoute route, Set<ResolvedRoute> alternatives) {
+ post(resolvedRouteStore.updateRoute(route, alternatives));
+ }
+
+ private void remove(IpPrefix prefix) {
+ post(resolvedRouteStore.removeRoute(prefix));
}
private void resolve(RouteSet routes) {
- Optional<ResolvedRoute> resolvedRoute =
- routes.routes().stream()
- .map(this::resolve)
- .filter(Objects::nonNull)
+ Set<ResolvedRoute> resolvedRoutes = routes.routes().stream()
+ .map(this::resolve)
+ .filter(Objects::nonNull)
+ .collect(Collectors.toSet());
+
+ Optional<ResolvedRoute> bestRoute = resolvedRoutes.stream()
.reduce(this::decide);
- if (resolvedRoute.isPresent()) {
- this.store(resolvedRoute.get());
+ if (bestRoute.isPresent()) {
+ store(bestRoute.get(), resolvedRoutes);
} else {
- post(resolvedRouteStore.removeRoute(routes.prefix()));
+ remove(routes.prefix());
}
}