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;
16 import java.util.concurrent.ExecutionException;
18 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
19 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
20 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
21 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
22 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
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.HwvtepPhysicalLocatorAugmentation;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepPhysicalPortAugmentation;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.PhysicalSwitchAugmentation;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitchesKey;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.physical.locator.set.attributes.LocatorSet;
30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointKey;
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.util.concurrent.CheckedFuture;
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 ReadOnlyTransaction transaction = db.newReadOnlyTransaction();
45 Map<InstanceIdentifier<Node>, Node> nodeCreateOrUpdate =
46 TransactUtils.extractCreatedOrUpdatedOrRemoved(changes, Node.class);
47 if (nodeCreateOrUpdate != null) {
48 for (Entry<InstanceIdentifier<Node>, Node> entry: nodeCreateOrUpdate.entrySet()) {
49 CheckedFuture<Optional<Node>, ReadFailedException> nodeFuture =
50 transaction.read(LogicalDatastoreType.OPERATIONAL, entry.getKey());
52 Optional<Node> nodeOptional = nodeFuture.get();
53 if (nodeOptional.isPresent()) {
54 operationalNodes.put(entry.getKey(), nodeOptional.get());
56 } catch (InterruptedException | ExecutionException e) {
57 LOG.warn("Error reading from datastore",e);
64 public Optional<Node> getGlobalNode(InstanceIdentifier<?> iid) {
65 InstanceIdentifier<Node> nodeIid = iid.firstIdentifierOf(Node.class);
66 return Optional.fromNullable(operationalNodes.get(nodeIid));
69 public Optional<LogicalSwitches> getLogicalSwitches(InstanceIdentifier<?> iid, LogicalSwitchesKey lswitchKey) {
70 Optional<Node> nodeOptional = getGlobalNode(iid);
71 if (nodeOptional.isPresent()) {
72 List<LogicalSwitches> lswitchList = nodeOptional.get().getAugmentation(HwvtepGlobalAugmentation.class).getLogicalSwitches();
73 if (lswitchList != null) {
74 for (LogicalSwitches lswitch: lswitchList) {
75 if (lswitch.getKey().equals(lswitchKey)) {
76 return Optional.fromNullable(lswitch);
81 return Optional.absent();
84 public Optional<PhysicalSwitchAugmentation> getPhysicalSwitchAugmentation(InstanceIdentifier<?> iid) {
85 Optional<Node> nodeOptional = getGlobalNode(iid);
86 if (nodeOptional.isPresent()) {
87 return Optional.fromNullable(nodeOptional.get().getAugmentation(PhysicalSwitchAugmentation.class));
89 return Optional.absent();
92 public Optional<LocatorSet> getPhysicalLocatorSet(InstanceIdentifier<?> iid) {
93 Optional<Node> nodeOptional = getGlobalNode(iid);
94 //TODO: physical locator set are under different logical switches
95 /*if (nodeOptional.isPresent()) {
96 return Optional.fromNullable(nodeOptional.get().getAugmentation(HwvtepGlobalAugmentation.class).getLogicalSwitches());
98 return Optional.absent();
101 public Optional<TerminationPoint> getHwvtepTerminationPoint(InstanceIdentifier<?> iid) {
103 Optional<Node> nodeOptional = getGlobalNode(iid);
104 if (nodeOptional.isPresent() && nodeOptional.get().getTerminationPoint() != null) {
105 TerminationPointKey key = iid.firstKeyOf(TerminationPoint.class, TerminationPointKey.class);
107 for (TerminationPoint tp:nodeOptional.get().getTerminationPoint()) {
108 if (tp.getKey().equals(key)) {
109 return Optional.of(tp);
115 return Optional.absent();
118 public Optional<HwvtepPhysicalLocatorAugmentation> getPhysicalLocatorAugmentation(InstanceIdentifier<?> iid) {
119 Optional<TerminationPoint> nodeOptional = getHwvtepTerminationPoint(iid);
120 if (nodeOptional.isPresent()) {
121 return Optional.fromNullable(nodeOptional.get().getAugmentation(HwvtepPhysicalLocatorAugmentation.class));
123 return Optional.absent();
126 public Optional<HwvtepPhysicalPortAugmentation> getPhysycalPortAugmentation(InstanceIdentifier<?> iid) {
127 Optional<TerminationPoint> tpOptional = getHwvtepTerminationPoint(iid);
128 if (tpOptional.isPresent()) {
129 return Optional.fromNullable(tpOptional.get().getAugmentation(HwvtepPhysicalPortAugmentation.class));
131 return Optional.absent();