Change logical-switch-ref to iid in hwvtep.yang
[ovsdb.git] / hwvtepsouthbound / hwvtepsouthbound-impl / src / main / java / org / opendaylight / ovsdb / hwvtepsouthbound / transactions / md / UcastMacsLocalUpdateCommand.java
1 /*
2  * Copyright (c) 2015 Ericsson India Global Services Pvt Ltd. 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
9 package org.opendaylight.ovsdb.hwvtepsouthbound.transactions.md;
10
11 import com.google.common.base.Optional;
12 import java.util.ArrayList;
13 import java.util.List;
14 import java.util.Map;
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.ovsdb.hwvtepsouthbound.HwvtepConnectionInstance;
18 import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepSouthboundMapper;
19 import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepSouthboundUtil;
20 import org.opendaylight.ovsdb.lib.message.TableUpdates;
21 import org.opendaylight.ovsdb.lib.notation.UUID;
22 import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
23 import org.opendaylight.ovsdb.lib.schema.typed.TyperUtils;
24 import org.opendaylight.ovsdb.schema.hardwarevtep.LogicalSwitch;
25 import org.opendaylight.ovsdb.schema.hardwarevtep.PhysicalLocator;
26 import org.opendaylight.ovsdb.schema.hardwarevtep.UcastMacsLocal;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpAddress;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentationBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepLogicalSwitchRef;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalLocatorRef;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalUcastMacs;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalUcastMacsBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
38 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
40
41 public class UcastMacsLocalUpdateCommand extends AbstractTransactionCommand {
42
43     private Map<UUID, UcastMacsLocal> updatedUMacsLocalRows;
44     private Map<UUID, PhysicalLocator> updatedPLocRows;
45     private Map<UUID, LogicalSwitch> updatedLSRows;
46
47     public UcastMacsLocalUpdateCommand(HwvtepConnectionInstance key, TableUpdates updates, DatabaseSchema dbSchema) {
48         super(key, updates, dbSchema);
49         updatedUMacsLocalRows = TyperUtils.extractRowsUpdated(UcastMacsLocal.class, getUpdates(), getDbSchema());
50         updatedPLocRows = TyperUtils.extractRowsUpdated(PhysicalLocator.class, getUpdates(), getDbSchema());
51         updatedLSRows = TyperUtils.extractRowsUpdated(LogicalSwitch.class, getUpdates(), getDbSchema());
52     }
53
54     @Override
55     public void execute(ReadWriteTransaction transaction) {
56         for (UcastMacsLocal ucastMacsLocal : updatedUMacsLocalRows.values()) {
57             updateData(transaction, ucastMacsLocal);
58         }
59     }
60
61     private void updateData(ReadWriteTransaction transaction, UcastMacsLocal ucml) {
62         final InstanceIdentifier<Node> connectionIId = getOvsdbConnectionInstance().getInstanceIdentifier();
63         Optional<Node> connection = HwvtepSouthboundUtil.readNode(transaction, connectionIId);
64         if (connection.isPresent()) {
65             // Update the connection node to let it know it manages this UCastMacsLocal
66             Node connectionNode = buildConnectionNode(ucml);
67             transaction.merge(LogicalDatastoreType.OPERATIONAL, connectionIId, connectionNode);
68             // TODO: Delete entries that are no longer needed
69         }
70     }
71
72     private Node buildConnectionNode(UcastMacsLocal ucml) {
73         //Update node with UcastMacsLocal reference
74         NodeBuilder connectionNode = new NodeBuilder();
75         connectionNode.setNodeId(getOvsdbConnectionInstance().getNodeId());
76         InstanceIdentifier<Node> nodeIid = getOvsdbConnectionInstance().getInstanceIdentifier();
77         HwvtepGlobalAugmentationBuilder hgAugmentationBuilder = new HwvtepGlobalAugmentationBuilder();
78         LocalUcastMacsBuilder ucmlBuilder = new LocalUcastMacsBuilder();
79         if (ucml.getIpAddr() != null && !ucml.getIpAddr().isEmpty()) {
80             ucmlBuilder.setIpaddr(new IpAddress(ucml.getIpAddr().toCharArray()));
81         }
82         ucmlBuilder.setMacEntryKey(new MacAddress(ucml.getMac()));
83         if (ucml.getLocatorColumn() != null && ucml.getLocatorColumn().getData() != null) {
84             UUID plocUUID = ucml.getLocatorColumn().getData();
85             PhysicalLocator physicalLocator = updatedPLocRows.get(plocUUID);
86             if (physicalLocator != null) {
87                 InstanceIdentifier<TerminationPoint> plIid =
88                         HwvtepSouthboundMapper.createInstanceIdentifier(nodeIid, physicalLocator);
89                 ucmlBuilder.setLocatorRef(new HwvtepPhysicalLocatorRef(plIid));
90             }
91         }
92         if (ucml.getLogicalSwitchColumn() != null && ucml.getLogicalSwitchColumn().getData() != null) {
93             UUID lsUUID = ucml.getLogicalSwitchColumn().getData();
94             LogicalSwitch logicalSwitch = updatedLSRows.get(lsUUID);
95             if (logicalSwitch != null) {
96                 InstanceIdentifier<LogicalSwitches> lSwitchIid =
97                         HwvtepSouthboundMapper.createInstanceIdentifier(getOvsdbConnectionInstance(), logicalSwitch);
98                 ucmlBuilder.setLogicalSwitchRef(new HwvtepLogicalSwitchRef(lSwitchIid));
99             }
100         }
101         List<LocalUcastMacs> umclList = new ArrayList<>();
102         umclList.add(ucmlBuilder.build());
103         hgAugmentationBuilder.setLocalUcastMacs(umclList);
104         connectionNode.addAugmentation(HwvtepGlobalAugmentation.class, hgAugmentationBuilder.build());
105         return connectionNode.build();
106     }
107
108 }