hwvtepsouthbound/hwvtepsouthbound-impl/src/main/java/org/opendaylight/ovsdb/hwvtepsouthbound/transact/HwvtepOperationalState.java

   1 /*
   2  * Copyright (c) 2015 China Telecom Beijing Research Institute 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.transact;
  10
  11 import com.google.common.base.Optional;
  12 import com.google.common.base.Preconditions;
  13 import com.google.common.collect.Maps;
  14 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  15 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
  16 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  17 import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepConnectionInstance;
  18 import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepDeviceInfo;
  19 import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepSouthboundUtil;
  20 import org.opendaylight.ovsdb.lib.notation.UUID;
  21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.EncapsulationTypeBase;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepNodeName;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalLocatorAugmentation;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalPortAugmentation;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.PhysicalSwitchAugmentation;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalMcastMacs;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalMcastMacsKey;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalUcastMacs;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LocalUcastMacsKey;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesKey;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacs;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacsKey;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteUcastMacs;
  37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteUcastMacsKey;
  38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.Switches;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical._switch.attributes.Tunnels;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical._switch.attributes.TunnelsKey;
  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.node.TerminationPoint;
  43 import org.opendaylight.yangtools.yang.binding.Identifiable;
  44 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  45 import org.slf4j.Logger;
  46 import org.slf4j.LoggerFactory;
  47
  48 import java.util.Collection;
  49 import java.util.Collections;
  50 import java.util.HashMap;
  51 import java.util.List;
  52 import java.util.Map;
  53 import java.util.Map.Entry;
  54 import java.util.Set;
  55 import java.util.concurrent.ConcurrentHashMap;
  56
  57 //TODO: need to be optimized, get entry by iid not name
  58 public class HwvtepOperationalState {
  59
  60     private static final Logger LOG = LoggerFactory.getLogger(HwvtepOperationalState.class);
  61
  62     private Map<InstanceIdentifier<Node>, Node> operationalNodes = new HashMap<>();
  63     private ReadWriteTransaction transaction;
  64     HashMap<InstanceIdentifier<TerminationPoint>, UUID> inflightLocators = Maps.newHashMap();
  65     private HwvtepDeviceInfo deviceInfo;
  66     private HwvtepConnectionInstance connectionInstance;
  67     private Map<Class<? extends Identifiable>, Map<InstanceIdentifier, UUID>> currentTxUUIDs = new ConcurrentHashMap<>();
  68     private Map<Class<? extends Identifiable>, Map<InstanceIdentifier, Boolean>> currentTxDeletedKeys = new ConcurrentHashMap<>();
  69
  70     public HwvtepOperationalState(DataBroker db, HwvtepConnectionInstance connectionInstance,
  71                                   Collection<DataTreeModification<Node>> changes) {
  72         this.connectionInstance = connectionInstance;
  73         this.deviceInfo = connectionInstance.getDeviceInfo();
  74         Map<InstanceIdentifier<Node>, Node> nodeCreateOrUpdate =
  75             TransactUtils.extractCreatedOrUpdatedOrRemoved(changes, Node.class);
  76         if (nodeCreateOrUpdate != null) {
  77             transaction = db.newReadWriteTransaction();
  78             for (Entry<InstanceIdentifier<Node>, Node> entry: nodeCreateOrUpdate.entrySet()) {
  79                 Optional<Node> readNode = HwvtepSouthboundUtil.readNode(transaction, entry.getKey());
  80                 //add related globalNode or physicalSwitchNode to operationalNodes map
  81                 //for example, when creating physical port, logical switch is needed
  82                 //but logical switch is in HwvtepGlobalAugmentation rather than PhysicalSwitchAugmentation
  83                 if (readNode.isPresent()) {
  84                     operationalNodes.put(entry.getKey(), readNode.get());
  85                     HwvtepGlobalAugmentation hgAugmentation = readNode.get().getAugmentation(HwvtepGlobalAugmentation.class);
  86                     PhysicalSwitchAugmentation psAugmentation = readNode.get().getAugmentation(PhysicalSwitchAugmentation.class);
  87                     if (hgAugmentation != null && hgAugmentation.getSwitches() != null) {
  88                         for (Switches pswitch : hgAugmentation.getSwitches()) {
  89                             @SuppressWarnings("unchecked")
  90                             InstanceIdentifier<Node> psNodeIid = (InstanceIdentifier<Node>) pswitch.getSwitchRef().getValue();
  91                             Optional<Node> psNode = HwvtepSouthboundUtil.readNode(transaction, psNodeIid);
  92                             if (psNode.isPresent()) {
  93                                 operationalNodes.put(psNodeIid, psNode.get());
  94                             }
  95                         }
  96                     }
  97                     if (psAugmentation != null) {
  98                         @SuppressWarnings("unchecked")
  99                         InstanceIdentifier<Node> hgNodeIid = (InstanceIdentifier<Node>) psAugmentation.getManagedBy().getValue();
 100                         Optional<Node> hgNode = HwvtepSouthboundUtil.readNode(transaction, hgNodeIid);
 101                         if (hgNode.isPresent()) {
 102                             operationalNodes.put(hgNodeIid, hgNode.get());
 103                         }
 104                     }
 105                 }
 106             }
 107         }
 108     }
 109
 110     public HwvtepOperationalState(HwvtepConnectionInstance connectionInstance) {
 111         this.connectionInstance = connectionInstance;
 112         this.deviceInfo = connectionInstance.getDeviceInfo();
 113         transaction = connectionInstance.getDataBroker().newReadWriteTransaction();
 114         Optional<Node> readNode = HwvtepSouthboundUtil.readNode(transaction, connectionInstance.getInstanceIdentifier());
 115         if (readNode.isPresent()) {
 116             operationalNodes.put(connectionInstance.getInstanceIdentifier(), readNode.get());
 117         }
 118     }
 119
 120     public Optional<Node> getGlobalNode(InstanceIdentifier<?> iid) {
 121         InstanceIdentifier<Node> nodeIid = iid.firstIdentifierOf(Node.class);
 122         return Optional.fromNullable(operationalNodes.get(nodeIid));
 123     }
 124
 125     public Optional<HwvtepGlobalAugmentation> getHwvtepGlobalAugmentation(InstanceIdentifier<?> iid) {
 126         Preconditions.checkNotNull(iid);
 127         Optional<Node> nodeOptional = getGlobalNode(iid);
 128         if (nodeOptional.isPresent()) {
 129             return Optional.fromNullable(nodeOptional.get().getAugmentation(HwvtepGlobalAugmentation.class));
 130         }
 131         return Optional.absent();
 132     }
 133
 134     public Optional<PhysicalSwitchAugmentation> getPhysicalSwitchAugmentation(InstanceIdentifier<?> iid) {
 135         Preconditions.checkNotNull(iid);
 136         Optional<Node> nodeOptional = getGlobalNode(iid);
 137         if (nodeOptional.isPresent()) {
 138             return Optional.fromNullable(nodeOptional.get().getAugmentation(PhysicalSwitchAugmentation.class));
 139         }
 140         return Optional.absent();
 141     }
 142
 143     public Optional<List<TerminationPoint>> getTerminationPointList(InstanceIdentifier<?> iid) {
 144         Preconditions.checkNotNull(iid);
 145         Optional<Node> nodeOptional = getGlobalNode(iid);
 146         if (nodeOptional.isPresent() && nodeOptional.get().getTerminationPoint() != null) {
 147             return Optional.fromNullable(nodeOptional.get().getTerminationPoint());
 148         }
 149         return Optional.absent();
 150     }
 151
 152     public Optional<LogicalSwitches> getLogicalSwitches(InstanceIdentifier<?> iid, LogicalSwitchesKey logicalSwitchesKey) {
 153         Preconditions.checkNotNull(iid);
 154         Optional<HwvtepGlobalAugmentation> nodeOptional = getHwvtepGlobalAugmentation(iid);
 155         if (nodeOptional.isPresent()) {
 156             HwvtepGlobalAugmentation hgAugmentation = nodeOptional.get();
 157             List<LogicalSwitches> lswitchList = null;
 158             if (hgAugmentation != null) {
 159                 lswitchList = hgAugmentation.getLogicalSwitches();
 160             }
 161             if (lswitchList != null) {
 162                 for (LogicalSwitches lswitch: lswitchList) {
 163                     if (lswitch.getKey().equals(logicalSwitchesKey)) {
 164                         return Optional.fromNullable(lswitch);
 165                     }
 166                 }
 167             }
 168         }
 169         return Optional.absent();
 170     }
 171
 172     public Optional<Tunnels> getTunnels(InstanceIdentifier<?> iid, TunnelsKey tunnelsKey) {
 173         Preconditions.checkNotNull(iid);
 174         Optional<PhysicalSwitchAugmentation> psOptional = getPhysicalSwitchAugmentation(iid);
 175         if (psOptional.isPresent()) {
 176             PhysicalSwitchAugmentation psAugmentation = psOptional.get();
 177             List<Tunnels> tunnelList = null;
 178             if (psAugmentation != null) {
 179                 tunnelList = psAugmentation.getTunnels();
 180             }
 181             if (tunnelList != null) {
 182                 for (Tunnels tunnel: tunnelList) {
 183                     if (tunnel.getKey().equals(tunnelsKey)) {
 184                         return Optional.fromNullable(tunnel);
 185                     }
 186                 }
 187             }
 188         }
 189         return Optional.absent();
 190     }
 191
 192     public Optional<HwvtepPhysicalPortAugmentation> getPhysicalPortAugmentation(InstanceIdentifier<?> iid,
 193             HwvtepNodeName hwvtepNodeName) {
 194         Preconditions.checkNotNull(iid);
 195         Optional<List<TerminationPoint>> nodeOptional = getTerminationPointList(iid);
 196         if (nodeOptional.isPresent()) {
 197             List<TerminationPoint> tpList = nodeOptional.get();
 198             for (TerminationPoint tp : tpList) {
 199                 HwvtepPhysicalPortAugmentation hppAugmentation = tp.getAugmentation(HwvtepPhysicalPortAugmentation.class);
 200                 if (hppAugmentation != null && hppAugmentation.getHwvtepNodeName().equals(hwvtepNodeName)) {
 201                     return Optional.fromNullable(hppAugmentation);
 202                 }
 203             }
 204         }
 205         return Optional.absent();
 206     }
 207
 208     public Optional<HwvtepPhysicalLocatorAugmentation> getPhysicalLocatorAugmentation(InstanceIdentifier<?> iid,
 209             IpAddress dstIp, Class<? extends EncapsulationTypeBase> encapType) {
 210         Preconditions.checkNotNull(iid);
 211         Optional<List<TerminationPoint>> nodeOptional = getTerminationPointList(iid);
 212         if (nodeOptional.isPresent()) {
 213             List<TerminationPoint> tpList = nodeOptional.get();
 214             for (TerminationPoint tp : tpList) {
 215                 HwvtepPhysicalLocatorAugmentation hppAugmentation = tp.getAugmentation(HwvtepPhysicalLocatorAugmentation.class);
 216                 if (hppAugmentation != null && hppAugmentation.getDstIp().equals(dstIp)
 217                         && hppAugmentation.getEncapsulationType().equals(encapType)) {
 218                     return Optional.fromNullable(hppAugmentation);
 219                 }
 220             }
 221         }
 222         return Optional.absent();
 223     }
 224
 225     public Optional<LocalMcastMacs> getLocalMcastMacs(InstanceIdentifier<?> iid, LocalMcastMacsKey key) {
 226         Preconditions.checkNotNull(iid);
 227         Optional<HwvtepGlobalAugmentation> nodeOptional = getHwvtepGlobalAugmentation(iid);
 228         if (nodeOptional.isPresent()) {
 229             HwvtepGlobalAugmentation hgAugmentation = nodeOptional.get();
 230             List<LocalMcastMacs> macList = null;
 231             if (hgAugmentation != null) {
 232                 macList = hgAugmentation.getLocalMcastMacs();
 233             }
 234             if (macList != null) {
 235                 for (LocalMcastMacs mac: macList) {
 236                     if (mac.getKey().equals(key)) {
 237                         return Optional.fromNullable(mac);
 238                     }
 239                 }
 240             }
 241         }
 242         return Optional.absent();
 243     }
 244
 245     public Optional<RemoteMcastMacs> getRemoteMcastMacs(InstanceIdentifier<?> iid, RemoteMcastMacsKey key) {
 246         Preconditions.checkNotNull(iid);
 247         Optional<HwvtepGlobalAugmentation> nodeOptional = getHwvtepGlobalAugmentation(iid);
 248         if (nodeOptional.isPresent()) {
 249             HwvtepGlobalAugmentation hgAugmentation = nodeOptional.get();
 250             List<RemoteMcastMacs> macList = null;
 251             if (hgAugmentation != null) {
 252                 macList = hgAugmentation.getRemoteMcastMacs();
 253             }
 254             if (macList != null) {
 255                 for (RemoteMcastMacs mac: macList) {
 256                     if (mac.getKey().equals(key)) {
 257                         return Optional.fromNullable(mac);
 258                     }
 259                 }
 260             }
 261         }
 262         return Optional.absent();
 263     }
 264
 265     public Optional<LocalUcastMacs> getLocalUcastMacs(InstanceIdentifier<?> iid, LocalUcastMacsKey key) {
 266         Preconditions.checkNotNull(iid);
 267         Optional<HwvtepGlobalAugmentation> nodeOptional = getHwvtepGlobalAugmentation(iid);
 268         if (nodeOptional.isPresent()) {
 269             HwvtepGlobalAugmentation hgAugmentation = nodeOptional.get();
 270             List<LocalUcastMacs> macList = null;
 271             if (hgAugmentation != null) {
 272                 macList = hgAugmentation.getLocalUcastMacs();
 273             }
 274             if (macList != null) {
 275                 for (LocalUcastMacs mac: macList) {
 276                     if (mac.getKey().equals(key)) {
 277                         return Optional.fromNullable(mac);
 278                     }
 279                 }
 280             }
 281         }
 282         return Optional.absent();
 283     }
 284
 285     public Optional<RemoteUcastMacs> getRemoteUcastMacs(InstanceIdentifier<?> iid, RemoteUcastMacsKey key) {
 286         Preconditions.checkNotNull(iid);
 287         Optional<HwvtepGlobalAugmentation> nodeOptional = getHwvtepGlobalAugmentation(iid);
 288         if (nodeOptional.isPresent()) {
 289             HwvtepGlobalAugmentation hgAugmentation = nodeOptional.get();
 290             List<RemoteUcastMacs> macList = null;
 291             if (hgAugmentation != null) {
 292                 macList = hgAugmentation.getRemoteUcastMacs();
 293             }
 294             if (macList != null) {
 295                 for (RemoteUcastMacs mac: macList) {
 296                     if (mac.getKey().equals(key)) {
 297                         return Optional.fromNullable(mac);
 298                     }
 299                 }
 300             }
 301         }
 302         return Optional.absent();
 303     }
 304
 305     public Optional<HwvtepPhysicalLocatorAugmentation> getPhysicalLocatorAugmentation(InstanceIdentifier<TerminationPoint> iid) {
 306         Optional<TerminationPoint> tp = HwvtepSouthboundUtil.readNode(transaction, iid);
 307         if (tp.isPresent()) {
 308             return Optional.fromNullable(tp.get().getAugmentation(HwvtepPhysicalLocatorAugmentation.class));
 309         }
 310         return Optional.absent();
 311     }
 312
 313     public Optional<LogicalSwitches> getLogicalSwitches(InstanceIdentifier<LogicalSwitches> iid) {
 314         Optional<LogicalSwitches> lswitch = HwvtepSouthboundUtil.readNode(transaction, iid);
 315         return lswitch;
 316     }
 317
 318     public Optional<Tunnels> getTunnels(InstanceIdentifier<Tunnels> iid) {
 319         Optional<Tunnels> tunnels = HwvtepSouthboundUtil.readNode(transaction, iid);
 320         return tunnels;
 321     }
 322
 323     public ReadWriteTransaction getReadWriteTransaction() {
 324         return transaction;
 325     }
 326
 327     public void setPhysicalLocatorInFlight(InstanceIdentifier<TerminationPoint> iid,
 328                                            UUID uuid) {
 329         inflightLocators.put(iid, uuid);
 330     }
 331
 332     public UUID getPhysicalLocatorInFlight(InstanceIdentifier<TerminationPoint> iid) {
 333         return inflightLocators.get(iid);
 334     }
 335
 336     public HwvtepConnectionInstance getConnectionInstance() {
 337         return connectionInstance;
 338     }
 339
 340     public HwvtepDeviceInfo getDeviceInfo() {
 341         return deviceInfo;
 342     }
 343
 344     public void updateCurrentTxData(Class<? extends Identifiable> cls, InstanceIdentifier key, UUID uuid) {
 345         HwvtepSouthboundUtil.updateData(currentTxUUIDs, cls, key, uuid);
 346     }
 347
 348     public void updateCurrentTxDeleteData(Class<? extends Identifiable> cls, InstanceIdentifier key) {
 349         HwvtepSouthboundUtil.updateData(currentTxDeletedKeys, cls, key, Boolean.TRUE);
 350     }
 351
 352     public UUID getUUIDFromCurrentTx(Class<? extends Identifiable> cls, InstanceIdentifier key) {
 353         return HwvtepSouthboundUtil.getData(currentTxUUIDs, cls, key);
 354     }
 355
 356     public boolean isKeyPartOfCurrentTx(Class<? extends Identifiable> cls, InstanceIdentifier key) {
 357         return HwvtepSouthboundUtil.containsKey(currentTxUUIDs, cls, key);
 358     }
 359
 360     public Set<InstanceIdentifier> getDeletedKeysInCurrentTx(Class<? extends Identifiable> cls) {
 361         if (currentTxDeletedKeys.containsKey(cls)) {
 362             return currentTxDeletedKeys.get(cls).keySet();
 363         }
 364         return Collections.EMPTY_SET;
 365     }
 366
 367 }