Improve registerMainClassTask implementation (#12632)

Register a new findMainClassTask or extend if the one already exists from Spring Boot
grails · Jul 25, 2022 · 5b42019 · 5b42019
1 parent 8cb293c
commit 5b42019
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/grails-gradle-plugin/src/main/groovy/org/grails/gradle/plugin/core/GrailsGradlePlugin.groovy b/grails-gradle-plugin/src/main/groovy/org/grails/gradle/plugin/core/GrailsGradlePlugin.groovy
@@ -496,13 +496,17 @@ class GrailsGradlePlugin extends GroovyPlugin {
     @CompileStatic
     protected void registerFindMainClassTask(Project project) {
         TaskContainer taskContainer = project.tasks
-        if (taskContainer.findByName("findMainClass") == null) {
-            def findMainClassTask = project.tasks.create(name: "findMainClass", type: FindMainClassTask, overwrite: true)
+        def findMainClassTask = taskContainer.findByName("findMainClass")
+        if (findMainClassTask == null) {
+            findMainClassTask = project.tasks.create(name: "findMainClass", type: FindMainClassTask, overwrite: true)
             findMainClassTask.mustRunAfter project.tasks.withType(GroovyCompile)
             def bootRepackageTask = project.tasks.findByName("bootRepackage")
             if (bootRepackageTask) {
                 bootRepackageTask.dependsOn findMainClassTask
             }
+        } else if (!FindMainClassTask.class.isAssignableFrom(findMainClassTask.class)) {
+            def grailsFindMainClass = project.tasks.create(name: "grailsFindMainClass", type: FindMainClassTask)
+            findMainClassTask.dependsOn(grailsFindMainClass)
         }
     }