package ietf-network 2015-06-08 yang models

[transportpce.git] / pce / src / main / java / org / opendaylight / transportpce / pce / PceCalculation.java

diff --git a/pce/src/main/java/org/opendaylight/transportpce/pce/PceCalculation.java b/pce/src/main/java/org/opendaylight/transportpce/pce/PceCalculation.java
index 3f6bccdee90d96acbf430a8c8eba9e3be373a649..aede67294f734b894fc1ab38cc10e7c66ca9d958 100644 (file)
--- a/pce/src/main/java/org/opendaylight/transportpce/pce/PceCalculation.java
+++ b/pce/src/main/java/org/opendaylight/transportpce/pce/PceCalculation.java
@@ -9,14 +9,12 @@
 package org.opendaylight.transportpce.pce;
 
 import com.google.common.base.Optional;
-
 import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
 import java.util.concurrent.ExecutionException;
-
 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
@@ -149,7 +147,7 @@ public class PceCalculation {
             return false;
         }
         this.allNodes = nw.getNode();
-        Network1 nw1 = nw.getAugmentation(Network1.class);
+        Network1 nw1 = nw.augmentation(Network1.class);
 
         this.allLinks = nw1.getLink();
         if ((this.allNodes == null) || this.allNodes.isEmpty()) {
@@ -353,7 +351,7 @@ public class PceCalculation {
             }
 
             // extract node type
-            Node1 node1 = node.getAugmentation(Node1.class);
+            Node1 node1 = node.augmentation(Node1.class);
             if (node1 == null) {
                 LOG.error("validateNode: no Node1 (type) Augmentation for node: [{}]. Node is ignored",
                         nodeId.getValue());