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 com.google.common.base.Optional;
  11 import com.google.common.base.Preconditions;
  12 import com.google.common.util.concurrent.CheckedFuture;
  13 import com.google.common.util.concurrent.FutureCallback;
  14 import com.google.common.util.concurrent.Futures;
  15 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  16 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNode;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnector;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeConnectorUpdated;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.FlowCapableNodeUpdated;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.TableKey;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorRef;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorRemoved;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorUpdated;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRemoved;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeUpdated;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.OpendaylightInventoryListener;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorBuilder;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeBuilder;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  37 import org.opendaylight.yangtools.yang.binding.DataObject;
  38 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
  40 import org.slf4j.Logger;
  41 import org.slf4j.LoggerFactory;
  42
  43 class NodeChangeCommiter implements OpendaylightInventoryListener {
  44     private static final Logger LOG = LoggerFactory.getLogger(NodeChangeCommiter.class);
  45
  46     private final FlowCapableInventoryProvider manager;
  47
  48     public NodeChangeCommiter(final FlowCapableInventoryProvider manager) {
  49         this.manager = Preconditions.checkNotNull(manager);
  50     }
  51
  52     @Override
  53     public synchronized void onNodeConnectorRemoved(final NodeConnectorRemoved connector) {
  54         LOG.debug("Node connector removed notification received.");
  55         manager.enqueue(new InventoryOperation() {
  56             @Override
  57             public void applyOperation(final ReadWriteTransaction tx) {
  58                 final NodeConnectorRef ref = connector.getNodeConnectorRef();
  59                 LOG.debug("removing node connector {} ", ref.getValue());
  60                 tx.delete(LogicalDatastoreType.OPERATIONAL, ref.getValue());
  61             }
  62         });
  63     }
  64
  65     @Override
  66     public synchronized void onNodeConnectorUpdated(final NodeConnectorUpdated connector) {
  67         LOG.debug("Node connector updated notification received.");
  68         manager.enqueue(new InventoryOperation() {
  69             @Override
  70             public void applyOperation(final ReadWriteTransaction tx) {
  71                 final NodeConnectorRef ref = connector.getNodeConnectorRef();
  72                 final NodeConnectorBuilder data = new NodeConnectorBuilder(connector);
  73                 data.setKey(new NodeConnectorKey(connector.getId()));
  74
  75                 final FlowCapableNodeConnectorUpdated flowConnector = connector
  76                         .getAugmentation(FlowCapableNodeConnectorUpdated.class);
  77                 if (flowConnector != null) {
  78                     final FlowCapableNodeConnector augment = InventoryMapping.toInventoryAugment(flowConnector);
  79                     data.addAugmentation(FlowCapableNodeConnector.class, augment);
  80                 }
  81                 InstanceIdentifier<NodeConnector> value = (InstanceIdentifier<NodeConnector>) ref.getValue();
  82                 LOG.debug("updating node connector : {}.", value);
  83                 NodeConnector build = data.build();
  84                 tx.put(LogicalDatastoreType.OPERATIONAL, value, build);
  85             }
  86         });
  87     }
  88
  89     @Override
  90     public synchronized void onNodeRemoved(final NodeRemoved node) {
  91         LOG.debug("Node removed notification received.");
  92         manager.enqueue(new InventoryOperation() {
  93             @Override
  94             public void applyOperation(final ReadWriteTransaction tx) {
  95                 final NodeRef ref = node.getNodeRef();
  96                 LOG.debug("removing node : {}", ref.getValue());
  97                 tx.delete(LogicalDatastoreType.OPERATIONAL, ref.getValue());
  98             }
  99         });
 100     }
 101
 102     @Override
 103     public synchronized void onNodeUpdated(final NodeUpdated node) {
 104         final FlowCapableNodeUpdated flowNode = node.getAugmentation(FlowCapableNodeUpdated.class);
 105         if (flowNode == null) {
 106             return;
 107         }
 108         LOG.debug("Node updated notification received.");
 109         manager.enqueue(new InventoryOperation() {
 110             @Override
 111             public void applyOperation(ReadWriteTransaction tx) {
 112                 final NodeRef ref = node.getNodeRef();
 113                 @SuppressWarnings("unchecked")
 114                 InstanceIdentifierBuilder<Node> builder = ((InstanceIdentifier<Node>) ref.getValue()).builder();
 115                 InstanceIdentifierBuilder<FlowCapableNode> augmentation = builder.augmentation(FlowCapableNode.class);
 116                 final InstanceIdentifier<FlowCapableNode> path = augmentation.build();
 117                 CheckedFuture readFuture = tx.read(LogicalDatastoreType.OPERATIONAL, path);
 118                 Futures.addCallback(readFuture, new FutureCallback<Optional<? extends DataObject>>() {
 119                     @Override
 120                     public void onSuccess(Optional<? extends DataObject> optional) {
 121                         enqueueWriteNodeDataTx(node, flowNode, path);
 122                         if (!optional.isPresent()) {
 123                             enqueuePutTable0Tx(ref);
 124                         }
 125                     }
 126
 127                     @Override
 128                     public void onFailure(Throwable throwable) {
 129                         LOG.debug(String.format("Can't retrieve node data for node %s. Writing node data with table0.", node));
 130                         enqueueWriteNodeDataTx(node, flowNode, path);
 131                         enqueuePutTable0Tx(ref);
 132                     }
 133                 });
 134             }
 135         });
 136     }
 137
 138     private void enqueueWriteNodeDataTx(final NodeUpdated node, final FlowCapableNodeUpdated flowNode, final InstanceIdentifier<FlowCapableNode> path) {
 139         manager.enqueue(new InventoryOperation() {
 140             @Override
 141             public void applyOperation(final ReadWriteTransaction tx) {
 142                 final NodeBuilder nodeBuilder = new NodeBuilder(node);
 143                 nodeBuilder.setKey(new NodeKey(node.getId()));
 144
 145                 final FlowCapableNode augment = InventoryMapping.toInventoryAugment(flowNode);
 146                 nodeBuilder.addAugmentation(FlowCapableNode.class, augment);
 147                 LOG.debug("updating node :{} ", path);
 148                 tx.put(LogicalDatastoreType.OPERATIONAL, path, augment);
 149             }
 150         });
 151     }
 152
 153     private void enqueuePutTable0Tx(final NodeRef ref) {
 154         manager.enqueue(new InventoryOperation() {
 155             @Override
 156             public void applyOperation(ReadWriteTransaction tx) {
 157                 final TableKey tKey = new TableKey((short) 0);
 158                 final InstanceIdentifier<Table> tableIdentifier =
 159                         ((InstanceIdentifier<Node>) ref.getValue()).augmentation(FlowCapableNode.class).child(Table.class, new TableKey(tKey));
 160                 TableBuilder tableBuilder = new TableBuilder();
 161                 Table table0 = tableBuilder.setId((short) 0).build();
 162                 LOG.debug("writing table :{} ", tableIdentifier);
 163                 tx.put(LogicalDatastoreType.OPERATIONAL, tableIdentifier, table0, true);
 164             }
 165         });
 166     }
 167 }