MigrateRestTemplateBuilderTimeoutByInt: Don't rely on cycles

See: openrewrite/rewrite#3921
openrewrite · Jan 18, 2024 · b76e034 · b76e034
1 parent 2ce6e11
commit b76e034
Showing 1 changed file with 12 additions and 14 deletions.
diff --git a/src/main/java/org/openrewrite/java/spring/boot2/MigrateRestTemplateBuilderTimeoutByInt.java b/src/main/java/org/openrewrite/java/spring/boot2/MigrateRestTemplateBuilderTimeoutByInt.java
@@ -15,10 +15,7 @@
  */
 package org.openrewrite.java.spring.boot2;
 
-import org.openrewrite.ExecutionContext;
-import org.openrewrite.Preconditions;
-import org.openrewrite.Recipe;
-import org.openrewrite.TreeVisitor;
+import org.openrewrite.*;
 import org.openrewrite.java.JavaIsoVisitor;
 import org.openrewrite.java.JavaParser;
 import org.openrewrite.java.JavaTemplate;
@@ -48,17 +45,18 @@ public TreeVisitor<?, ExecutionContext> getVisitor() {
                     @Override
                     public J.MethodInvocation visitMethodInvocation(J.MethodInvocation method, ExecutionContext ctx) {
                         J.MethodInvocation m = super.visitMethodInvocation(method, ctx);
-                        if (connectionTimeout.matches(method) || readTimeout.matches(method)) {
+                        updateCursor(m);
+                        if (connectionTimeout.matches(m) || readTimeout.matches(m)) {
                             m = JavaTemplate
-                                .builder("Duration.ofMillis(#{any(int)})")
-                                .imports("java.time.Duration")
-                                .javaParser(JavaParser.fromJavaVersion()
-                                    .classpathFromResources(ctx, "spring-boot-2.*"))
-                                .build()
-                                .apply(
-                                    getCursor(),
-                                    m.getCoordinates().replaceArguments(),
-                                    m.getArguments().get(0));
+                                    .builder("Duration.ofMillis(#{any(int)})")
+                                    .imports("java.time.Duration")
+                                    .javaParser(JavaParser.fromJavaVersion()
+                                            .classpathFromResources(ctx, "spring-boot-2.*"))
+                                    .build()
+                                    .apply(
+                                            getCursor(),
+                                            m.getCoordinates().replaceArguments(),
+                                            m.getArguments().get(0));
                             maybeAddImport("java.time.Duration");
                         }
                         return m;