【Groovy】编译时元编程 ( 编译时方法拦截 | 在 MyASTTransformation#visit 方法中找到要拦截的方法 )-阿里云开发者社区

【Groovy】编译时元编程 ( 编译时方法拦截 | 在 MyASTTransformation#visit 方法中找到要拦截的方法 )

2022-02-09 164

版权

本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。

简介： 【Groovy】编译时元编程 ( 编译时方法拦截 | 在 MyASTTransformation#visit 方法中找到要拦截的方法 )

一、在 MyASTTransformation#visit 方法中找到要拦截的方法

在 ASTTransformation 接口实现类的

void visit(ASTNode[] nodes, SourceUnit source)

方法中 , 其中 ASTNode[] nodes 参数是 AST 语法树根节点数组 , 每个数组元素都是一个 ModuleNode 对应一个 Groovy 脚本 ;

SourceUnit source 是源单元 , 可以通过该对象拿到源文件 ;

source.AST 是单个 ModuleNode 对象 , 对应一个 Groovy 脚本 ;

1、获取 ClassNode 节点集合

source.AST.classes 就是一个 Groovy 脚本中定义的类节点数组 ; 这是在 ModuleNode 中的 ClassNode 类节点封装在了 List classes = new LinkedList(); 成员中 ;

2、查找指定的 ClassNode 节点

使用

source.AST.classes.find {
            // 查找名称为 Student 的类
            // it 是 ClassNode 节点
            it.name == "Student"
        }

代码 , 可以查找到名称为 “Student” 的 ClassNode 节点 , 也就是 Student 类对应的节点 ;

集合的 find 方法原型如下 , 得到的是一个集合元素对象 ; 该方法返回的是集合中第一个与闭包条件匹配的集合元素 ;

/**
     * 查找与闭包条件匹配的第一个值。例子：
     * <pre class="groovyTestCase">def list = [1,2,3]
     * assert 2 == list.find { it {@code >} 1 }
     * </pre>
     *
     * @param self    a Collection
     * @param closure a closure condition
     * @return the first Object found, in the order of the collections iterator, or null if no element matches
     * @since 1.0
     */
    public static <T> T find(Collection<T> self, @ClosureParams(FirstParam.FirstGenericType.class) Closure closure) {
        BooleanClosureWrapper bcw = new BooleanClosureWrapper(closure);
        for (T value : self) {
            if (bcw.call(value)) {
                return value;
            }
        }
        return null;
    }

3、获取指定 ClassNode 节点下的 MethodNode 节点集合

再进一步 , 如果获取的 ClassNode 节点不为空 , 则获取该节点下的 MethodNode 节点集合 , 使用 ?.methods 代码获取 ,

source.AST.classes.find {
            // 查找名称为 Student 的类
            // it 是 ClassNode 节点
            it.name == "Student"
        }?.methods

ClassNode 的 getMethods 方法原型如下 :

public class ClassNode extends AnnotatedNode implements Opcodes {
    /**
     * @return 与此相关的方法 {@code ClassNode}
     */
    public List<MethodNode> getMethods() {
        if (redirect != null)
            return redirect.getMethods();
        lazyClassInit();
        return methodsList;
    }
}

4、查找指定的 MethodNode 节点

查找 List 集合中 , 名称为 “hello” 的节点 , 也就是查找 Student 类中的 hello 方法对应的 MethodNode 节点 ;

source.AST.classes.find {
            // 查找名称为 Student 的类
            // it 是 ClassNode 节点
            it.name == "Student"
        }?.methods?.find {
            // 查找 Student 类下名称为 hello 的方法
            // it 是 MethodNode 节点
            it.name == "hello"
        }