diff options
Diffstat (limited to 'javaparser-core/src/main/java/com/github/javaparser/ast/stmt/ForeachStmt.java')
-rw-r--r-- | javaparser-core/src/main/java/com/github/javaparser/ast/stmt/ForeachStmt.java | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/javaparser-core/src/main/java/com/github/javaparser/ast/stmt/ForeachStmt.java b/javaparser-core/src/main/java/com/github/javaparser/ast/stmt/ForeachStmt.java index 5656bc505..3467466c1 100644 --- a/javaparser-core/src/main/java/com/github/javaparser/ast/stmt/ForeachStmt.java +++ b/javaparser-core/src/main/java/com/github/javaparser/ast/stmt/ForeachStmt.java @@ -165,6 +165,18 @@ public final class ForeachStmt extends Statement implements NodeWithBody<Foreach public boolean replace(Node node, Node replacementNode) { if (node == null) return false; + if (node == body) { + setBody((Statement) replacementNode); + return true; + } + if (node == iterable) { + setIterable((Expression) replacementNode); + return true; + } + if (node == variable) { + setVariable((VariableDeclarationExpr) replacementNode); + return true; + } return super.replace(node, replacementNode); } } |