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;
14 import java.util.Map.Entry;
15 import java.util.concurrent.ExecutionException;
17 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
18 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
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.binding.api.DataObjectModification.ModificationType;
22 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
23 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.HwvtepGlobalAugmentation;
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.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 extractCreatedOrUpdatedOrRemoved(changes, Node.class);
47 //TransactUtils.extractCreatedOrUpdatedOrRemoved(changes, Node.class);
48 if (nodeCreateOrUpdate != null) {
49 for (Entry<InstanceIdentifier<Node>, Node> entry: nodeCreateOrUpdate.entrySet()) {
50 CheckedFuture<Optional<Node>, ReadFailedException> nodeFuture =
51 transaction.read(LogicalDatastoreType.OPERATIONAL, entry.getKey());
53 Optional<Node> nodeOptional = nodeFuture.get();
54 if (nodeOptional.isPresent()) {
55 operationalNodes.put(entry.getKey(), nodeOptional.get());
57 } catch (InterruptedException | ExecutionException e) {
58 LOG.warn("Error reading from datastore",e);
65 private Node getCreated(DataObjectModification<Node> mod) {
66 if((mod.getModificationType() == ModificationType.WRITE)
67 && (mod.getDataBefore() == null)){
68 return mod.getDataAfter();
73 private Node getRemoved(DataObjectModification<Node> mod) {
74 if(mod.getModificationType() == ModificationType.DELETE){
75 return mod.getDataBefore();
80 private Node getUpdated(DataObjectModification<Node> mod) {
82 switch(mod.getModificationType()) {
83 case SUBTREE_MODIFIED:
84 node = mod.getDataAfter();
87 if(mod.getDataBefore() != null) {
88 node = mod.getDataAfter();
97 private Node getOriginal(DataObjectModification<Node> mod) {
99 switch(mod.getModificationType()) {
100 case SUBTREE_MODIFIED:
101 node = mod.getDataBefore();
104 if(mod.getDataBefore() != null) {
105 node = mod.getDataBefore();
109 node = mod.getDataBefore();
117 private Map<InstanceIdentifier<Node>, Node> extractCreatedOrUpdatedOrRemoved(
118 Collection<DataTreeModification<Node>> changes, Class<Node> class1) {
119 // TODO Auto-generated method stub
120 Map<InstanceIdentifier<Node>, Node> result = new HashMap<InstanceIdentifier<Node>, Node>();
121 for (DataTreeModification<Node> change : changes) {
122 final InstanceIdentifier<Node> key = change.getRootPath().getRootIdentifier();
123 final DataObjectModification<Node> mod = change.getRootNode();
124 Node created = getCreated(mod);
125 result.put(key, created);
126 Node updated = getUpdated(mod);
127 result.put(key, updated);
128 Node deleted = getRemoved(mod);
129 result.put(key, deleted);
134 public Optional<Node> getGlobalNode(InstanceIdentifier<?> iid) {
135 InstanceIdentifier<Node> nodeIid = iid.firstIdentifierOf(Node.class);
136 return Optional.fromNullable(operationalNodes.get(nodeIid));
139 public Optional<LogicalSwitches> getLogicalSwitches(InstanceIdentifier<?> iid) {
140 Optional<Node> nodeOptional = getGlobalNode(iid);
141 /*if (nodeOptional.isPresent()) {
142 return Optional.fromNullable(nodeOptional.get().getAugmentation(HwvtepLogicalSwitchAugmentation.class));
144 return Optional.absent();
147 public Optional<PhysicalSwitchAugmentation> getPhysicalSwitchAugmentation(InstanceIdentifier<?> iid) {
148 Optional<Node> nodeOptional = getGlobalNode(iid);
149 if (nodeOptional.isPresent()) {
150 return Optional.fromNullable(nodeOptional.get().getAugmentation(PhysicalSwitchAugmentation.class));
152 return Optional.absent();
155 public Optional<LocatorSet> getPhysicalLocatorSet(InstanceIdentifier<?> iid) {
156 Optional<Node> nodeOptional = getGlobalNode(iid);
157 /*if (nodeOptional.isPresent()) {
158 return Optional.fromNullable(nodeOptional.get().getAugmentation(HwvtepGlobalAugmentation.class).getLogicalSwitches());
160 return Optional.absent();
163 public Optional<TerminationPoint> getHwvtepTerminationPoint(InstanceIdentifier<?> iid) {
165 Optional<Node> nodeOptional = getGlobalNode(iid);
166 if (nodeOptional.isPresent() && nodeOptional.get().getTerminationPoint() != null) {
167 TerminationPointKey key = iid.firstKeyOf(TerminationPoint.class, TerminationPointKey.class);
169 for (TerminationPoint tp:nodeOptional.get().getTerminationPoint()) {
170 if (tp.getKey().equals(key)) {
171 return Optional.of(tp);
177 return Optional.absent();
180 public Optional<HwvtepPhysicalLocatorAugmentation> getPhysicalLocatorAugmentation(InstanceIdentifier<?> iid) {
181 Optional<TerminationPoint> nodeOptional = getHwvtepTerminationPoint(iid);
182 if (nodeOptional.isPresent()) {
183 return Optional.fromNullable(nodeOptional.get().getAugmentation(HwvtepPhysicalLocatorAugmentation.class));
185 return Optional.absent();
188 public Optional<HwvtepPhysicalPortAugmentation> getPhysycalPortAugmentation(InstanceIdentifier<?> iid) {
189 Optional<TerminationPoint> tpOptional = getHwvtepTerminationPoint(iid);
190 if (tpOptional.isPresent()) {
191 return Optional.fromNullable(tpOptional.get().getAugmentation(HwvtepPhysicalPortAugmentation.class));
193 return Optional.absent();