aboutsummaryrefslogtreecommitdiff
path: root/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java')
-rw-r--r--src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java7
1 files changed, 7 insertions, 0 deletions
diff --git a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
index 6352f4a..a98728b 100644
--- a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
+++ b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
@@ -142,6 +142,13 @@ public class NodeStructure {
}
@Override
+ public Stream<? extends Node> visit(TryNode tryStatement) {
+ return Stream.concat(
+ tryStatement.getBody().stream(),
+ tryStatement.getExceptionHandlers().stream().flatMap(handler -> handler.getBody().stream()));
+ }
+
+ @Override
public Stream<? extends Node> visit(FormalTypeParameterNode parameter) {
return Stream.empty();
}