Merge pull request #43 from ethlo/issue-42

Fixes #42

Author: ethlo

pom.xml

@@ -3,7 +3,7 @@
     <modelVersion>4.0.0</modelVersion>
     <groupId>com.ethlo.persistence.tools</groupId>
     <artifactId>eclipselink-maven-plugin</artifactId>
-    <version>3.0.1</version>
+    <version>3.0.2</version>
     <packaging>maven-plugin</packaging>
     <name>Eclipselink Maven Plugin</name>
     <description>Maven Plugin for Eclipselink supporting static weaving, meta-model generation and DDL generation

src/main/java/com/ethlo/persistence/tools/eclipselink/EclipselinkModelGenMojo.java

@@ -193,7 +193,17 @@ public void execute() throws MojoExecutionException
 
     private Set<File> getSourceFiles()
     {
-        if (source == null || !source.exists())
+        final Set<File> fromSource = getFilesFromDirectory(source);
+        final Set<File> fromGenerated = getFilesFromDirectory(generatedSourcesDirectory);
+        final Set<File> all = new TreeSet<>();
+        all.addAll(fromSource);
+        all.addAll(fromGenerated);
+        return all;
+    }
+
+    private Set<File> getFilesFromDirectory(File dir)
+    {
+        if (dir == null || !dir.exists())
         {
             return new TreeSet<>();
         }
@@ -208,12 +218,12 @@ private Set<File> getSourceFiles()
             }
         }
 
-        Set<File> files = new HashSet<>();
-        final Scanner scanner = buildContext.newScanner(source);
+        final Set<File> files = new HashSet<>();
+        final Scanner scanner = buildContext.newScanner(dir);
         scanner.setIncludes(filters);
         scanner.scan();
 
-        String[] includedFiles = scanner.getIncludedFiles();
+        final String[] includedFiles = scanner.getIncludedFiles();
         if (includedFiles != null)
         {
             for (String includedFile : includedFiles)