commit a9f8ae4f84d0c10b3b21c48ffb25be0315d8ecc5
author Charles Chan <rascov@gmail.com> Fri Mar 17 18:36:26 2017 -0700
committer Jonathan Hart <jono@onlab.us> Wed Mar 22 20:33:09 2017 +0000
tree c3788ee1486ad60459c9d1e493a0779cb230ce1a
parent f2e7a34ae826f30f5f8fe5b2bd7925186c6fd662

Support tagged next hop in Segment Routing

Change-Id: I4e1a77b797a487a647c1b8e437ba8426f84a5ae0

apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/RouteHandler.java

diff --git a/apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/RouteHandler.java b/apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/RouteHandler.java
index 6fe9127..d97b5ac 100644
--- a/apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/RouteHandler.java
+++ b/apps/segmentrouting/src/main/java/org/onosproject/segmentrouting/RouteHandler.java
@@ -39,14 +39,13 @@
     }
 
     protected void init(DeviceId deviceId) {
-        srManager.routeService.getNextHops().forEach(nextHop -> {
-            if (nextHop.location().deviceId().equals(deviceId)) {
-                srManager.routeService.getRoutesForNextHop(nextHop.ip()).forEach(route -> {
-                    ResolvedRoute resolvedRoute =
-                            new ResolvedRoute(route, nextHop.mac(), nextHop.location());
-                    processRouteAddedInternal(resolvedRoute);
-                });
-            }
+        srManager.routeService.getRouteTables().forEach(routeTableId -> {
+            srManager.routeService.getRoutes(routeTableId).forEach(routeInfo -> {
+                routeInfo.allRoutes().stream()
+                        .filter(resolvedRoute -> resolvedRoute.location() != null &&
+                                resolvedRoute.location().deviceId().equals(deviceId))
+                        .forEach(this::processRouteAddedInternal);
+            });
         });
     }
 
@@ -58,9 +57,7 @@
     private void processRouteAddedInternal(ResolvedRoute route) {
         IpPrefix prefix = route.prefix();
         MacAddress nextHopMac = route.nextHopMac();
-        // TODO ResolvedRoute does not contain VLAN information.
-        //      Therefore we only support untagged nexthop for now.
-        VlanId nextHopVlan = VlanId.NONE;
+        VlanId nextHopVlan = route.nextHopVlan();
         ConnectPoint location = route.location();
 
         srManager.deviceConfiguration.addSubnet(location, prefix);
@@ -83,9 +80,7 @@
     private void processRouteRemovedInternal(ResolvedRoute route) {
         IpPrefix prefix = route.prefix();
         MacAddress nextHopMac = route.nextHopMac();
-        // TODO ResolvedRoute does not contain VLAN information.
-        //      Therefore we only support untagged nexthop for now.
-        VlanId nextHopVlan = VlanId.NONE;
+        VlanId nextHopVlan = route.nextHopVlan();
         ConnectPoint location = route.location();
 
         srManager.deviceConfiguration.removeSubnet(location, prefix);