Clustering support for existing design (FRM,IM,SM) - add retry mechanisms to avoid...

[openflowplugin.git] / applications / statistics-manager / src / main / java / org / opendaylight / openflowplugin / applications / statistics / manager / impl / StatPermCollectorImpl.java

diff --git a/applications/statistics-manager/src/main/java/org/opendaylight/openflowplugin/applications/statistics/manager/impl/StatPermCollectorImpl.java b/applications/statistics-manager/src/main/java/org/opendaylight/openflowplugin/applications/statistics/manager/impl/StatPermCollectorImpl.java
index 32c19e7187d2a9b14cb863cf033521bdd943cd55..a740c21c4bc9dedfccd324ad47cdfe7ba9f0430a 100644 (file)
--- a/applications/statistics-manager/src/main/java/org/opendaylight/openflowplugin/applications/statistics/manager/impl/StatPermCollectorImpl.java
+++ b/applications/statistics-manager/src/main/java/org/opendaylight/openflowplugin/applications/statistics/manager/impl/StatPermCollectorImpl.java
@@ -19,11 +19,16 @@ import java.util.concurrent.ExecutorService;
 import java.util.concurrent.Executors;
 import java.util.concurrent.ThreadFactory;
 
+import com.google.common.base.Optional;
+import org.opendaylight.controller.md.sal.common.api.clustering.Entity;
+import org.opendaylight.controller.md.sal.common.api.clustering.EntityOwnershipState;
 import org.opendaylight.openflowplugin.applications.statistics.manager.StatPermCollector;
 import org.opendaylight.openflowplugin.applications.statistics.manager.StatisticsManager;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.transaction.rev150304.TransactionId;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.table.types.rev131026.TableId;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 import org.slf4j.Logger;
@@ -257,6 +262,13 @@ public class StatPermCollectorImpl implements StatPermCollector {
 
     private void collectStatCrossNetwork() {
         for (final Entry<InstanceIdentifier<Node>, StatNodeInfoHolder> nodeEntity : statNodeHolder.entrySet()) {
+            final NodeKey nodeKey = nodeEntity.getKey().firstKeyOf(Node.class);
+            if (!this.isThisInstanceNodeOwner(nodeKey.getId())) {
+                continue;
+            }
+            LOG.trace("collectStatCrossNetwork: Controller is owner of the " +
+                    "node {}, so collecting the statistics.",nodeKey);
+
             final List<StatCapabTypes> listNeededStat = nodeEntity.getValue().getStatMarkers();
             final NodeRef actualNodeRef = nodeEntity.getValue().getNodeRef();
             final Short maxTables = nodeEntity.getValue().getMaxTables();
@@ -317,6 +329,24 @@ public class StatPermCollectorImpl implements StatPermCollector {
         }
     }
 
+    private boolean isThisInstanceNodeOwner(NodeId nodeId) {
+        final Entity deviceEntity = new Entity("openflow",nodeId.getValue());
+        if(manager.getOwnershipService().isCandidateRegistered(deviceEntity)) {
+            Optional<EntityOwnershipState> deviceOwnershipState = manager.getOwnershipService()
+                    .getOwnershipState(deviceEntity);
+
+            if(deviceOwnershipState.isPresent()) {
+                return deviceOwnershipState.get().isOwner();
+            } else {
+                LOG.error("Node {} is connected to the controller but ownership state is missing.");
+            }
+        } else {
+            LOG.warn("Node {} is connected to the controller but it did not" +
+                    "registered for the device ownership.",nodeId);
+        }
+        return false;
+    }
+
     private class StatNodeInfoHolder {
         private final NodeRef nodeRef;
         private final List<StatCapabTypes> statMarkers;