X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?a=blobdiff_plain;f=opendaylight%2Fsal%2Fyang-prototype%2Fcode-generator%2Fbinding-java-api-generator%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fsal%2Fjava%2Fapi%2Fgenerator%2FEnumGenerator.java;h=f2acefd169fcc16588f8fb65fac9cd6d80e00ddd;hb=4f104c6b522822ad1bea0ab5799eca0fa6a98d99;hp=760365d171ae3fcf017537978c2f915269cd50a6;hpb=a881650131c718f5189d975e95f44b3f4450903a;p=controller.git

diff --git a/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/EnumGenerator.java b/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/EnumGenerator.java
index 760365d171..f2acefd169 100644
--- a/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/EnumGenerator.java
+++ b/opendaylight/sal/yang-prototype/code-generator/binding-java-api-generator/src/main/java/org/opendaylight/controller/sal/java/api/generator/EnumGenerator.java
@@ -7,8 +7,6 @@
  */
 package org.opendaylight.controller.sal.java.api.generator;
 
-import static org.opendaylight.controller.sal.java.api.generator.Constants.NL;
-
 import java.io.IOException;
 import java.io.StringWriter;
 import java.io.Writer;
@@ -22,25 +20,11 @@ public class EnumGenerator implements CodeGenerator {
     @Override
     public Writer generate(Type type) throws IOException {
         final Writer writer = new StringWriter();
-
         if (type instanceof Enumeration) {
-            Enumeration enums = (Enumeration) type;
-            writer.write(GeneratorUtil.createPackageDeclaration(enums.getPackageName()));
-            writer.write(NL + NL);
-            writer.write(GeneratorUtil.createEnum(enums, ""));
+            final Enumeration enums = (Enumeration) type;
+            final EnumTemplate enumTemplate = new EnumTemplate(enums);
+            writer.write(enumTemplate.generate().toString());
         }
-
-        return writer;
-    }
-
-    public Writer generateInnerEnumeration(Type type, String indent) throws IOException {
-        final Writer writer = new StringWriter();
-
-        if (type instanceof Enumeration) {
-            Enumeration enums = (Enumeration) type;
-            writer.write(GeneratorUtil.createEnum(enums, indent));
-        }
-
         return writer;
     }