Reduce failures through getParentTreeCursor()

openrewrite · Oct 14, 2024 · d284269 · d284269
1 parent e009626
commit d284269
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/main/java/org/openrewrite/java/migrate/javax/AddColumnAnnotation.java b/src/main/java/org/openrewrite/java/migrate/javax/AddColumnAnnotation.java
@@ -91,7 +91,7 @@ public J.VariableDeclarations visitVariableDeclarations(J.VariableDeclarations m
                         // Update existing @Column annotation
                         J.VariableDeclarations updatedVariable = (J.VariableDeclarations) new AddOrUpdateAnnotationAttribute(
                                 "javax.persistence.Column", "name", "element", true)
-                                .getVisitor().visit(multiVariable, ctx, getCursor());
+                                .getVisitor().visit(multiVariable, ctx, getCursor().getParentTreeCursor());
                         return super.visitVariableDeclarations(updatedVariable, ctx);
                     }
                 }

diff --git a/src/main/java/org/openrewrite/java/migrate/javax/AddScopeToInjectedClass.java b/src/main/java/org/openrewrite/java/migrate/javax/AddScopeToInjectedClass.java
@@ -85,7 +85,7 @@ public J.CompilationUnit visitCompilationUnit(J.CompilationUnit compilationUnit,
                 for (J.ClassDeclaration aClass : cu.getClasses()) {
                     if (aClass.getType() != null && injectedTypes.contains(aClass.getType().getFullyQualifiedName())) {
                         return (J.CompilationUnit) new AnnotateTypesVisitor(JAVAX_ENTERPRISE_CONTEXT_DEPENDENT)
-                                .visit(cu, injectedTypes, getCursor().getParent());
+                                .visit(cu, injectedTypes, getCursor().getParentTreeCursor());
                     }
                 }
                 return cu;