Prevent class cast exception on returnedClass()

openrewrite · Jan 1, 2025 · c71d9f7 · c71d9f7
1 parent 1a6f87d
commit c71d9f7
Show file tree

Hide file tree

Showing 2 changed files with 328 additions and 150 deletions.
diff --git a/src/main/java/org/openrewrite/hibernate/MigrateUserType.java b/src/main/java/org/openrewrite/hibernate/MigrateUserType.java
@@ -97,7 +97,11 @@ public J.MethodDeclaration visitMethodDeclaration(J.MethodDeclaration method, At
 
                     @Override
                     public J.Return visitReturn(J.Return _return, AtomicReference<J.FieldAccess> ref) {
-                        ref.set((J.FieldAccess) _return.getExpression());
+                        Expression expression = _return.getExpression();
+                        if (expression instanceof J.FieldAccess &&
+                                "class".equals(((J.FieldAccess) expression).getSimpleName())) {
+                            ref.set((J.FieldAccess) expression);
+                        }
                         return _return;
                     }
                 }.visitNonNull(cd, reference);