X-Git-Url: https://git.opendaylight.org/gerrit/gitweb?p=controller.git;a=blobdiff_plain;f=opendaylight%2Fconfig%2Fyang-jmx-generator-plugin%2Fsrc%2Fmain%2Fjava%2Forg%2Fopendaylight%2Fcontroller%2Fconfig%2Fyangjmxgenerator%2Fplugin%2Fjava%2FJavaFileInputBuilder.java;h=c498c5bf9fe9b702f6f7f0f835d9efde964a111d;hp=c739f57730c3920caa641881d64b3dd652527b08;hb=135a5848aa950dda6998203876b1daeb89a0a1c5;hpb=b23703bef6c3aaafe2dc83608a03b738ad42f945

diff --git a/opendaylight/config/yang-jmx-generator-plugin/src/main/java/org/opendaylight/controller/config/yangjmxgenerator/plugin/java/JavaFileInputBuilder.java b/opendaylight/config/yang-jmx-generator-plugin/src/main/java/org/opendaylight/controller/config/yangjmxgenerator/plugin/java/JavaFileInputBuilder.java
index c739f57730..c498c5bf9f 100644
--- a/opendaylight/config/yang-jmx-generator-plugin/src/main/java/org/opendaylight/controller/config/yangjmxgenerator/plugin/java/JavaFileInputBuilder.java
+++ b/opendaylight/config/yang-jmx-generator-plugin/src/main/java/org/opendaylight/controller/config/yangjmxgenerator/plugin/java/JavaFileInputBuilder.java
@@ -8,14 +8,13 @@
 
 package org.opendaylight.controller.config.yangjmxgenerator.plugin.java;
 
-import com.google.common.base.Optional;
-import org.opendaylight.controller.config.yangjmxgenerator.plugin.ftl.model.Annotation;
+import static com.google.common.base.Preconditions.checkNotNull;
 
+import com.google.common.base.Optional;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
-
-import static com.google.common.base.Preconditions.checkNotNull;
+import org.opendaylight.controller.config.yangjmxgenerator.plugin.ftl.model.Annotation;
 
 public class JavaFileInputBuilder {
 
@@ -27,6 +26,8 @@ public class JavaFileInputBuilder {
 
     private final List<String> classAnnotations = new ArrayList<>();
 
+    private final List<FullyQualifiedName> importFQNs = new ArrayList<>();
+
     private final List<FullyQualifiedName> extendsFQNs = new ArrayList<>();
 
     private final List<FullyQualifiedName> implementsFQNs = new ArrayList<>();
@@ -45,6 +46,10 @@ public class JavaFileInputBuilder {
         classAnnotations.add(checkNotNull(annotation));
     }
 
+    public void addImportFQN(FullyQualifiedName fqn) {
+        importFQNs.add(fqn);
+    }
+
     public void addExtendsFQN(FullyQualifiedName fqn) {
         extendsFQNs.add(fqn);
     }
@@ -130,6 +135,11 @@ public class JavaFileInputBuilder {
                 return header;
             }
 
+            @Override
+            public List<FullyQualifiedName> getImports() {
+                return Collections.unmodifiableList(importFQNs);
+            }
+
             @Override
             public Optional<String> getClassJavaDoc() {
                 return classJavaDoc;
@@ -159,6 +169,7 @@ public class JavaFileInputBuilder {
             public List<String> getBodyElements() {
                 return Collections.unmodifiableList(bodyElements);
             }
+
         };
     }
 }