multiFileYangTranslator
Change-Id: I2adfef3acaec4bd74ba2c487404d2c655b800988
diff --git a/src/main/java/org/onosproject/yangutils/translator/tojava/javamodel/YangJavaUnion.java b/src/main/java/org/onosproject/yangutils/translator/tojava/javamodel/YangJavaUnion.java
index 7af19c6..5c3a144 100644
--- a/src/main/java/org/onosproject/yangutils/translator/tojava/javamodel/YangJavaUnion.java
+++ b/src/main/java/org/onosproject/yangutils/translator/tojava/javamodel/YangJavaUnion.java
@@ -16,10 +16,10 @@
package org.onosproject.yangutils.translator.tojava.javamodel;
import java.io.IOException;
+
import org.onosproject.yangutils.datamodel.YangUnion;
import org.onosproject.yangutils.translator.tojava.JavaCodeGenerator;
import org.onosproject.yangutils.translator.tojava.JavaFileInfo;
-import org.onosproject.yangutils.translator.tojava.JavaImportData;
import org.onosproject.yangutils.translator.tojava.TempJavaCodeFragmentFiles;
import org.onosproject.yangutils.translator.tojava.utils.YangPluginConfig;
@@ -29,7 +29,9 @@
/**
* Represents union information extended to support java code generation.
*/
-public class YangJavaUnion extends YangUnion implements JavaCodeGeneratorInfo, JavaCodeGenerator {
+public class YangJavaUnion
+ extends YangUnion
+ implements JavaCodeGeneratorInfo, JavaCodeGenerator {
/**
* Contains the information of the java file being generated.
@@ -37,12 +39,6 @@
private JavaFileInfo javaFileInfo;
/**
- * Contains information of the imports to be inserted in the java file
- * generated.
- */
- private JavaImportData javaImportData;
-
- /**
* File handle to maintain temporary java code fragments as per the code
* snippet types.
*/
@@ -54,7 +50,6 @@
public YangJavaUnion() {
super();
setJavaFileInfo(new JavaFileInfo());
- setJavaImportData(new JavaImportData());
getJavaFileInfo().setGeneratedFileTypes(GENERATE_UNION_CLASS);
}
@@ -82,27 +77,6 @@
}
/**
- * Returns the data of java imports to be included in generated file.
- *
- * @return data of java imports to be included in generated file
- */
- @Override
- public JavaImportData getJavaImportData() {
- return javaImportData;
- }
-
- /**
- * Sets the data of java imports to be included in generated file.
- *
- * @param javaImportData data of java imports to be included in generated
- * file
- */
- @Override
- public void setJavaImportData(JavaImportData javaImportData) {
- this.javaImportData = javaImportData;
- }
-
- /**
* Returns the temporary file handle.
*
* @return temporary file handle
@@ -134,7 +108,8 @@
* @throws IOException IO operations fails
*/
@Override
- public void generateCodeEntry(YangPluginConfig yangPlugin) throws IOException {
+ public void generateCodeEntry(YangPluginConfig yangPlugin)
+ throws IOException {
generateCodeOfNode(this, yangPlugin);
}
@@ -144,7 +119,8 @@
* @throws IOException IO operations fails
*/
@Override
- public void generateCodeExit() throws IOException {
+ public void generateCodeExit()
+ throws IOException {
getTempJavaCodeFragmentFiles().generateJavaFile(GENERATE_UNION_CLASS, this);
}
}