From 4068d39e430d99b0a4c6469b6ec0c1e909509dea Mon Sep 17 00:00:00 2001
From: Clayton Walker <cwalker@sofi.org>
Date: Mon, 12 Dec 2022 19:22:41 -0700
Subject: [PATCH] Fix configuration cache

---
 .../GoogleJavaFormatPlugin.groovy                      | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/main/groovy/com/github/sherter/googlejavaformatgradleplugin/GoogleJavaFormatPlugin.groovy b/src/main/groovy/com/github/sherter/googlejavaformatgradleplugin/GoogleJavaFormatPlugin.groovy
index 8c88523..3dd1e68 100644
--- a/src/main/groovy/com/github/sherter/googlejavaformatgradleplugin/GoogleJavaFormatPlugin.groovy
+++ b/src/main/groovy/com/github/sherter/googlejavaformatgradleplugin/GoogleJavaFormatPlugin.groovy
@@ -38,10 +38,16 @@ class GoogleJavaFormatPlugin implements Plugin<Project> {
 
         SharedContext context = new SharedContext(project, extension)
         TaskConfigurator configurator = new TaskConfigurator(context)
-        project.gradle.taskGraph.beforeTask { Task task ->
-            if (task.project == this.project && task instanceof FormatTask) {
+        if (SUPPORTS_LAZY_TASKS) {
+            project.tasks.withType(FormatTask).configureEach { FormatTask task ->
                 task.accept(configurator)
             }
+        } else {
+            project.gradle.taskGraph.beforeTask { Task task ->
+                if (task.project == this.project && task instanceof FormatTask) {
+                    task.accept(configurator)
+                }
+            }
         }
     }