Bug 1250 - increase UT coverage
[openflowplugin.git] / openflowplugin / src / main / java / org / opendaylight / openflowplugin / openflow / md / util / InventoryDataServiceUtil.java
1 /**
2  * Copyright (c) 2013 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.openflowplugin.openflow.md.util;
9
10 import com.google.common.base.Optional;
11 import java.math.BigInteger;
12 import java.util.List;
13 import java.util.concurrent.ExecutionException;
14 import org.apache.commons.lang3.StringUtils;
15 import org.opendaylight.controller.md.sal.binding.api.ReadTransaction;
16 import org.opendaylight.controller.md.sal.binding.api.ReadWriteTransaction;
17 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
18 import org.opendaylight.openflowplugin.openflow.md.core.session.OFSessionUtil;
19 import org.opendaylight.openflowplugin.api.openflow.md.util.OpenflowVersion;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorId;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorRef;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorUpdatedBuilder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeRef;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeUpdatedBuilder;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnector;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.node.NodeConnectorKey;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
32 import org.opendaylight.yangtools.yang.binding.DataObject;
33 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
34 import org.slf4j.Logger;
35 import org.slf4j.LoggerFactory;
36
37 public abstract class InventoryDataServiceUtil {
38     private final static Logger LOG = LoggerFactory.getLogger(InventoryDataServiceUtil.class);
39
40     public final static String OF_URI_PREFIX = "openflow:";
41     /*
42      * Get an InstanceIdentifier for the Nodes class that is the root of the
43      * inventory tree We use this alot, so its worth keeping around
44      */
45     private static InstanceIdentifier<Nodes> nodesIdentifier = InstanceIdentifier.create(Nodes.class);
46
47     public static Nodes checkForNodes() {
48         Nodes nodes = null;
49         LOG.error("Before Nodes - nodes: " + nodes);
50         try {
51             nodes = getDataObject(OFSessionUtil.getSessionManager().getDataBroker().newReadOnlyTransaction(), nodesIdentifier);
52         } catch (Exception e) {
53             LOG.error(
54                     "Caught exception from OFSessionUtil.getSessionManager().getDataBroker().newReadWriteTransaction()",
55                     e);
56         }
57         LOG.error("After Nodes- nodes: " + nodes);
58         return nodes;
59     }
60
61     public static List<Node> readAllNodes() {
62         Nodes nodes = getDataObject(OFSessionUtil.getSessionManager().getDataBroker().newReadOnlyTransaction(), nodesIdentifier);
63         return nodes.getNode();
64     }
65
66     public static Node readNode(InstanceIdentifier<Node> instance) {
67         return (Node) getDataObject(OFSessionUtil.getSessionManager().getDataBroker().newReadOnlyTransaction(), instance);
68     }
69
70     public static void putNodeConnector(InstanceIdentifier<Node> instance, NodeConnector nodeConnector) {
71         ReadWriteTransaction transaction = OFSessionUtil.getSessionManager().getDataBroker().newReadWriteTransaction();
72         InstanceIdentifier<NodeConnector> nodeConnectorID = instance.child(NodeConnector.class, nodeConnector.getKey());
73         transaction.merge(LogicalDatastoreType.OPERATIONAL, nodeConnectorID, nodeConnector);
74         transaction.submit();
75     }
76
77     public static void putNodeConnector(NodeKey nodeKey, NodeConnector nodeConnector) {
78         InstanceIdentifier<Node> instance = nodeKeyToInstanceIdentifier(nodeKey);
79         putNodeConnector(instance, nodeConnector);
80     }
81
82     public static void putNodeConnector(NodeId nodeId, NodeConnector nodeConnector) {
83         putNodeConnector(new NodeKey(nodeId), nodeConnector);
84     }
85
86     public static InstanceIdentifier<Node> identifierFromDatapathId(BigInteger datapathId) {
87         NodeKey nodeKey = nodeKeyFromDatapathId(datapathId);
88         return InstanceIdentifier.builder(Nodes.class).child(Node.class, nodeKey).toInstance();
89     }
90
91     public static NodeKey nodeKeyFromDatapathId(BigInteger datapathId) {
92         return new NodeKey(nodeIdFromDatapathId(datapathId));
93     }
94
95     public static NodeUpdatedBuilder nodeUpdatedBuilderFromDataPathId(BigInteger datapathId) {
96         NodeUpdatedBuilder builder = new NodeUpdatedBuilder();
97         builder.setId(nodeIdFromDatapathId(datapathId));
98         builder.setNodeRef(nodeRefFromNodeKey(new NodeKey(builder.getId())));
99         return builder;
100     }
101
102     public static NodeId nodeIdFromDatapathId(BigInteger datapathId) {
103         // FIXME: Convert to textual representation of datapathID
104         String current = datapathId.toString();
105         return new NodeId(OF_URI_PREFIX + current);
106     }
107
108     public static BigInteger dataPathIdFromNodeId(NodeId nodeId) {
109         String dpids = nodeId.getValue().replace(OF_URI_PREFIX, "");
110         BigInteger dpid = new BigInteger(dpids);
111         return dpid;
112     }
113
114
115     public static NodeRef nodeRefFromNodeKey(NodeKey nodeKey) {
116         return new NodeRef(nodeKeyToInstanceIdentifier(nodeKey));
117     }
118
119     public static InstanceIdentifier<Node> nodeKeyToInstanceIdentifier(NodeKey nodeKey) {
120         return InstanceIdentifier.builder(Nodes.class).child(Node.class, nodeKey).toInstance();
121     }
122
123     public static NodeConnectorId nodeConnectorIdfromDatapathPortNo(BigInteger datapathid, Long portNo,
124                                                                     OpenflowVersion ofVersion) {
125         String logicalName = OpenflowPortsUtil.getPortLogicalName(ofVersion, portNo);
126         return new NodeConnectorId(OF_URI_PREFIX + datapathid + ":" + (logicalName == null ? portNo : logicalName));
127     }
128
129     public static Long portNumberfromNodeConnectorId(OpenflowVersion ofVersion, NodeConnectorId ncId) {
130         return portNumberfromNodeConnectorId(ofVersion, ncId.getValue());
131     }
132
133     public static Long portNumberfromNodeConnectorId(OpenflowVersion ofVersion, String ncId) {
134         String[] split = ncId.split(":");
135
136         // It can happen that token length will be just 1 i.e 2 or CONTROLLER
137         // If the length is just one then this cannot be the new MD-SAL style node connector Id which
138         // is of the form openflow:1:3.
139
140         String portNoString = split[split.length - 1];
141         Long portNo = OpenflowPortsUtil.getPortFromLogicalName(ofVersion, portNoString);
142         return portNo;
143     }
144
145
146     public static NodeConnectorRef nodeConnectorRefFromDatapathIdPortno(BigInteger datapathId, Long portNo, OpenflowVersion ofVersion) {
147         return new NodeConnectorRef(nodeConnectorInstanceIdentifierFromDatapathIdPortno(datapathId, portNo, ofVersion));
148     }
149
150     public static InstanceIdentifier<NodeConnector> nodeConnectorInstanceIdentifierFromDatapathIdPortno(
151             BigInteger datapathId, Long portNo, OpenflowVersion ofVersion) {
152         NodeId nodeId = nodeIdFromDatapathId(datapathId);
153         NodeConnectorId nodeConnectorId = nodeConnectorIdfromDatapathPortNo(datapathId, portNo, ofVersion);
154         return InstanceIdentifier.builder(Nodes.class) //
155                 .child(Node.class, new NodeKey(nodeId)) //
156                 .child(NodeConnector.class, new NodeConnectorKey(nodeConnectorId)).toInstance();
157     }
158
159     public static NodeConnectorUpdatedBuilder nodeConnectorUpdatedBuilderFromDatapathIdPortNo(BigInteger datapathId,
160                                                                                               Long portNo, OpenflowVersion ofVersion) {
161         NodeConnectorUpdatedBuilder builder = new NodeConnectorUpdatedBuilder();
162         builder.setId(InventoryDataServiceUtil.nodeConnectorIdfromDatapathPortNo(datapathId, portNo, ofVersion));
163         builder.setNodeConnectorRef(InventoryDataServiceUtil.nodeConnectorRefFromDatapathIdPortno(datapathId, portNo, ofVersion));
164         return builder;
165     }
166
167     public static NodeConnectorBuilder nodeConnectorBuilderFromDatapathIdPortNo(BigInteger datapathId,
168                                                                                 Long portNo, OpenflowVersion ofVersion) {
169         NodeConnectorBuilder builder = new NodeConnectorBuilder();
170         builder.setId(InventoryDataServiceUtil.nodeConnectorIdfromDatapathPortNo(datapathId, portNo, ofVersion));
171         return builder;
172     }
173
174     /**
175      * @param dataPathId
176      * @return string of size 16, padded with '0'
177      */
178     public static String bigIntegerToPaddedHex(BigInteger dataPathId) {
179         return StringUtils.leftPad(dataPathId.toString(16), 16, "0");
180     }
181
182     //TODO : create new module openflowplugin-util, move there this method along with TestProviderTransactionUtil#getDataObject
183     private static <T extends DataObject> T getDataObject(ReadTransaction readOnlyTransaction, InstanceIdentifier<T> identifier) {
184         Optional<T> optionalData = null;
185         try {
186             optionalData = readOnlyTransaction.read(LogicalDatastoreType.OPERATIONAL, identifier).get();
187             if (optionalData.isPresent()) {
188                 return optionalData.get();
189             }
190         } catch (ExecutionException | InterruptedException e) {
191             LOG.error("Read transaction for identifier {} failed.", identifier, e);
192         }
193         return null;
194     }
195
196 }