[groovy] branch master updated: Tweak the general visiting methods

[sunlan]

This is an automated email from the ASF dual-hosted git repository.

sunlan pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/groovy.git


The following commit(s) were added to refs/heads/master by this push:
     new ac96f6b  Tweak the general visiting methods
ac96f6b is described below

commit ac96f6b574129fe7ab13cbebd23d7a1931409d50
Author: Daniel Sun <sun...@apache.org>
AuthorDate: Fri Dec 4 00:20:20 2020 +0800

    Tweak the general visiting methods
---
 src/main/java/org/codehaus/groovy/ast/GroovyCodeVisitor.java | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/main/java/org/codehaus/groovy/ast/GroovyCodeVisitor.java 
b/src/main/java/org/codehaus/groovy/ast/GroovyCodeVisitor.java
index ba705d3..8419bd1 100644
--- a/src/main/java/org/codehaus/groovy/ast/GroovyCodeVisitor.java
+++ b/src/main/java/org/codehaus/groovy/ast/GroovyCodeVisitor.java
@@ -68,6 +68,7 @@ import org.codehaus.groovy.ast.stmt.ExpressionStatement;
 import org.codehaus.groovy.ast.stmt.ForStatement;
 import org.codehaus.groovy.ast.stmt.IfStatement;
 import org.codehaus.groovy.ast.stmt.ReturnStatement;
+import org.codehaus.groovy.ast.stmt.Statement;
 import org.codehaus.groovy.ast.stmt.SwitchStatement;
 import org.codehaus.groovy.ast.stmt.SynchronizedStatement;
 import org.codehaus.groovy.ast.stmt.ThrowStatement;
@@ -202,8 +203,13 @@ public interface GroovyCodeVisitor {
         if (list != null) list.forEach(expr -> expr.visit(this));
     }
 
-    default void visit(ASTNode astNode) {
-        if (null == astNode) return;
-        astNode.visit(this);
+    default void visit(Statement statement) {
+        if (null == statement) return;
+        statement.visit(this);
+    }
+
+    default void visit(Expression expression) {
+        if (null == expression) return;
+        expression.visit(this);
     }
 }