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