fix ArbitratorReconciliationManagerImpl endReconciliation() remove

JIRA: OPNFLWPLUG-1048
Change-Id: If26656fa6f65d04435c94c0b7258144f45dc989e
Signed-off-by: Michael Vorburger <vorburger@redhat.com>

diff --git a/applications/arbitratorreconciliation/impl/src/main/java/org/opendaylight/openflowplugin/applications/arbitratorreconciliation/impl/ArbitratorReconciliationManagerImpl.java b/applications/arbitratorreconciliation/impl/src/main/java/org/opendaylight/openflowplugin/applications/arbitratorreconciliation/impl/ArbitratorReconciliationManagerImpl.java
index 55486918a04ccf1f6e102086a39ba66cfdadb16f..9ff3c889dad462198037166e13c203bd8bda9231 100644 (file)
--- a/applications/arbitratorreconciliation/impl/src/main/java/org/opendaylight/openflowplugin/applications/arbitratorreconciliation/impl/ArbitratorReconciliationManagerImpl.java
+++ b/applications/arbitratorreconciliation/impl/src/main/java/org/opendaylight/openflowplugin/applications/arbitratorreconciliation/impl/ArbitratorReconciliationManagerImpl.java
@@ -197,10 +197,9 @@ public class ArbitratorReconciliationManagerImpl implements ArbitratorReconcileS
 
     @Override
     public ListenableFuture<Boolean> endReconciliation(DeviceInfo node) {
-        LOG.trace("Stopping arbitrator reconciliation for node {}", node.getDatapathId());
-        InstanceIdentifier<FlowCapableNode> connectedNode = node.getNodeInstanceIdentifier()
-                .augmentation(FlowCapableNode.class);
-        bundleIdMap.remove(connectedNode);
+        BigInteger datapathId = node.getDatapathId();
+        LOG.trace("Stopping arbitrator reconciliation for node {}", datapathId);
+        bundleIdMap.remove(datapathId);
         deregisterRpc(node);
         return Futures.immediateFuture(true);
     }