Merge "Return only the Identifier and not the whole Node object"
[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 import java.util.Map;
6 import java.util.Map.Entry;
7
8 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
9 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
10 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
11 import org.opendaylight.ovsdb.lib.message.TableUpdates;
12 import org.opendaylight.ovsdb.lib.notation.UUID;
13 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
14 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
15 import org.opendaylight.ovsdb.schema.openvswitch.Bridge;
16 import org.opendaylight.ovsdb.schema.openvswitch.Controller;
17 import org.opendaylight.ovsdb.southbound.OvsdbClientKey;
18 import org.opendaylight.ovsdb.southbound.SouthboundConstants;
19 import org.opendaylight.ovsdb.southbound.SouthboundMapper;
20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.DatapathId;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentationBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeName;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeRef;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentation;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeAugmentationBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbNodeRef;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntry;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.node.attributes.ManagedNodeEntryBuilder;
31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
34 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
35 import org.slf4j.Logger;
36 import org.slf4j.LoggerFactory;
37
38 import com.google.common.base.Optional;
39
40 public class OvsdbBridgeUpdateCommand extends AbstractTransactionCommand {
41     private static final Logger LOG = LoggerFactory.getLogger(OvsdbBridgeUpdateCommand.class);
42
43     public OvsdbBridgeUpdateCommand(OvsdbClientKey key, TableUpdates updates,
44             DatabaseSchema dbSchema) {
45         super(key,updates,dbSchema);
46     }
47
48     @Override
49     public void execute(ReadWriteTransaction transaction) {
50         Map<UUID,Bridge> updatedBridgeRows = TyperUtils.extractRowsUpdated(Bridge.class, getUpdates(), getDbSchema());
51         Map<UUID,Controller> updatedControllerRows = TyperUtils.extractRowsUpdated(Controller.class, getUpdates(), getDbSchema());
52         for(Entry<UUID, Bridge> entry : updatedBridgeRows.entrySet()) {
53             Bridge bridge = entry.getValue();
54             final InstanceIdentifier<Node> nodePath = getKey().toInstanceIndentifier();
55             Optional<Node> node = Optional.absent();
56             try{
57                 node = transaction.read(LogicalDatastoreType.OPERATIONAL, nodePath).checkedGet();
58             }catch (final ReadFailedException e) {
59                 LOG.debug("Read Operational/DS for Node fail! {}", nodePath, e);
60             }
61             if(node.isPresent()){
62                 LOG.info("Node {} is present",node);
63                 NodeBuilder managedNodeBuilder = new NodeBuilder();
64                 NodeId manageNodeId = SouthboundMapper.createManagedNodeId(getKey(), new OvsdbBridgeName(bridge.getName()));
65                 managedNodeBuilder.setNodeId(manageNodeId);
66                 OvsdbBridgeAugmentationBuilder ovsdbManagedNodeBuilder = new OvsdbBridgeAugmentationBuilder();
67                 ovsdbManagedNodeBuilder.setBridgeName(new OvsdbBridgeName(bridge.getName()));
68                 ovsdbManagedNodeBuilder.setBridgeUuid(new Uuid(bridge.getUuid().toString()));
69                 DatapathId dpid= SouthboundMapper.createDatapathId(bridge);
70                 if(dpid != null) {
71                     ovsdbManagedNodeBuilder.setDatapathId(dpid);
72                 }
73                 if(SouthboundMapper.createMdsalProtocols(bridge) != null
74                         && SouthboundMapper.createMdsalProtocols(bridge).size() > 0) {
75                     ovsdbManagedNodeBuilder.setProtocolEntry(SouthboundMapper.createMdsalProtocols(bridge));
76                 }
77
78                 if(!SouthboundMapper.createControllerEntries(bridge, updatedControllerRows).isEmpty()) {
79                     ovsdbManagedNodeBuilder.setControllerEntry(SouthboundMapper.createControllerEntries(bridge, updatedControllerRows));
80                 }
81
82                 if(bridge.getFailModeColumn() != null &&
83                         bridge.getFailModeColumn().getData() != null &&
84                         !bridge.getFailModeColumn().getData().isEmpty()) {
85                     String[] failmodeArray = new String[bridge.getFailModeColumn().getData().size()];
86                     bridge.getFailModeColumn().getData().toArray(failmodeArray);
87                     ovsdbManagedNodeBuilder.setFailMode(SouthboundConstants.OVSDB_FAIL_MODE_MAP.inverse().get(failmodeArray[0]));
88                 }
89                 ovsdbManagedNodeBuilder.setManagedBy(new OvsdbNodeRef(nodePath));
90                 managedNodeBuilder.addAugmentation(OvsdbBridgeAugmentation.class, ovsdbManagedNodeBuilder.build());
91
92                 InstanceIdentifier<Node> managedNodePath = SouthboundMapper.createInstanceIdentifier(manageNodeId);
93
94                 LOG.debug("Store managed node augmentation data {}",ovsdbManagedNodeBuilder.toString());
95                 transaction.merge(LogicalDatastoreType.OPERATIONAL, managedNodePath, managedNodeBuilder.build());
96
97                 //Update node with managed node reference
98                 NodeBuilder nodeBuilder = new NodeBuilder();
99                 nodeBuilder.setNodeId(SouthboundMapper.createNodeId(getKey().getIp(),getKey().getPort()));
100
101                 OvsdbNodeAugmentationBuilder ovsdbNodeBuilder = new OvsdbNodeAugmentationBuilder();
102                 List<ManagedNodeEntry> managedNodes = new ArrayList<ManagedNodeEntry>();
103                 ManagedNodeEntry managedNodeEntry = new ManagedNodeEntryBuilder().setBridgeRef(new OvsdbBridgeRef(managedNodePath)).build();
104                 managedNodes.add(managedNodeEntry);
105                 ovsdbNodeBuilder.setManagedNodeEntry(managedNodes);
106
107                 nodeBuilder.addAugmentation(OvsdbNodeAugmentation.class, ovsdbNodeBuilder.build());
108
109                 LOG.debug("Update node with managed node ref {}",ovsdbNodeBuilder.toString());
110                 transaction.merge(LogicalDatastoreType.OPERATIONAL, nodePath, nodeBuilder.build());
111
112             }
113         }
114     }
115 }