Update Renderer Module
[transportpce.git] / renderer / src / test / java / org / opendaylight / transportpce / renderer / utils / WaveLengthServiceUtils.java
1 /*
2  * Copyright © 2018 Orange 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
9 package org.opendaylight.transportpce.renderer.utils;
10
11 import java.util.concurrent.ExecutionException;
12 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
13 import org.opendaylight.transportpce.common.NetworkUtils;
14 import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
15 //import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev170929.Node1;
16 //import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev170929.TerminationPoint1;
17 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev171215.Node1;
18 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev171215.TerminationPoint1;
19 import org.opendaylight.yang.gen.v1.http.org.openroadm.network.topology.rev171215.TerminationPoint1Builder;
20 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev150608.Network;
21 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev150608.NetworkId;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev150608.NetworkKey;
23 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev150608.NodeId;
24 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev150608.network.NodeKey;
25 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev150608.TpId;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev150608.network.node.TerminationPointKey;
27 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
28 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier.InstanceIdentifierBuilder;
29
30 public final class WaveLengthServiceUtils {
31
32     private WaveLengthServiceUtils() {
33
34     }
35
36     private static InstanceIdentifierBuilder<TerminationPoint1> createTerminationPoint1IIDBuilder(String nodeId,
37         String tpId) {
38         return InstanceIdentifier
39             .builder(Network.class, new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
40             .child(org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev150608.network
41                 .Node.class, new NodeKey(new NodeId(nodeId))).augmentation(
42                 org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev150608.Node1.class)
43             .child(org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.topology.rev150608
44                     .network.node.TerminationPoint.class,
45                 new TerminationPointKey(new TpId(tpId))).augmentation(TerminationPoint1.class);
46     }
47
48     public static void putTerminationPoint1ToDatastore(String nodeId, String tpId, TerminationPoint1 terminationPoint1,
49         DeviceTransactionManager deviceTransactionManager)
50         throws ExecutionException, InterruptedException {
51         TransactionUtils
52             .writeTransaction(deviceTransactionManager, nodeId, LogicalDatastoreType.CONFIGURATION,
53                 createTerminationPoint1IIDBuilder(nodeId, tpId).build(), terminationPoint1);
54     }
55
56     public static TerminationPoint1 getTerminationPoint1FromDatastore(String nodeId, String tpId,
57         DeviceTransactionManager deviceTransactionManager)
58         throws ExecutionException, InterruptedException {
59         InstanceIdentifier<TerminationPoint1> tpIID = createTerminationPoint1IIDBuilder(nodeId, tpId).build();
60         return (TerminationPoint1) TransactionUtils
61             .readTransaction(deviceTransactionManager, nodeId, LogicalDatastoreType.CONFIGURATION, tpIID);
62     }
63
64     private static InstanceIdentifier<Node1> createNode1IID(String nodeId) {
65         return InstanceIdentifier
66             .builder(Network.class, new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID)))
67             .child(org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev150608
68                 .network.Node.class, new NodeKey(new NodeId(nodeId)))
69             .augmentation(Node1.class)
70             .build();
71     }
72
73     public static void putNode1ToDatastore(String nodeId, Node1 node1,
74         DeviceTransactionManager deviceTransactionManager)
75         throws ExecutionException, InterruptedException {
76         InstanceIdentifier<Node1>
77             nodeIID = createNode1IID(nodeId);
78         TransactionUtils
79             .writeTransaction(deviceTransactionManager, nodeId,
80                 LogicalDatastoreType.CONFIGURATION, nodeIID, node1);
81     }
82
83     public static Node1 getNode1FromDatastore(String nodeId, DeviceTransactionManager deviceTransactionManager)
84         throws ExecutionException, InterruptedException {
85         InstanceIdentifier<Node1>
86             nodeIID = createNode1IID(nodeId);
87         return (Node1) TransactionUtils
88             .readTransaction(deviceTransactionManager, nodeId, LogicalDatastoreType.CONFIGURATION, nodeIID);
89     }
90 }