commit 5652766058d572554d5850f1c051e813e0e986b2
author Gaurav Agrawal <gaurav.agrawal@huawei.com> Wed Apr 20 15:49:17 2016 +0530
committer Thomas Vachuska <tom@onlab.us> Tue Apr 26 22:38:16 2016 +0000
tree 0817a3596ceabd9f48ca0753551fa87dd8b493e8
parent 427d29374a38a03e551a61bc7fd09e970382c347

[ONOS-4302] RPC translator implementation

Change-Id: I457f12a2b0edaadee5ff888e0297b40854d53096

utils/yangutils/src/main/java/org/onosproject/yangutils/translator/tojava/utils/ClassDefinitionGenerator.java

diff --git a/utils/yangutils/src/main/java/org/onosproject/yangutils/translator/tojava/utils/ClassDefinitionGenerator.java b/utils/yangutils/src/main/java/org/onosproject/yangutils/translator/tojava/utils/ClassDefinitionGenerator.java
index 0db41c9..ec58b26 100644
--- a/utils/yangutils/src/main/java/org/onosproject/yangutils/translator/tojava/utils/ClassDefinitionGenerator.java
+++ b/utils/yangutils/src/main/java/org/onosproject/yangutils/translator/tojava/utils/ClassDefinitionGenerator.java
@@ -19,6 +19,7 @@
 import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.BUILDER_CLASS_MASK;
 import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.BUILDER_INTERFACE_MASK;
 import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_ENUM_CLASS;
+import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_RPC_INTERFACE;
 import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_TYPEDEF_CLASS;
 import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.GENERATE_UNION_CLASS;
 import static org.onosproject.yangutils.translator.tojava.GeneratedJavaFileType.IMPL_CLASS_MASK;
@@ -63,7 +64,7 @@
     public static String generateClassDefinition(int genFileTypes, String yangName) {
 
         /**
-         * based on the file type and the YANG name of the file, generate the
+         * Based on the file type and the YANG name of the file, generate the
          * class / interface definition start.
          */
         if ((genFileTypes & INTERFACE_MASK) != 0) {
@@ -80,6 +81,8 @@
             return getTypeClassDefinition(yangName);
         } else if ((genFileTypes & GENERATE_ENUM_CLASS) != 0) {
             return getEnumClassDefinition(yangName);
+        } else if ((genFileTypes & GENERATE_RPC_INTERFACE) != 0) {
+            return getRpcInterfaceDefinition(yangName);
         }
         return null;
     }
@@ -88,7 +91,7 @@
      * Returns enum file class definition.
      *
      * @param yangName class name
-     * @return enum file class definiton
+     * @return enum file class definition
      */
     private static String getEnumClassDefinition(String yangName) {
         return PUBLIC + SPACE + ENUM + SPACE + yangName + SPACE + OPEN_CURLY_BRACKET + NEW_LINE;
@@ -155,4 +158,14 @@
     private static String getTypeClassDefinition(String yangName) {
         return PUBLIC + SPACE + FINAL + SPACE + CLASS + SPACE + yangName + SPACE + OPEN_CURLY_BRACKET + NEW_LINE;
     }
+
+    /**
+     * Returns rpc file interface definition.
+     *
+     * @param yangName file name
+     * @return definition
+     */
+    private static String getRpcInterfaceDefinition(String yangName) {
+        return INTERFACE + SPACE + yangName + SPACE + OPEN_CURLY_BRACKET + NEW_LINE;
+    }
 }