Refactor SupportedIfCapability usage
[transportpce.git] / common / src / main / java / org / opendaylight / transportpce / common / mapping / MappingUtilsImpl.java
1 /*
2  * Copyright © 2017 AT&T 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.transportpce.common.mapping;
9
10 import java.util.ArrayList;
11 import java.util.HashMap;
12 import java.util.List;
13 import java.util.Map;
14 import java.util.Optional;
15 import java.util.concurrent.ExecutionException;
16 import org.opendaylight.mdsal.binding.api.DataBroker;
17 import org.opendaylight.mdsal.binding.api.ReadTransaction;
18 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
19 import org.opendaylight.transportpce.common.StringConstants;
20 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220316.Network;
21 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220316.mc.capabilities.McCapabilities;
22 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220316.network.Nodes;
23 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220316.network.NodesKey;
24 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220316.network.nodes.NodeInfo;
25 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.If100GE;
26 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.If100GEODU4;
27 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.If10GE;
28 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.If10GEODU2;
29 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.If10GEODU2e;
30 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.If1GE;
31 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.If1GEODU0;
32 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.If400GE;
33 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.IfOCH;
34 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.IfOCHOTU2EODU2E;
35 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.IfOCHOTU2ODU2;
36 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.IfOCHOTU4ODU4;
37 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.IfOTUCnODUCn;
38 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.IfOtsiOtsigroup;
39 import org.opendaylight.yang.gen.v1.http.org.openroadm.port.types.rev201211.SupportedIfCapability;
40 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
41 import org.slf4j.Logger;
42 import org.slf4j.LoggerFactory;
43
44 public class MappingUtilsImpl implements MappingUtils {
45
46     private static final Logger LOG = LoggerFactory.getLogger(MappingUtilsImpl.class);
47
48     private final DataBroker dataBroker;
49
50     private static Map<String, Class<? extends SupportedIfCapability>> capTypeClassMap = new HashMap<>() {
51         {
52             put("IfOtsiOtsigroup", IfOtsiOtsigroup.class);
53             put("IfOTUCnODUCn", IfOTUCnODUCn.class);
54             put("IfOCHOTU4ODU4", IfOCHOTU4ODU4.class);
55             put("IfOCH", IfOCH.class);
56             put("If100GEODU4", If100GEODU4.class);
57             put("If10GEODU2e", If10GEODU2e.class);
58             put("If10GEODU2", If10GEODU2.class);
59             put("If1GEODU0", If1GEODU0.class);
60             put("If400GE", If400GE.class);
61             put("If100GE", If100GE.class);
62             put("If10GE", If10GE.class);
63             put("If1GE", If1GE.class);
64             put("IfOCHOTU2EODU2E", IfOCHOTU2EODU2E.class);
65             put("IfOCHOTU2ODU2", IfOCHOTU2ODU2.class);
66         }
67     };
68
69     public MappingUtilsImpl(DataBroker dataBroker) {
70
71         this.dataBroker = dataBroker;
72
73     }
74
75     public String getOpenRoadmVersion(String nodeId) {
76         /*
77          * Getting physical mapping corresponding to logical connection point
78          */
79         InstanceIdentifier<NodeInfo> nodeInfoIID = InstanceIdentifier.builder(Network.class).child(Nodes.class,
80                 new NodesKey(nodeId)).child(NodeInfo.class).build();
81         try (ReadTransaction readTx = dataBroker.newReadOnlyTransaction()) {
82             Optional<NodeInfo> nodeInfoObj =
83                     readTx.read(LogicalDatastoreType.CONFIGURATION, nodeInfoIID).get();
84             if (nodeInfoObj.isPresent()) {
85                 NodeInfo nodInfo = nodeInfoObj.get();
86                 switch (nodInfo.getOpenroadmVersion()) {
87                     case _71:
88                         return StringConstants.OPENROADM_DEVICE_VERSION_7_1;
89                     case _221:
90                         return StringConstants.OPENROADM_DEVICE_VERSION_2_2_1;
91                     case _121:
92                         return StringConstants.OPENROADM_DEVICE_VERSION_1_2_1;
93                     default:
94                         LOG.warn("unknown openROADM device version");
95                 }
96             } else {
97                 LOG.warn("Could not find mapping for nodeId {}", nodeId);
98             }
99         } catch (InterruptedException | ExecutionException ex) {
100             LOG.error("Unable to read mapping for nodeId {}",nodeId, ex);
101         }
102         return null;
103     }
104
105     /*
106     * (non-Javadoc)
107     *
108     * @see org.opendaylight.transportpce.common.mapping.MappingUtils#getMcCapabilitiesForNode(java.lang.String)
109     */
110     @Override
111     public List<McCapabilities> getMcCapabilitiesForNode(String nodeId) {
112         List<McCapabilities> mcCapabilities = new ArrayList<>();
113         InstanceIdentifier<Nodes> nodePortMappingIID = InstanceIdentifier.builder(Network.class)
114                 .child(Nodes.class, new NodesKey(nodeId)).build();
115         try (ReadTransaction readTx = this.dataBroker.newReadOnlyTransaction()) {
116             Optional<Nodes> nodePortMapObject = readTx.read(LogicalDatastoreType.CONFIGURATION, nodePortMappingIID)
117                     .get();
118             if (nodePortMapObject.isPresent()) {
119                 LOG.info("Found node {}", nodeId);
120                 Nodes node = nodePortMapObject.get();
121                 mcCapabilities.addAll(node.nonnullMcCapabilities().values());
122             }
123         } catch (ExecutionException e) {
124             LOG.error("Something went wrong while getting node {}", nodeId, e);
125         } catch (InterruptedException e) {
126             LOG.error("Request interrupted for node {} interrupted", nodeId, e);
127             Thread.currentThread().interrupt();
128         }
129         LOG.info("Capabilitities for node {}: {}", nodeId, mcCapabilities);
130         return mcCapabilities;
131     }
132
133     public static Class<? extends SupportedIfCapability> convertSupIfCapa(String ifCapType) {
134         if (!capTypeClassMap.containsKey(ifCapType)) {
135             LOG.error("supported-if-capability {} not supported", ifCapType);
136             return null;
137         }
138         return capTypeClassMap.get(ifCapType);
139     }
140 }