TSC-101: Fixup Augmentable and Identifiable methods change
[ovsdb.git] / hwvtepsouthbound / hwvtepsouthbound-impl / src / main / java / org / opendaylight / ovsdb / hwvtepsouthbound / transactions / md / HwvtepMcastMacsRemoteUpdateCommand.java
1 /*
2  * Copyright (c) 2015, 2017 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 java.util.ArrayList;
12 import java.util.List;
13 import java.util.Map;
14 import java.util.Map.Entry;
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.HwvtepSouthboundConstants;
19 import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepSouthboundMapper;
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.McastMacsRemote;
26 import org.opendaylight.ovsdb.schema.hardwarevtep.PhysicalLocator;
27 import org.opendaylight.ovsdb.schema.hardwarevtep.PhysicalLocatorSet;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.MacAddress;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Uuid;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentationBuilder;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepLogicalSwitchRef;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalLocatorRef;
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.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacs;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacsBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacsKey;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.locator.set.attributes.LocatorSet;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.locator.set.attributes.LocatorSetBuilder;
41 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
42 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
43 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
44 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
45
46 public class HwvtepMcastMacsRemoteUpdateCommand extends AbstractTransactionCommand {
47
48     private final Map<UUID, McastMacsRemote> updatedMMacsRemoteRows;
49     private final Map<UUID, PhysicalLocatorSet> updatedPLocSetRows;
50     private final Map<UUID, PhysicalLocator> updatedPLocRows;
51
52     public HwvtepMcastMacsRemoteUpdateCommand(HwvtepConnectionInstance key, TableUpdates updates,
53             DatabaseSchema dbSchema) {
54         super(key, updates, dbSchema);
55         updatedMMacsRemoteRows = TyperUtils.extractRowsUpdated(McastMacsRemote.class, getUpdates(), getDbSchema());
56         updatedPLocSetRows = TyperUtils.extractRowsUpdated(PhysicalLocatorSet.class, getUpdates(), getDbSchema());
57         updatedPLocRows = TyperUtils.extractRowsUpdated(PhysicalLocator.class, getUpdates(), getDbSchema());
58     }
59
60     @Override
61     public void execute(ReadWriteTransaction transaction) {
62         for (Entry<UUID, McastMacsRemote> entry : updatedMMacsRemoteRows.entrySet()) {
63             updateData(transaction, entry.getValue());
64         }
65     }
66
67     private void updateData(ReadWriteTransaction transaction, McastMacsRemote macRemote) {
68         final InstanceIdentifier<Node> connectionIId = getOvsdbConnectionInstance().getInstanceIdentifier();
69         Node connectionNode = buildConnectionNode(macRemote);
70         transaction.merge(LogicalDatastoreType.OPERATIONAL, connectionIId, connectionNode);
71         InstanceIdentifier<RemoteMcastMacs> macIid = getMacIid(connectionIId, connectionNode);
72         getOvsdbConnectionInstance().getDeviceInfo().updateDeviceOperData(RemoteMcastMacs.class,
73                 macIid, macRemote.getUuid(), macRemote);
74     }
75
76     InstanceIdentifier<RemoteMcastMacs> getMacIid(InstanceIdentifier<Node> connectionIId, Node connectionNode) {
77         RemoteMcastMacsKey macsKey =
78                 connectionNode.augmentation(HwvtepGlobalAugmentation.class).getRemoteMcastMacs().get(0).key();
79         InstanceIdentifier<RemoteMcastMacs> key = connectionIId.augmentation(HwvtepGlobalAugmentation.class)
80                 .child(RemoteMcastMacs.class, macsKey);
81         return key;
82     }
83
84     private Node buildConnectionNode(McastMacsRemote macRemote) {
85         NodeBuilder connectionNode = new NodeBuilder();
86         connectionNode.setNodeId(getOvsdbConnectionInstance().getNodeId());
87         RemoteMcastMacsBuilder macRemoteBuilder = new RemoteMcastMacsBuilder();
88         if (macRemote.getMac().equals(HwvtepSouthboundConstants.UNKNOWN_DST_STRING)) {
89             macRemoteBuilder.setMacEntryKey(HwvtepSouthboundConstants.UNKNOWN_DST_MAC);
90         } else {
91             macRemoteBuilder.setMacEntryKey(new MacAddress(macRemote.getMac()));
92         }
93         macRemoteBuilder.setMacEntryUuid(new Uuid(macRemote.getUuid().toString()));
94         setIpAddress(macRemoteBuilder, macRemote);
95         setLocatorSet(macRemoteBuilder, macRemote);
96         setLogicalSwitch(macRemoteBuilder, macRemote);
97
98         List<RemoteMcastMacs> macRemoteList = new ArrayList<>();
99         macRemoteList.add(macRemoteBuilder.build());
100
101         HwvtepGlobalAugmentationBuilder hgAugmentationBuilder = new HwvtepGlobalAugmentationBuilder();
102         hgAugmentationBuilder.setRemoteMcastMacs(macRemoteList);
103         connectionNode.addAugmentation(HwvtepGlobalAugmentation.class, hgAugmentationBuilder.build());
104         return connectionNode.build();
105     }
106
107     private void setLogicalSwitch(RemoteMcastMacsBuilder macRemoteBuilder, McastMacsRemote macRemote) {
108         if (macRemote.getLogicalSwitchColumn() != null && macRemote.getLogicalSwitchColumn().getData() != null) {
109             UUID lsUUID = macRemote.getLogicalSwitchColumn().getData();
110             LogicalSwitch logicalSwitch = getOvsdbConnectionInstance().getDeviceInfo().getLogicalSwitch(lsUUID);
111             if (logicalSwitch != null) {
112                 InstanceIdentifier<LogicalSwitches> switchIid =
113                         HwvtepSouthboundMapper.createInstanceIdentifier(getOvsdbConnectionInstance(), logicalSwitch);
114                 macRemoteBuilder.setLogicalSwitchRef(new HwvtepLogicalSwitchRef(switchIid));
115             }
116         }
117     }
118
119     private void setIpAddress(RemoteMcastMacsBuilder macRemoteBuilder, McastMacsRemote macRemote) {
120         if (macRemote.getIpAddr() != null && !macRemote.getIpAddr().isEmpty()) {
121             macRemoteBuilder.setIpaddr(new IpAddress(macRemote.getIpAddr().toCharArray()));
122         }
123     }
124
125     private void setLocatorSet(RemoteMcastMacsBuilder macRemoteBuilder, McastMacsRemote macRemote) {
126         if (macRemote.getLocatorSetColumn() != null && macRemote.getLocatorSetColumn().getData() != null) {
127             UUID locSetUUID = macRemote.getLocatorSetColumn().getData();
128             PhysicalLocatorSet plSet = updatedPLocSetRows.get(locSetUUID);
129             if (plSet != null) {
130                 if (plSet.getLocatorsColumn() != null && plSet.getLocatorsColumn().getData() != null
131                         && !plSet.getLocatorsColumn().getData().isEmpty()) {
132                     List<LocatorSet> plsList = new ArrayList<>();
133                     for (UUID locUUID : plSet.getLocatorsColumn().getData()) {
134                         PhysicalLocator locator = updatedPLocRows.get(locUUID);
135                         if (locator == null) {
136                             locator = (PhysicalLocator) getOvsdbConnectionInstance()
137                                     .getDeviceInfo().getDeviceOperData(TerminationPoint.class, locUUID);
138                         }
139                         InstanceIdentifier<TerminationPoint> tpIid = HwvtepSouthboundMapper.createInstanceIdentifier(
140                                 getOvsdbConnectionInstance().getInstanceIdentifier(), locator);
141                         plsList.add(new LocatorSetBuilder()
142                                 .setLocatorRef(new HwvtepPhysicalLocatorRef(tpIid)).build());
143                     }
144                     macRemoteBuilder.setLocatorSet(plsList);
145                 }
146             }
147         }
148     }
149
150 }