Simplify getBridgeNode()
[ovsdb.git] / southbound / southbound-impl / src / main / java / org / opendaylight / ovsdb / southbound / ovsdb / transact / BridgeOperationalState.java
1 /*
2  * Copyright (c) 2015 Cisco Systems, Inc. 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 package org.opendaylight.ovsdb.southbound.ovsdb.transact;
9
10 import java.util.Collection;
11 import java.util.Map;
12 import java.util.Optional;
13 import org.opendaylight.mdsal.binding.api.DataBroker;
14 import org.opendaylight.mdsal.binding.api.DataTreeModification;
15 import org.opendaylight.mdsal.binding.api.ReadTransaction;
16 import org.opendaylight.ovsdb.southbound.SouthboundUtil;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbBridgeAugmentation;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.OvsdbTerminationPointAugmentation;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ControllerEntry;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ControllerEntryKey;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ProtocolEntry;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.rev150105.ovsdb.bridge.attributes.ProtocolEntryKey;
23 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
24 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
25 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPointKey;
26 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
27 import org.slf4j.Logger;
28 import org.slf4j.LoggerFactory;
29
30 public class BridgeOperationalState {
31     private static final Logger LOG = LoggerFactory.getLogger(BridgeOperationalState.class);
32     private final DataBroker db;
33
34     public BridgeOperationalState(DataBroker db, DataChangeEvent changes) {
35         this.db = db;
36     }
37
38     public BridgeOperationalState(DataBroker db, Collection<DataTreeModification<Node>> changes) {
39         this.db = db;
40     }
41
42     @SuppressWarnings("IllegalCatch")
43     public Optional<Node> getBridgeNode(InstanceIdentifier<?> iid) {
44         InstanceIdentifier<Node> nodeIid = iid.firstIdentifierOf(Node.class);
45         try (ReadTransaction transaction = db.newReadOnlyTransaction()) {
46             return SouthboundUtil.readNode(transaction, nodeIid);
47         } catch (Exception exp) {
48             LOG.error("Error in getting the brideNode for {}", iid, exp);
49             return Optional.empty();
50         }
51     }
52
53     public Optional<OvsdbBridgeAugmentation> getOvsdbBridgeAugmentation(InstanceIdentifier<?> iid) {
54         return getBridgeNode(iid)
55             .flatMap(node -> Optional.ofNullable(node.augmentation(OvsdbBridgeAugmentation.class)));
56     }
57
58     public Optional<TerminationPoint> getBridgeTerminationPoint(InstanceIdentifier<?> iid) {
59         if (iid != null) {
60             Optional<Node> nodeOptional = getBridgeNode(iid);
61             if (nodeOptional.isPresent()) {
62                 Node node = nodeOptional.orElseThrow();
63                 TerminationPointKey key = iid.firstKeyOf(TerminationPoint.class);
64                 if (key != null) {
65                     final TerminationPoint tp = node.nonnullTerminationPoint().get(key);
66                     if (tp != null) {
67                         return Optional.of(tp);
68                     }
69                 }
70             } else {
71                 LOG.debug("TerminationPoints or Operational BridgeNode missing for {}", iid);
72             }
73         }
74         return Optional.empty();
75     }
76
77     public Optional<OvsdbTerminationPointAugmentation> getOvsdbTerminationPointAugmentation(InstanceIdentifier<?> iid) {
78         return getBridgeTerminationPoint(iid)
79             .flatMap(tp -> Optional.ofNullable(tp.augmentation(OvsdbTerminationPointAugmentation.class)));
80     }
81
82     public Optional<ControllerEntry> getControllerEntry(InstanceIdentifier<?> iid) {
83         if (iid != null) {
84             Optional<OvsdbBridgeAugmentation> ovsdbBridgeOptional = getOvsdbBridgeAugmentation(iid);
85             if (ovsdbBridgeOptional.isPresent()) {
86                 Map<ControllerEntryKey, ControllerEntry> entries =
87                     ovsdbBridgeOptional.orElseThrow().getControllerEntry();
88                 if (entries != null) {
89                     ControllerEntryKey key = iid.firstKeyOf(ControllerEntry.class);
90                     if (key != null) {
91                         ControllerEntry entry = entries.get(key);
92                         if (entry != null) {
93                             return Optional.of(entry);
94                         }
95                     }
96                 }
97             }
98         }
99         return Optional.empty();
100     }
101
102     public Optional<ProtocolEntry> getProtocolEntry(InstanceIdentifier<ProtocolEntry> iid) {
103         if (iid != null) {
104             Optional<OvsdbBridgeAugmentation> ovsdbBridgeOptional = getOvsdbBridgeAugmentation(iid);
105             if (ovsdbBridgeOptional.isPresent()) {
106                 Map<ProtocolEntryKey, ProtocolEntry> entries = ovsdbBridgeOptional.orElseThrow().getProtocolEntry();
107                 if (entries != null) {
108                     ProtocolEntryKey key = iid.firstKeyOf(ProtocolEntry.class);
109                     if (key != null) {
110                         ProtocolEntry entry = entries.get(key);
111                         if (entry != null) {
112                             return Optional.of(entry);
113                         }
114                     }
115                 }
116             }
117         }
118         return Optional.empty();
119     }
120 }