文章目錄

• 一、利用注解進行 AST 語法樹轉換
• • 1、定義注解并使用 GroovyASTTransformationClass 注明 AST 轉換接口
  • 2、AST 轉換接口實現
  • 3、定義 Groovy 類并使用 @Compile 注解修飾需要攔截的方法
  • 4、執行結果

一、利用注解進行 AST 語法樹轉換

---

1、定義注解并使用 GroovyASTTransformationClass 注明 AST 轉換接口

首先 , 定義 Compile 注解 , 該注解名稱是任意字符串 , @Target(ElementType.METHOD) 表示該注解作用于方法上 , @GroovyASTTransformationClass("MyASTTransformation") 表示該注解修飾的節點對應的 AST 轉換接口實現類是 MyASTTransformation ;

import org.codehaus.groovy.transform.GroovyASTTransformationClassimport java.lang.annotation.ElementType import java.lang.annotation.Target/*** 該注解作用于方法上*/ @Target(ElementType.METHOD) @GroovyASTTransformationClass("MyASTTransformation") @interface Compile { }

2、AST 轉換接口實現

然后 , 實現 MyASTTransformation 類 , 該類繼承 ASTTransformation 接口 ;

import org.codehaus.groovy.ast.ASTNode import org.codehaus.groovy.ast.builder.AstBuilder import org.codehaus.groovy.ast.stmt.BlockStatement 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 源單元 , 可以通過該對象拿到源文件*/@Overridevoid visit(ASTNode[] nodes, SourceUnit source) {// 獲取 Groovy.groovy 腳本中的類集合 , 并進行遍歷// 在 ModuleNode 中的類節點封裝在了如下成員中// List<ClassNode> classes = new LinkedList<ClassNode>();source.AST.classes.find {// 查找名稱為 Student 的類// it 是 ClassNode 節點it.name == "Student"}?.methods?.find {// 查找 Student 類下名稱為 hello 的方法// it 是 MethodNode 節點it.name == "hello"}?.with {// 找到了 Student 下的 hello 方法// 在 MethodNode 節點下調用// it 就是 MethodNode 節點BlockStatement blockStatement = code// 清空 BlockStatement 中的 List<Statement> statements 成員// 方法攔截清空 , 就不再執行原本的方法// 方法注入不清空 , 會執行原來的方法內容//blockStatement.statements.clear()// 創建方法節點def methods = new AstBuilder().buildFromSpec {expression {methodCall {variable('this')constant('println')argumentList {constant('hello buildFromSpec')}}}}// 將方法節點添加到 hello 方法中//blockStatement.statements.addAll(methods)// 創建方法節點def methods2 = new AstBuilder().buildFromString('println "hello buildFromString"')// 將方法節點添加到 hello 方法中//blockStatement.statements.addAll(methods2)// 創建方法節點, 注意此處拿到的是def methods3 = new AstBuilder().buildFromCode {println "hello buildFromCode"}// 將方法節點添加到 hello 方法中blockStatement.statements.addAll(methods3[0].statements)}} }

3、定義 Groovy 類并使用 @Compile 注解修飾需要攔截的方法

最后 , 實現 Groovy 類 , 在該類的方法上使用 @Compile 注解 ;

class Student{def name@Compiledef hello(){println "hello"} }def student = new Student() student.hello()

4、執行結果

執行上述 Groovy 腳本 , 執行結果為 :

hello hello buildFromCode

總結

以上是生活随笔為你收集整理的【Groovy】编译时元编程 ( 利用注解进行 AST 语法树转换 | 定义注解并使用 GroovyASTTransformationClass 注明 AST 转换接口 | AST 转换接口实现 )的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。