Refactor SupportedIfCapability usage
[transportpce.git] / networkmodel / src / main / java / org / opendaylight / transportpce / networkmodel / util / TopologyUtils.java
index af2646f10b696bc6bc772158e289e7f0b9f2e9ca..9cca982f5d385ece698573c487849cabf06386db 100644 (file)
@@ -7,26 +7,43 @@
  */
 package org.opendaylight.transportpce.networkmodel.util;
 
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
+import java.util.stream.Collectors;
 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
 import org.opendaylight.transportpce.common.NetworkUtils;
 import org.opendaylight.transportpce.common.network.NetworkTransactionService;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev181130.Link1;
-import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev181130.Link1Builder;
+import org.opendaylight.transportpce.networkmodel.dto.TopologyShard;
+import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220316.mapping.Mapping;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev211210.Link1;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev211210.Link1Builder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev211210.TerminationPoint1;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev211210.TerminationPoint1Builder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.state.types.rev191129.State;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.equipment.states.types.rev191129.AdminStates;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.Networks;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.Network;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.NetworkKey;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.Node;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.network.NodeKey;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.LinkId;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Network1;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Node1;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.TpId;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.Link;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.LinkBuilder;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.LinkKey;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.link.DestinationBuilder;
 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.link.SourceBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.node.TerminationPoint;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.node.TerminationPointBuilder;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.node.TerminationPointKey;
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -44,13 +61,13 @@ public final class TopologyUtils {
 
         // Create Destination for link
         DestinationBuilder dstNodeBldr = new DestinationBuilder()
-            .setDestTp(destTp)
+            .setDestTp(new TpId(destTp))
             .setDestNode(new NodeId(dstNode));
 
         // Create Source for the link
         SourceBuilder srcNodeBldr = new SourceBuilder()
             .setSourceNode(new NodeId(srcNode))
-            .setSourceTp(srcTp);
+            .setSourceTp(new TpId(srcTp));
 
         LinkId linkId;
         LinkId oppositeLinkId;
@@ -117,4 +134,99 @@ public final class TopologyUtils {
             return false;
         }
     }
+
+    public static AdminStates setNetworkAdminState(String adminState) {
+        if (adminState == null) {
+            return null;
+        }
+        switch (adminState) {
+            case "InService":
+                return AdminStates.InService;
+            case "OutOfService":
+                return AdminStates.OutOfService;
+            case "Maintenance":
+                return AdminStates.Maintenance;
+            default:
+                return null;
+        }
+    }
+
+    public static State setNetworkOperState(String operState) {
+        if (operState == null) {
+            return null;
+        }
+        switch (operState) {
+            case "InService":
+                return State.InService;
+            case "OutOfService":
+                return State.OutOfService;
+            case "Degraded":
+                return State.Degraded;
+            default:
+                return null;
+        }
+    }
+
+    public static TopologyShard updateTopologyShard(String abstractNodeid, Mapping mapping, Map<NodeKey, Node> nodes,
+            Map<LinkKey, Link> links) {
+        // update termination-point corresponding to the mapping
+        List<TerminationPoint> topoTps = new ArrayList<>();
+        TerminationPoint tp = nodes.get(new NodeKey(new NodeId(abstractNodeid))).augmentation(Node1.class)
+            .getTerminationPoint().get(new TerminationPointKey(new TpId(mapping.getLogicalConnectionPoint())));
+        TerminationPoint1Builder tp1Bldr = new TerminationPoint1Builder(tp.augmentation(TerminationPoint1.class));
+        if (!tp1Bldr.getAdministrativeState().getName().equals(mapping.getPortAdminState())) {
+            tp1Bldr.setAdministrativeState(AdminStates.valueOf(mapping.getPortAdminState()));
+        }
+        if (!tp1Bldr.getOperationalState().getName().equals(mapping.getPortOperState())) {
+            tp1Bldr.setOperationalState(State.valueOf(mapping.getPortOperState()));
+        }
+        TerminationPointBuilder tpBldr = new TerminationPointBuilder(tp).addAugmentation(tp1Bldr.build());
+        topoTps.add(tpBldr.build());
+
+        if (links == null) {
+            return new TopologyShard(null, null, topoTps);
+        }
+        String tpId = tpBldr.getTpId().getValue();
+        // update links terminating on the given termination-point
+        List<Link> filteredTopoLinks = links.values().stream()
+            .filter(l1 -> (l1.getSource().getSourceNode().getValue().equals(abstractNodeid)
+                && l1.getSource().getSourceTp().getValue().equals(tpId))
+                || (l1.getDestination().getDestNode().getValue().equals(abstractNodeid)
+                && l1.getDestination().getDestTp().getValue().equals(tpId)))
+            .collect(Collectors.toList());
+        List<Link> topoLinks = new ArrayList<>();
+        for (Link link : filteredTopoLinks) {
+            TerminationPoint otherLinkTp;
+            if (link.getSource().getSourceNode().getValue().equals(abstractNodeid)) {
+                otherLinkTp = nodes
+                    .get(new NodeKey(new NodeId(link.getDestination().getDestNode().getValue())))
+                    .augmentation(Node1.class)
+                    .getTerminationPoint()
+                    .get(new TerminationPointKey(new TpId(link.getDestination().getDestTp().getValue())));
+            } else {
+                otherLinkTp = nodes
+                    .get(new NodeKey(new NodeId(link.getSource().getSourceNode().getValue())))
+                    .augmentation(Node1.class)
+                    .getTerminationPoint()
+                    .get(new TerminationPointKey(new TpId(link.getSource().getSourceTp().getValue())));
+            }
+            Link1Builder link1Bldr = new Link1Builder(link.augmentation(Link1.class));
+            if (tpBldr.augmentation(TerminationPoint1.class).getAdministrativeState().equals(AdminStates.InService)
+                && otherLinkTp.augmentation(TerminationPoint1.class)
+                    .getAdministrativeState().equals(AdminStates.InService)) {
+                link1Bldr.setAdministrativeState(AdminStates.InService);
+            } else {
+                link1Bldr.setAdministrativeState(AdminStates.OutOfService);
+            }
+            if (tpBldr.augmentation(TerminationPoint1.class).getOperationalState().equals(State.InService)
+                && otherLinkTp.augmentation(TerminationPoint1.class)
+                    .getOperationalState().equals(State.InService)) {
+                link1Bldr.setOperationalState(State.InService);
+            } else {
+                link1Bldr.setOperationalState(State.OutOfService);
+            }
+            topoLinks.add(new LinkBuilder(link).addAugmentation(link1Bldr.build()).build());
+        }
+        return new TopologyShard(null, topoLinks, topoTps);
+    }
 }