opendaylight/md-sal/inventory-manager/src/main/java/org/opendaylight/controller/md/inventory/manager/NodeChangeCommiter.java

   1 /**
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.md.inventory.manager;
   9
  10 import org.opendaylight.controller.sal.binding.api.data.DataModificationTransaction;
  11 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  12 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnector;
  13 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnectorUpdated;
  14 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeUpdated;
  15 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorRef;
  16 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorRemoved;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorUpdated;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRemoved;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeUpdated;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.OpendaylightInventoryListener;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorBuilder;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  28 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  29 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  30 import org.slf4j.Logger;
  31 import org.slf4j.LoggerFactory;
  32
  33 import com.google.common.base.Preconditions;
  34
  35 class NodeChangeCommiter implements OpendaylightInventoryListener {
  36     private static final Logger LOG = LoggerFactory.getLogger(NodeChangeCommiter.class);
  37
  38     private final FlowCapableInventoryProvider manager;
  39
  40     public NodeChangeCommiter(final FlowCapableInventoryProvider manager) {
  41         this.manager = Preconditions.checkNotNull(manager);
  42     }
  43
  44     @Override
  45     public synchronized void onNodeConnectorRemoved(final NodeConnectorRemoved connector) {
  46         manager.enqueue(new InventoryOperation() {
  47             @Override
  48             public void applyOperation(final DataModificationTransaction tx) {
  49                 final NodeConnectorRef ref = connector.getNodeConnectorRef();
  50                 LOG.debug("removing node connector {} ", ref.getValue());
  51                 tx.removeOperationalData(ref.getValue());
  52             }
  53         });
  54     }
  55
  56     @Override
  57     public synchronized void onNodeConnectorUpdated(final NodeConnectorUpdated connector) {
  58         manager.enqueue(new InventoryOperation() {
  59             @Override
  60             public void applyOperation(final DataModificationTransaction tx) {
  61                 final NodeConnectorRef ref = connector.getNodeConnectorRef();
  62                 final NodeConnectorBuilder data = new NodeConnectorBuilder(connector);
  63                 data.setKey(new NodeConnectorKey(connector.getId()));
  64
  65                 final FlowCapableNodeConnectorUpdated flowConnector = connector
  66                         .getAugmentation(FlowCapableNodeConnectorUpdated.class);
  67                 if (flowConnector != null) {
  68                     final FlowCapableNodeConnector augment = InventoryMapping.toInventoryAugment(flowConnector);
  69                     data.addAugmentation(FlowCapableNodeConnector.class, augment);
  70                 }
  71                 InstanceIdentifier<? extends Object> value = ref.getValue();
  72                 LOG.debug("updating node connector : {}.", value);
  73                 NodeConnector build = data.build();
  74                 tx.putOperationalData(value, build);
  75             }
  76         });
  77     }
  78
  79     @Override
  80     public synchronized void onNodeRemoved(final NodeRemoved node) {
  81         manager.enqueue(new InventoryOperation() {
  82             @Override
  83             public void applyOperation(final DataModificationTransaction tx) {
  84                 final NodeRef ref = node.getNodeRef();
  85                 LOG.debug("removing node : {}", ref.getValue());
  86                 tx.removeOperationalData((ref.getValue()));
  87             }
  88         });
  89     }
  90
  91     @Override
  92     public synchronized void onNodeUpdated(final NodeUpdated node) {
  93         final FlowCapableNodeUpdated flowNode = node.getAugmentation(FlowCapableNodeUpdated.class);
  94         if (flowNode == null) {
  95             return;
  96         }
  97
  98         manager.enqueue(new InventoryOperation() {
  99             @Override
 100             public void applyOperation(final DataModificationTransaction tx) {
 101                 final NodeRef ref = node.getNodeRef();
 102                 final NodeBuilder nodeBuilder = new NodeBuilder(node);
 103                 nodeBuilder.setKey(new NodeKey(node.getId()));
 104
 105                 final FlowCapableNode augment = InventoryMapping.toInventoryAugment(flowNode);
 106                 nodeBuilder.addAugmentation(FlowCapableNode.class, augment);
 107
 108                 @SuppressWarnings("unchecked")
 109                 InstanceIdentifierBuilder<Node> builder = ((InstanceIdentifier<Node>) ref.getValue()).builder();
 110                 InstanceIdentifierBuilder<FlowCapableNode> augmentation = builder.augmentation(FlowCapableNode.class);
 111                 final InstanceIdentifier<FlowCapableNode> path = augmentation.build();
 112                 LOG.debug("updating node :{} ", path);
 113                 tx.putOperationalData(path, augment);
 114             }
 115         });
 116     }
 117 }