From ecfac0740033b048b55001332eb11131b64ddcaf Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 17 Apr 2016 15:33:54 +0100
Subject: Add try nodes

---
 .../java/org/zwobble/couscous/ast/structure/NodeStructure.java     | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java')

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
@@ -141,6 +141,13 @@ public class NodeStructure {
                     whileLoop.getBody().stream());
             }
 
+            @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();
-- 
cgit v1.2.3