Fix index out of bounds

openrewrite · Jan 21, 2025 · f36a1d9 · f36a1d9
1 parent d49141c
commit f36a1d9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/rewrite-java/src/main/java/org/openrewrite/java/tree/J.java b/rewrite-java/src/main/java/org/openrewrite/java/tree/J.java
@@ -5122,7 +5122,14 @@ final class SwitchExpression implements J, Expression, TypedTree {
                 public J visitBlock(Block block, AtomicReference<JavaType> javaType) {
                     if (!block.getStatements().isEmpty()) {
                         Case caze = (Case) block.getStatements().get(0);
-                        javaType.set(caze.getExpressions().get(0).getType());
+                        for (J j : caze.getCaseLabels()) {
+                            if (j instanceof TypedTree) {
+                                if (((TypedTree) j).getType() != null) {
+                                    javaType.set(((TypedTree) j).getType());
+                                    break;
+                                }
+                            }
+                        }
                     }
                     return block;
                 }