[ONOS-5186]adding isDescendant node augmented flag in YANG node to handle augment scenario in YMS
Change-Id: I39dc042825feb4a668933fca4ed67dfaa106da1b
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();
+ }
+ }
}