Add a UsesType precondition to ReplaceConstant

amishra-u · Dec 28, 2024 · ef9c7d5 · ef9c7d5
1 parent 85db15f
commit ef9c7d5
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/rewrite-java/src/main/java/org/openrewrite/java/ReplaceConstant.java b/rewrite-java/src/main/java/org/openrewrite/java/ReplaceConstant.java
@@ -19,6 +19,7 @@
 import lombok.Value;
 import org.jspecify.annotations.Nullable;
 import org.openrewrite.*;
+import org.openrewrite.java.search.UsesType;
 import org.openrewrite.java.tree.J;
 import org.openrewrite.java.tree.JavaType;
 import org.openrewrite.java.tree.TypeUtils;
@@ -53,7 +54,7 @@ public String getDescription() {
 
     @Override
     public TreeVisitor<?, ExecutionContext> getVisitor() {
-        return new JavaVisitor<ExecutionContext>() {
+        JavaVisitor<ExecutionContext> replacementVisitor = new JavaVisitor<ExecutionContext>() {
             J.@Nullable Literal literal;
 
             @Override
@@ -117,5 +118,9 @@ private J.Literal buildLiteral() {
                 return literal;
             }
         };
+        return Preconditions.check(
+                new UsesType<>(owningType, true),
+                replacementVisitor
+        );
     }
 }