Skip to content

Commit fc4a1a0

Browse files
committed
Prepare LombokValToFinalVar for Lombok type attribution
1 parent 41a76c8 commit fc4a1a0

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

src/main/java/org/openrewrite/java/migrate/lombok/LombokValToFinalVar.java

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,8 @@ public J.CompilationUnit visitCompilationUnit(J.CompilationUnit compilationUnit,
8080
@Override
8181
public J.VariableDeclarations visitVariableDeclarations(J.VariableDeclarations mv, ExecutionContext ctx) {
8282
J.VariableDeclarations varDecls = super.visitVariableDeclarations(mv, ctx);
83-
if (TypeUtils.isOfClassType(varDecls.getType(), LOMBOK_VAL)) {
83+
if (TypeUtils.isOfClassType(varDecls.getType(), LOMBOK_VAL) ||
84+
(varDecls.getTypeExpression() instanceof J.Identifier && ((J.Identifier) varDecls.getTypeExpression()).getSimpleName().equals("val"))) {
8485
maybeRemoveImport(LOMBOK_VAL);
8586

8687
J.VariableDeclarations.NamedVariable nv = mv.getVariables().get(0);

src/test/java/org/openrewrite/java/migrate/lombok/LombokValToFinalVarTest.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,6 +64,7 @@ void bar() {
6464
void replaceAssignmentVar() {
6565
//language=java
6666
rewriteRun(
67+
spec -> spec.typeValidationOptions(TypeValidation.builder().identifiers(false).build()),
6768
version(
6869
java(
6970
"""

0 commit comments

Comments
 (0)