Only classes with suffix 'Entity' / performance / fail-safe

Author: pun-ky

README.md

@@ -16,4 +16,23 @@ Plugin for enhancing JPA entity classes at compile time to be ready for OpenJPA.
 
 ## License
 
-Apache License 2.0
+Apache License 2.0
+
+## Usage
+
+By default, plugin enhances all classes that have suffix `Entity`, e.g `UserEntity`, `PostEntity` etc.
+But that behavior can be customized to any needs, see below example.
+
+```
+
+apply plugin: 'neva.openjpa'
+
+enhanceEntities {
+
+    includeFilter = {
+        include '**/*Entity.class'
+    }
+
+}
+
+```

build.gradle

@@ -9,14 +9,14 @@ buildscript {
     }
 }
 
-group 'com.neva.gradle.plugin'
-version '1.0.0'
-defaultTasks = ['clean', 'install']
-
 apply plugin: 'groovy'
-apply plugin: 'maven'
+apply plugin: 'maven-publish'
 apply plugin: 'com.gradle.plugin-publish'
 
+group = 'com.neva.gradle.plugin'
+version = '1.0.1'
+defaultTasks = ['clean', 'install']
+
 repositories {
     jcenter()
     mavenCentral()
@@ -26,7 +26,7 @@ dependencies {
     compile gradleApi()
     compile localGroovy()
 
-    compile 'org.apache.openjpa:openjpa:2.3.0'
+    compile 'org.apache.openjpa:openjpa:2.4.1'
 }
 
 jar {
@@ -38,6 +38,14 @@ jar {
     }
 }
 
+publishing {
+    publications {
+        mavenJava(MavenPublication) {
+            from components.java
+        }
+    }
+}
+
 pluginBundle {
     website = 'http://javarel.neva.zone/'
     vcsUrl = 'https://github.com/neva-dev/gradle-openjpa-plugin'

src/main/groovy/com/neva/gradle/plugin/openjpa/task/EnhanceTask.groovy

@@ -8,12 +8,15 @@ class EnhanceTask extends AbstractTask {
 
     static final NAME = "enhanceEntities"
 
+    def includeFilter = {
+        include '**/*Entity.class'
+    }
+
     EnhanceTask() {
         description = "Enhance JPA entity classes using OpenJPA Enhancer"
 
         configure {
-            outputs.upToDateWhen { false }
-
+            inputs.dir(inputDir)
             dependsOn([project.compileJava, project.processResources])
             project.classes.dependsOn this
         }
@@ -22,9 +25,7 @@ class EnhanceTask extends AbstractTask {
     @TaskAction
     def run() {
         // define the entity classes
-        def entityFiles = project.fileTree(project.sourceSets.main.output.classesDir).matching {
-            include '**/*.class'
-        }
+        def entityFiles = project.fileTree(inputDir).matching(this.includeFilter)
 
         println "Enhancing with OpenJPA, the following files..."
         entityFiles.getFiles().each {
@@ -51,4 +52,8 @@ class EnhanceTask extends AbstractTask {
         }
     }
 
+    def getInputDir() {
+        return project.sourceSets.main.output.classesDir
+    }
+
 }