From bacc781ea755d3771e1c8d09e1fbfd873872f750 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 15 Nov 2015 15:31:09 +0000
Subject: Allow descendants of nodes to be traversed

---
 src/main/java/org/zwobble/couscous/ast/MethodNode.java | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

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

diff --git a/src/main/java/org/zwobble/couscous/ast/MethodNode.java b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
index d37f7d7..3e97a36 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
@@ -1,10 +1,9 @@
 package org.zwobble.couscous.ast;
 
-import java.util.List;
-
-import org.zwobble.couscous.ast.visitors.NodeVisitor;
-
 import com.google.common.collect.ImmutableList;
+import org.zwobble.couscous.ast.visitors.NodeMapper;
+
+import java.util.List;
 
 public class MethodNode implements CallableNode {
     public static Builder staticMethod(String name) {
@@ -115,10 +114,10 @@ public class MethodNode implements CallableNode {
     public List<StatementNode> getBody() {
         return body;
     }
-    
+
     @Override
-    public void accept(NodeVisitor visitor) {
-        visitor.visit(this);
+    public <T> T accept(NodeMapper<T> visitor) {
+        return visitor.visit(this);
     }
     
     @Override
-- 
cgit v1.2.3