upgrade portmapping YANG to introduce OTN
[transportpce.git] / networkmodel / src / main / java / org / opendaylight / transportpce / networkmodel / util / TopologyUtils.java
1 /*
2  * Copyright © 2019 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.networkmodel.util;
9
10 import java.util.concurrent.ExecutionException;
11 import java.util.concurrent.TimeUnit;
12 import java.util.concurrent.TimeoutException;
13
14 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
15 import org.opendaylight.transportpce.common.NetworkUtils;
16 import org.opendaylight.transportpce.common.network.NetworkTransactionService;
17 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev181130.Link1;
18 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.network.rev181130.Link1Builder;
19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.Networks;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.Network;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.NetworkKey;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.LinkId;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.Network1;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.Link;
27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.LinkBuilder;
28 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.LinkKey;
29 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.link.DestinationBuilder;
30 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev180226.networks.network.link.SourceBuilder;
31 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
32 import org.slf4j.Logger;
33 import org.slf4j.LoggerFactory;
34
35 public final class TopologyUtils {
36
37     private static final Logger LOG = LoggerFactory.getLogger(TopologyUtils.class);
38
39     private TopologyUtils() {
40     }
41
42     // This method returns the linkBuilder object for given source and destination
43     public static LinkBuilder createLink(String srcNode, String dstNode, String srcTp, String destTp) {
44
45         // Create Destination for link
46         DestinationBuilder dstNodeBldr = new DestinationBuilder()
47             .setDestTp(destTp)
48             .setDestNode(new NodeId(dstNode));
49
50         // Create Source for the link
51         SourceBuilder srcNodeBldr = new SourceBuilder()
52             .setSourceNode(new NodeId(srcNode))
53             .setSourceTp(srcTp);
54
55         // set link builder attribute
56         LinkBuilder lnkBldr = new LinkBuilder()
57             .setDestination(dstNodeBldr.build())
58             .setSource(srcNodeBldr.build())
59             .setLinkId(LinkIdUtil.buildLinkId(srcNode, srcTp, dstNode, destTp));
60         lnkBldr.withKey(new LinkKey(lnkBldr.getLinkId()));
61
62         //set opposite link
63         LinkId oppositeLinkId = LinkIdUtil.getOppositeLinkId(srcNode, srcTp, dstNode, destTp);
64         Link1 lnk1 = new Link1Builder().setOppositeLink(oppositeLinkId).build();
65         lnkBldr.addAugmentation(Link1.class,lnk1);
66         return lnkBldr;
67     }
68
69     // This method returns the linkBuilder object for given source and destination
70     public static boolean deleteLink(String srcNode, String dstNode, String srcTp, String destTp,
71                                      NetworkTransactionService networkTransactionService) {
72         LOG.info("deleting link for {}-{}", srcNode, dstNode);
73         LinkId linkId = LinkIdUtil.buildLinkId(srcNode, srcTp, dstNode, destTp);
74         if (deleteLinkLinkId(linkId, networkTransactionService)) {
75             LOG.debug("Link Id {} updated to have admin state down");
76             return true;
77         } else {
78             LOG.debug("Link Id not found for Source {} and Dest {}", srcNode, dstNode);
79             return false;
80         }
81     }
82
83     // This method returns the linkBuilder object for given source and destination
84     public static boolean deleteLinkLinkId(LinkId linkId , NetworkTransactionService networkTransactionService) {
85         LOG.info("deleting link for LinkId: {}", linkId);
86         try {
87             InstanceIdentifier.InstanceIdentifierBuilder<Link> linkIID = InstanceIdentifier.builder(Networks.class)
88                 .child(Network.class, new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
89                 .augmentation(Network1.class).child(Link.class, new LinkKey(linkId));
90             java.util.Optional<Link> link =
91                 networkTransactionService.read(LogicalDatastoreType.CONFIGURATION,linkIID.build()).get();
92             if (link.isPresent()) {
93                 LinkBuilder linkBuilder = new LinkBuilder(link.get());
94                 Link1Builder link1Builder = new Link1Builder(linkBuilder.augmentation(Link1.class));
95                 linkBuilder.removeAugmentation(Link1.class);
96                 linkBuilder.addAugmentation(Link1.class,link1Builder.build());
97                 networkTransactionService.merge(LogicalDatastoreType.CONFIGURATION, linkIID.build(),
98                     linkBuilder.build());
99                 networkTransactionService.commit().get(1, TimeUnit.SECONDS);
100                 return true;
101             } else {
102                 LOG.error("No link found for given LinkId: {}",
103                     linkId);
104                 return false;
105             }
106
107         } catch (InterruptedException | ExecutionException | TimeoutException e) {
108             LOG.error(e.getMessage(), e);
109             return false;
110         }
111     }
112 }