2 * Copyright (c) 2015 China Telecom Beijing Research Institute and others. All rights reserved.
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
9 package org.opendaylight.ovsdb.hwvtepsouthbound.transact;
11 import java.util.Collection;
12 import java.util.HashMap;
13 import java.util.List;
15 import java.util.Map.Entry;
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;
39 public class HwvtepOperationalState {
40 private static final Logger LOG = LoggerFactory.getLogger(HwvtepOperationalState.class);
41 private Map<InstanceIdentifier<Node>, Node> operationalNodes = new HashMap<>();
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());
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());
80 private Optional<Node> getGlobalNode(InstanceIdentifier<?> iid) {
81 InstanceIdentifier<Node> nodeIid = iid.firstIdentifierOf(Node.class);
82 return Optional.fromNullable(operationalNodes.get(nodeIid));
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));
91 return Optional.absent();
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));
100 return Optional.absent();
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());
109 return Optional.absent();
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();
121 if (lswitchList != null) {
122 for (LogicalSwitches lswitch: lswitchList) {
123 if (lswitch.getKey().equals(logicalSwitchesKey)) {
124 return Optional.fromNullable(lswitch);
129 return Optional.absent();
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);
145 return Optional.absent();
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);
162 return Optional.absent();