commit b20ba9e1e2415ac2b31b188c70835cb1c805fde8
author Vidyashree Rama <vidyashree.rama@huawei.com> Fri Sep 02 14:12:25 2016 +0530
committer Vidyashree-Huawei <vidyashree.rama@huawei.com> Tue Sep 06 07:00:05 2016 +0000
tree d519bb7874b3c2825f6a567f4d7839389d616d1c
parent 1edde62ac08f67ecb4d6d3ab3b055f1a913f8cf3

[ONOS-5186]adding isDescendant node augmented flag in YANG node to handle augment scenario in YMS

Change-Id: I39dc042825feb4a668933fca4ed67dfaa106da1b

plugin/src/main/java/org/onosproject/yangutils/linker/impl/YangResolutionInfoImpl.java

diff --git a/plugin/src/main/java/org/onosproject/yangutils/linker/impl/YangResolutionInfoImpl.java b/plugin/src/main/java/org/onosproject/yangutils/linker/impl/YangResolutionInfoImpl.java
index f0d4608..1281bb8 100644
--- a/plugin/src/main/java/org/onosproject/yangutils/linker/impl/YangResolutionInfoImpl.java
+++ b/plugin/src/main/java/org/onosproject/yangutils/linker/impl/YangResolutionInfoImpl.java
@@ -1249,8 +1249,8 @@
                 if (targetNode instanceof YangAugmentableNode) {
                     detectCollisionForAugmentedNode(targetNode, augment);
                     ((YangAugmentableNode) targetNode).addAugmentation(augment);
-                    ((YangAugmentableNode) targetNode).setIsAugmented(true);
                     augment.setAugmentedNode(targetNode);
+                    setAugmentedFlagInAncestors(targetNode);
                     Resolvable resolvable = (Resolvable) entityToResolve;
                     resolvable.setResolvableStatus(RESOLVED);
                 } else {
@@ -1842,4 +1842,16 @@
         return null;
     }
 
+    /**
+     * Sets descendant node augmented flag in ancestors.
+     *
+     * @param targetNode augmented YANG node
+     */
+    private void setAugmentedFlagInAncestors(YangNode targetNode) {
+        targetNode = targetNode.getParent();
+        while (targetNode != null) {
+            targetNode.setDescendantNodeAugmented(true);
+            targetNode = targetNode.getParent();
+        }
+    }
 }