Added Removal of bridges from operational store
[ovsdb.git] / southbound / southbound-impl / src / main / java / org / opendaylight / ovsdb / southbound / transactions / md / OvsdbBridgeUpdateCommand.java
1 package org.opendaylight.ovsdb.southbound.transactions.md;
2
3 import java.util.ArrayList;
4 import java.util.List;
5
6 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
7 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
8 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
9 import org.opendaylight.ovsdb.lib.message.TableUpdates;
10 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
11 import org.opendaylight.ovsdb.lib.schema.typed.TypedBaseTable;
12 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
13 import org.opendaylight.ovsdb.southbound.OvsdbClientKey;
14 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
15 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeRef;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbManagedNodeAugmentation;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbManagedNodeAugmentationBuilder;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentationBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeRef;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntry;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntryBuilder;
24 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
25 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
26 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
30
31 import com.google.common.base.Optional;
32
33 public class OvsdbBridgeUpdateCommand implements TransactionCommand {
34     private static final Logger LOG = LoggerFactory.getLogger(OvsdbBridgeUpdateCommand.class);
35
36     private TableUpdates updates;
37     private DatabaseSchema dbSchema;
38
39     private OvsdbClientKey key;
40
41     public OvsdbBridgeUpdateCommand(OvsdbClientKey key,TableUpdates updates, DatabaseSchema dbSchema) {
42         this.updates = updates;
43         this.dbSchema = dbSchema;
44         this.key = key;
45     }
46     @Override
47     public void execute(ReadWriteTransaction transaction) {
48         List<TypedBaseTable<?>> updatedRows = TransactionUtils.extractRowsUpdated(Bridge.class, updates, dbSchema);
49         for(TypedBaseTable<?> updatedRow : updatedRows) {
50             if(updatedRow instanceof Bridge) {
51                 Bridge bridge = (Bridge)updatedRow;
52                 final InstanceIdentifier<Node> nodePath = key.toInstanceIndentifier();
53                 Optional<Node> node = Optional.absent();
54                 try{
55                     node = transaction.read(LogicalDatastoreType.OPERATIONAL, nodePath).checkedGet();
56                 }catch (final ReadFailedException e) {
57                     LOG.debug("Read Operational/DS for Node fail! {}", nodePath, e);
58                 }
59                 if(node.isPresent()){
60                     LOG.info("Node {} is present",node);
61                     NodeBuilder managedNodeBuilder = new NodeBuilder();
62                     NodeId manageNodeId = SouthboundMapper.createManagedNodeId(key, bridge.getUuid());
63                     managedNodeBuilder.setNodeId(manageNodeId);
64
65                     OvsdbManagedNodeAugmentationBuilder ovsdbManagedNodeBuilder = new OvsdbManagedNodeAugmentationBuilder();
66                     ovsdbManagedNodeBuilder.setBridgeName(bridge.getName());
67                     ovsdbManagedNodeBuilder.setBridgeUuid(new Uuid(bridge.getUuid().toString()));
68                     ovsdbManagedNodeBuilder.setManagedBy(new OvsdbNodeRef(nodePath));
69                     managedNodeBuilder.addAugmentation(OvsdbManagedNodeAugmentation.class, ovsdbManagedNodeBuilder.build());
70
71                     InstanceIdentifier<Node> managedNodePath = SouthboundMapper.createInstanceIdentifier(manageNodeId);
72
73                     LOG.debug("Store managed node augmentation data {}",ovsdbManagedNodeBuilder.toString());
74                     transaction.put(LogicalDatastoreType.OPERATIONAL, managedNodePath, managedNodeBuilder.build());
75
76                     //Update node with managed node reference
77                     NodeBuilder nodeBuilder = new NodeBuilder();
78                     nodeBuilder.setNodeId(SouthboundMapper.createNodeId(key.getIp(),key.getPort()));
79
80                     OvsdbNodeAugmentationBuilder ovsdbNodeBuilder = new OvsdbNodeAugmentationBuilder();
81                     List<ManagedNodeEntry> managedNodes = new ArrayList<ManagedNodeEntry>();
82                     ManagedNodeEntry entry = new ManagedNodeEntryBuilder().setBridgeRef(new OvsdbBridgeRef(managedNodePath)).build();
83                     managedNodes.add(entry);
84                     ovsdbNodeBuilder.setManagedNodeEntry(managedNodes);
85
86                     nodeBuilder.addAugmentation(OvsdbNodeAugmentation.class, ovsdbNodeBuilder.build());
87
88                     LOG.debug("Update node with managed node ref {}",ovsdbNodeBuilder.toString());
89                     transaction.merge(LogicalDatastoreType.OPERATIONAL, nodePath, nodeBuilder.build());
90
91                 }
92             }
93         }
94     }
95 }