diff options
| -rw-r--r-- | src/delombok/lombok/delombok/PrettyPrinter.java | 19 | ||||
| -rw-r--r-- | test/pretty/resource/after/tryWithResourcesVarRef.java | 9 | ||||
| -rw-r--r-- | test/pretty/resource/before/TryWithResourcesVarRef.java | 10 |
3 files changed, 33 insertions, 5 deletions
diff --git a/src/delombok/lombok/delombok/PrettyPrinter.java b/src/delombok/lombok/delombok/PrettyPrinter.java index 84c342f0..353bd753 100644 --- a/src/delombok/lombok/delombok/PrettyPrinter.java +++ b/src/delombok/lombok/delombok/PrettyPrinter.java @@ -1268,9 +1268,14 @@ public class PrettyPrinter extends JCTree.Visitor { break; case 1: print("("); - JCVariableDecl decl = (JCVariableDecl) resources.get(0); - flagMod = -1L & ~FINAL; - printVarDefInline(decl); + JCTree resource = (JCTree) resources.get(0); + if (resource instanceof JCVariableDecl) { + JCVariableDecl decl = (JCVariableDecl) resource; + flagMod = -1L & ~FINAL; + printVarDefInline(decl); + } else { + print(resource); + } print(") "); break; default: @@ -1279,8 +1284,12 @@ public class PrettyPrinter extends JCTree.Visitor { int c = 0; for (Object i : resources) { align(); - flagMod = -1L & ~FINAL; - printVarDefInline((JCVariableDecl) i); + if (i instanceof JCVariableDecl) { + flagMod = -1L & ~FINAL; + printVarDefInline((JCVariableDecl) i); + } else { + print((JCTree) i); + } if (++c == len) { print(") "); } else { diff --git a/test/pretty/resource/after/tryWithResourcesVarRef.java b/test/pretty/resource/after/tryWithResourcesVarRef.java new file mode 100644 index 00000000..5117f706 --- /dev/null +++ b/test/pretty/resource/after/tryWithResourcesVarRef.java @@ -0,0 +1,9 @@ +import java.io.PrintWriter; +public class TryWithResourcesVarRef { + { + PrintWriter pw = new PrintWriter(System.out); + try (pw) { + pw.println(); + } + } +} diff --git a/test/pretty/resource/before/TryWithResourcesVarRef.java b/test/pretty/resource/before/TryWithResourcesVarRef.java new file mode 100644 index 00000000..6c039eb7 --- /dev/null +++ b/test/pretty/resource/before/TryWithResourcesVarRef.java @@ -0,0 +1,10 @@ +//version 9: +import java.io.PrintWriter; +public class TryWithResourcesVarRef { + { + PrintWriter pw = new PrintWriter(System.out); + try (pw) { + pw.println(); + } + } +} |