Fixed main issues of SwitchManager
[groupbasedpolicy.git] / renderers / ofoverlay / src / main / java / org / opendaylight / groupbasedpolicy / renderer / ofoverlay / node / FlowCapableNodeListener.java
1 package org.opendaylight.groupbasedpolicy.renderer.ofoverlay.node;
2
3 import static com.google.common.base.Preconditions.checkNotNull;
4
5 import java.util.Map.Entry;
6
7 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
8 import org.opendaylight.controller.md.sal.binding.api.DataChangeListener;
9 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker.DataChangeScope;
10 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
11 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
12 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
13 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
14 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
17 import org.opendaylight.yangtools.concepts.ListenerRegistration;
18 import org.opendaylight.yangtools.yang.binding.DataObject;
19 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
20 import org.slf4j.Logger;
21 import org.slf4j.LoggerFactory;
22
23 public class FlowCapableNodeListener implements DataChangeListener, AutoCloseable {
24
25     private static final Logger LOG = LoggerFactory.getLogger(FlowCapableNodeListener.class);
26
27     private final static InstanceIdentifier<FlowCapableNode> fcNodeIid = InstanceIdentifier.builder(Nodes.class)
28         .child(Node.class)
29         .augmentation(FlowCapableNode.class)
30         .build();
31     private final SwitchManager switchManager;
32     private final ListenerRegistration<DataChangeListener> listenerRegistration;
33
34     public FlowCapableNodeListener(DataBroker dataProvider, SwitchManager switchManager) {
35         this.switchManager = checkNotNull(switchManager);
36         listenerRegistration = checkNotNull(dataProvider).registerDataChangeListener(LogicalDatastoreType.OPERATIONAL,
37                 fcNodeIid, this, DataChangeScope.BASE);
38     }
39
40     @Override
41     public void onDataChanged(AsyncDataChangeEvent<InstanceIdentifier<?>, DataObject> change) {
42         for (Entry<InstanceIdentifier<?>, DataObject> fcNodeEntry : change.getCreatedData().entrySet()) {
43             if (FlowCapableNode.class.equals(fcNodeEntry.getKey().getTargetType())) {
44                 NodeId nodeId = fcNodeEntry.getKey().firstKeyOf(Node.class, NodeKey.class).getId();
45                 FlowCapableNode fcNode = (FlowCapableNode) fcNodeEntry.getValue();
46                 LOG.trace("FlowCapableNode created. NodeId: {} FlowCapableNode: {}", nodeId.getValue(), fcNode);
47                 switchManager.updateSwitch(nodeId, fcNode);
48             }
49         }
50         for (Entry<InstanceIdentifier<?>, DataObject> fcNodeEntry : change.getUpdatedData().entrySet()) {
51             if (FlowCapableNode.class.equals(fcNodeEntry.getKey().getTargetType())) {
52                 NodeId nodeId = fcNodeEntry.getKey().firstKeyOf(Node.class, NodeKey.class).getId();
53                 FlowCapableNode fcNode = (FlowCapableNode) fcNodeEntry.getValue();
54                 LOG.trace("FlowCapableNode updated. NodeId: {} FlowCapableNode: {}", nodeId.getValue(), fcNode);
55                 switchManager.updateSwitch(nodeId, fcNode);
56             }
57         }
58         for (InstanceIdentifier<?> removedFcNodeIid : change.getRemovedPaths()) {
59             if (FlowCapableNode.class.equals(removedFcNodeIid.getTargetType())) {
60                 NodeId nodeId = removedFcNodeIid.firstKeyOf(Node.class, NodeKey.class).getId();
61                 LOG.trace("FlowCapableNode removed. NodeId: {}", nodeId.getValue());
62                 switchManager.updateSwitch(nodeId, null);
63             }
64         }
65     }
66
67     @Override
68     public void close() throws Exception {
69         if (listenerRegistration != null) {
70             listenerRegistration.close();
71         }
72     }
73
74 }