commit d2c7f8d58da3bd19c73b4d9e2ebe075cbf6a273e
author Vidyashree Rama <vidyashree.rama@huawei.com> Wed Nov 02 12:38:12 2016 +0530
committer Vidyashree-Huawei <vidyashree.rama@huawei.com> Thu Nov 03 07:31:17 2016 +0000
tree b566ab11105c9db0d40c02c173be925dba4d72d7
parent 110f7f76e0499c2f04a0edcfccde0636bca82287

[ONOS-5518] Augment create issue

Change-Id: Ib602521d6f812c19e47af315aa29282b882a6b2c

generator/src/main/java/org/onosproject/yangutils/translator/tojava/utils/JavaFileGeneratorUtils.java

diff --git a/generator/src/main/java/org/onosproject/yangutils/translator/tojava/utils/JavaFileGeneratorUtils.java b/generator/src/main/java/org/onosproject/yangutils/translator/tojava/utils/JavaFileGeneratorUtils.java
index d69f348..b48f48a 100644
--- a/generator/src/main/java/org/onosproject/yangutils/translator/tojava/utils/JavaFileGeneratorUtils.java
+++ b/generator/src/main/java/org/onosproject/yangutils/translator/tojava/utils/JavaFileGeneratorUtils.java
@@ -87,6 +87,7 @@
 import static org.onosproject.yangutils.translator.tojava.JavaQualifiedTypeInfoTranslator.getQualifiedTypeInfoOfCurNode;
 import static org.onosproject.yangutils.translator.tojava.YangJavaModelUtils.getNodesPackage;
 import static org.onosproject.yangutils.translator.tojava.utils.ClassDefinitionGenerator.generateClassDefinition;
+import static org.onosproject.yangutils.utils.UtilConstants.AUGMENTED;
 import static org.onosproject.yangutils.utils.UtilConstants.CLOSE_CURLY_BRACKET;
 import static org.onosproject.yangutils.utils.UtilConstants.ERROR_MSG_FOR_GEN_CODE;
 import static org.onosproject.yangutils.utils.UtilConstants.LEAFREF;
@@ -95,6 +96,7 @@
 import static org.onosproject.yangutils.utils.UtilConstants.PACKAGE;
 import static org.onosproject.yangutils.utils.UtilConstants.PERIOD;
 import static org.onosproject.yangutils.utils.UtilConstants.SEMI_COLON;
+import static org.onosproject.yangutils.utils.UtilConstants.SET_METHOD_PREFIX;
 import static org.onosproject.yangutils.utils.UtilConstants.SLASH;
 import static org.onosproject.yangutils.utils.UtilConstants.SPACE;
 import static org.onosproject.yangutils.utils.io.impl.JavaDocGen.JavaDocType.BUILDER_CLASS;
@@ -620,7 +622,7 @@
         List<YangAugment> augments = getListOfAugments(parent);
         for (YangAugment augment : augments) {
             nodeId = augment.getTargetNode().get(0).getNodeIdentifier();
-
+            augment.setSetterMethodName(getAugmentSetterName(augment));
             if (validateNodeIdentifierInSet(nodeId, targets)) {
                 targets.add(augment.getTargetNode().get(0));
             }
@@ -641,6 +643,18 @@
         return augments;
     }
 
+    /* Returns setter method name for augment.*/
+    private static String getAugmentSetterName(YangAugment augment) {
+        YangAtomicPath atomicPath = augment.getTargetNode().get(0);
+        YangNode augmentedNode = atomicPath.getResolvedNode();
+        String setterName = SET_METHOD_PREFIX + AUGMENTED
+                + getCapitalCase(((JavaFileInfoContainer) augmentedNode.getParent())
+                                         .getJavaFileInfo().getJavaName())
+                + getCapitalCase(((JavaFileInfoContainer) augmentedNode)
+                                         .getJavaFileInfo().getJavaName());
+        return setterName;
+    }
+
     /*Validates the set for duplicate names of node identifiers.*/
     private static boolean validateNodeIdentifierInSet(
             YangNodeIdentifier nodeId, List<YangAtomicPath> targets) {