From eee46d13e629c79f4acb60cc29aa773e2ed735c1 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 28 Feb 2016 13:54:49 +0000
Subject: Change method bodies to be optional

---
 src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(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 2424bbe..f1f37a6 100644
--- a/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
+++ b/src/main/java/org/zwobble/couscous/ast/structure/NodeStructure.java
@@ -3,6 +3,7 @@ package org.zwobble.couscous.ast.structure;
 import org.zwobble.couscous.ast.*;
 import org.zwobble.couscous.ast.visitors.NodeMapper;
 
+import java.util.Collection;
 import java.util.function.Predicate;
 import java.util.stream.Stream;
 
@@ -158,7 +159,7 @@ public class NodeStructure {
                 return concatStreams(
                     methodNode.getAnnotations().stream(),
                     methodNode.getArguments().stream(),
-                    methodNode.getBody().stream());
+                    methodNode.getBody().map(Collection::stream).orElse(Stream.empty()));
             }
 
             @Override
-- 
cgit v1.2.3