Groovy编译时元编程 ( ASTTransformation#visit 方法简介 | org.codehaus.groovy.ast.ModuleNode 脚本节点 )

Posted 2022-02-18 韩曙亮

文章目录

• 一、ASTTransformation#visit 方法简介
• 二、org.codehaus.groovy.ast.ModuleNode 脚本节点

一、ASTTransformation#visit 方法简介

---

在上一篇博客 【Groovy】编译时元编程 ( 编译 ASTTransformation | 打包 ASTTransformation 字节码文件 | 编译 Groovy 类同进行编译时处理 ) 中 , 写的进行 Groovy 编译时处理 ASTTransformation 如下 :

import org.codehaus.groovy.ast.ASTNode
import org.codehaus.groovy.control.SourceUnit
import org.codehaus.groovy.transform.ASTTransformation
import org.codehaus.groovy.transform.GroovyASTTransformation

@GroovyASTTransformation
class MyASTTransformation implements ASTTransformation 
    /**
     * 编译时处理方法
     * @param nodes AST 抽象语法树节点 , 是 ASTNode 数组类型
     * @param source 源单元 , 可以通过该对象拿到源文件
     */
    @Override
    void visit(ASTNode[] nodes, SourceUnit source) 
        println nodes
        println source
    

打印结果为 :

[org.codehaus.groovy.ast.ModuleNode@2609b277]
org.codehaus.groovy.control.SourceUnit@1fd14d74

在 visit 方法中 , ASTNode[] nodes 参数是 AST 语法树根节点 , SourceUnit source 参数代表源码 ;

二、org.codehaus.groovy.ast.ModuleNode 脚本节点

---

ASTNode[] nodes 参数是 AST 语法树节点 , 该数组的值是

[org.codehaus.groovy.ast.ModuleNode@2609b277]

AST 语法树节点是 org.codehaus.groovy.ast.ModuleNode 类型对象 ;

org.codehaus.groovy.ast.ModuleNode 是 Groovy 脚本的语法树节点 ;

• 其内部封装的 List<ClassNode> classes = new LinkedList<ClassNode>(); 成员表示该脚本中定义了多少个 Groovy 类节点 ;
• private List<MethodNode> methods = new ArrayList<MethodNode>(); 成员表示该脚本中有多少个方法 ;
• private Map<String, ImportNode> imports = new HashMap<String, ImportNode>(); 成员表示导入了多少类 ;

通过 ModuleNode 类 , 可以拿到 Groovy 脚本中几乎所有的内容 ;

org.codehaus.groovy.ast.ModuleNode 原型如下 :

/**
 * 表示一个模块，该模块通常由一个类声明组成，
 * 但可以包括一些导入、一些语句和多个类，
 * 这些类与Python或Ruby中的脚本等语句混合在一起
 *
 * @author Jochen Theodorou
 * @author Paul King
 * @author <a href="mailto:[e-mail]">James Strachan</a>
 */
public class ModuleNode extends ASTNode implements Opcodes 

    private BlockStatement statementBlock = new BlockStatement();
    List<ClassNode> classes = new LinkedList<ClassNode>();
    private List<MethodNode> methods = new ArrayList<MethodNode>();
    private Map<String, ImportNode> imports = new HashMap<String, ImportNode>();
    private List<ImportNode> starImports = new ArrayList<ImportNode>();
    private Map<String, ImportNode> staticImports = new LinkedHashMap<String, ImportNode>();
    private Map<String, ImportNode> staticStarImports = new LinkedHashMap<String, ImportNode>();
    private CompileUnit unit;
    private PackageNode packageNode;
    private String description;
    private boolean createClassForStatements = true;
    private transient SourceUnit context;
    private boolean importsResolved = false;
    private ClassNode scriptDummy;
    private String mainClassName = null;