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 java.util.Collection;
  12 import java.util.HashMap;
  13 import java.util.List;
  14 import java.util.Map;
  15 import java.util.Map.Entry;
  16
  17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  18 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
  19 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
  20 import org.opendaylight.ovsdb.hwvtepsouthbound.HwvtepSouthboundUtil;
  21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.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.LogicalSwitches;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesKey;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.Switches;
  31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
  33 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  34 import org.slf4j.Logger;
  35 import org.slf4j.LoggerFactory;
  36 import com.google.common.base.Optional;
  37 import com.google.common.base.Preconditions;
  38
  39 public class HwvtepOperationalState {
  40     private static final Logger LOG = LoggerFactory.getLogger(HwvtepOperationalState.class);
  41     private Map<InstanceIdentifier<Node>, Node> operationalNodes = new HashMap<>();
  42
  43     public HwvtepOperationalState(DataBroker db, Collection<DataTreeModification<Node>> changes) {
  44         Map<InstanceIdentifier<Node>, Node> nodeCreateOrUpdate =
  45             TransactUtils.extractCreatedOrUpdatedOrRemoved(changes, Node.class);
  46         if (nodeCreateOrUpdate != null) {
  47             final ReadWriteTransaction transaction = db.newReadWriteTransaction();
  48             for (Entry<InstanceIdentifier<Node>, Node> entry: nodeCreateOrUpdate.entrySet()) {
  49                 Optional<Node> readNode = HwvtepSouthboundUtil.readNode(transaction, entry.getKey());
  50                 //add related globalNode or physicalSwitchNode to operationalNodes map
  51                 //for example, when creating physical port, logical switch is needed
  52                 //but logical switch is in HwvtepGlobalAugmentation rather than PhysicalSwitchAugmentation
  53                 if (readNode.isPresent()) {
  54                     operationalNodes.put(entry.getKey(), readNode.get());
  55                     HwvtepGlobalAugmentation hgAugmentation = readNode.get().getAugmentation(HwvtepGlobalAugmentation.class);
  56                     PhysicalSwitchAugmentation psAugmentation = readNode.get().getAugmentation(PhysicalSwitchAugmentation.class);
  57                     if (hgAugmentation != null) {
  58                         for (Switches pswitch : hgAugmentation.getSwitches()) {
  59                             @SuppressWarnings("unchecked")
  60                             InstanceIdentifier<Node> psNodeIid = (InstanceIdentifier<Node>) pswitch.getSwitchRef().getValue();
  61                             Optional<Node> psNode = HwvtepSouthboundUtil.readNode(transaction, psNodeIid);
  62                             if (psNode.isPresent()) {
  63                                 operationalNodes.put(psNodeIid, psNode.get());
  64                             }
  65                         }
  66                     }
  67                     if (psAugmentation != null) {
  68                         @SuppressWarnings("unchecked")
  69                         InstanceIdentifier<Node> hgNodeIid = (InstanceIdentifier<Node>) psAugmentation.getManagedBy().getValue();
  70                         Optional<Node> hgNode = HwvtepSouthboundUtil.readNode(transaction, hgNodeIid);
  71                         if (hgNode.isPresent()) {
  72                             operationalNodes.put(hgNodeIid, hgNode.get());
  73                         }
  74                     }
  75                 }
  76             }
  77         }
  78     }
  79
  80     private Optional<Node> getGlobalNode(InstanceIdentifier<?> iid) {
  81         InstanceIdentifier<Node> nodeIid = iid.firstIdentifierOf(Node.class);
  82         return Optional.fromNullable(operationalNodes.get(nodeIid));
  83     }
  84
  85     public Optional<HwvtepGlobalAugmentation> getHwvtepGlobalAugmentation(InstanceIdentifier<?> iid) {
  86         Preconditions.checkNotNull(iid);
  87         Optional<Node> nodeOptional = getGlobalNode(iid);
  88         if (nodeOptional.isPresent()) {
  89             return Optional.fromNullable(nodeOptional.get().getAugmentation(HwvtepGlobalAugmentation.class));
  90         }
  91         return Optional.absent();
  92     }
  93
  94     public Optional<PhysicalSwitchAugmentation> getPhysicalSwitchAugmentation(InstanceIdentifier<?> iid) {
  95         Preconditions.checkNotNull(iid);
  96         Optional<Node> nodeOptional = getGlobalNode(iid);
  97         if (nodeOptional.isPresent()) {
  98             return Optional.fromNullable(nodeOptional.get().getAugmentation(PhysicalSwitchAugmentation.class));
  99         }
 100         return Optional.absent();
 101     }
 102
 103     public Optional<List<TerminationPoint>> getTerminationPointList(InstanceIdentifier<?> iid) {
 104         Preconditions.checkNotNull(iid);
 105         Optional<Node> nodeOptional = getGlobalNode(iid);
 106         if (nodeOptional.isPresent() && nodeOptional.get().getTerminationPoint() != null) {
 107             return Optional.fromNullable(nodeOptional.get().getTerminationPoint());
 108         }
 109         return Optional.absent();
 110     }
 111
 112     public Optional<LogicalSwitches> getLogicalSwitches(InstanceIdentifier<?> iid, LogicalSwitchesKey logicalSwitchesKey) {
 113         Preconditions.checkNotNull(iid);
 114         Optional<HwvtepGlobalAugmentation> nodeOptional = getHwvtepGlobalAugmentation(iid);
 115         if (nodeOptional.isPresent()) {
 116             HwvtepGlobalAugmentation hgAugmentation = nodeOptional.get();
 117             List<LogicalSwitches> lswitchList = null;
 118             if (hgAugmentation != null) {
 119                 lswitchList = hgAugmentation.getLogicalSwitches();
 120             }
 121             if (lswitchList != null) {
 122                 for (LogicalSwitches lswitch: lswitchList) {
 123                     if (lswitch.getKey().equals(logicalSwitchesKey)) {
 124                         return Optional.fromNullable(lswitch);
 125                     }
 126                 }
 127             }
 128         }
 129         return Optional.absent();
 130     }
 131
 132     public Optional<HwvtepPhysicalPortAugmentation> getPhysicalPortAugmentation(InstanceIdentifier<?> iid,
 133             HwvtepNodeName hwvtepNodeName) {
 134         Preconditions.checkNotNull(iid);
 135         Optional<List<TerminationPoint>> nodeOptional = getTerminationPointList(iid);
 136         if (nodeOptional.isPresent()) {
 137             List<TerminationPoint> tpList = nodeOptional.get();
 138             for (TerminationPoint tp : tpList) {
 139                 HwvtepPhysicalPortAugmentation hppAugmentation = tp.getAugmentation(HwvtepPhysicalPortAugmentation.class);
 140                 if (hppAugmentation.getHwvtepNodeName().equals(hwvtepNodeName)) {
 141                     return Optional.fromNullable(hppAugmentation);
 142                 }
 143             }
 144         }
 145         return Optional.absent();
 146     }
 147
 148     public Optional<HwvtepPhysicalLocatorAugmentation> getPhysicalLocatorAugmentation(InstanceIdentifier<?> iid,
 149             IpAddress dstIp, Class<? extends EncapsulationTypeBase> encapType) {
 150         Preconditions.checkNotNull(iid);
 151         Optional<List<TerminationPoint>> nodeOptional = getTerminationPointList(iid);
 152         if (nodeOptional.isPresent()) {
 153             List<TerminationPoint> tpList = nodeOptional.get();
 154             for (TerminationPoint tp : tpList) {
 155                 HwvtepPhysicalLocatorAugmentation hppAugmentation = tp.getAugmentation(HwvtepPhysicalLocatorAugmentation.class);
 156                 if (hppAugmentation.getDstIp().equals(dstIp)
 157                         && hppAugmentation.getEncapsulationType().equals(encapType)) {
 158                     return Optional.fromNullable(hppAugmentation);
 159                 }
 160             }
 161         }
 162         return Optional.absent();
 163     }
 164 }