Gen. Abstract*ModuleFactory with WORKING handleChangedClass()
[controller.git] / opendaylight / config / yang-jmx-generator-plugin / src / main / java / org / opendaylight / controller / config / yangjmxgenerator / plugin / java / JavaFileInputBuilder.java
index c739f57730c3920caa641881d64b3dd652527b08..c498c5bf9fe9b702f6f7f0f835d9efde964a111d 100644 (file)
@@ -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);
             }
+
         };
     }
 }