文章目錄

• 一、在 MyASTTransformation#visit 方法中進(jìn)行方法攔截
• 二、完整代碼示例及進(jìn)行編譯時處理的編譯過程
• • 1、Groovy 腳本 Groovy.groovy
  • 2、ASTTransformation 接口實現(xiàn) MyASTTransformation.groovy
  • 3、配置 ASTTransformation
  • 3、使用命令行進(jìn)行編譯時處理

一、在 MyASTTransformation#visit 方法中進(jìn)行方法攔截

---

在上一篇博客 【Groovy】編譯時元編程 ( 編譯時方法攔截 | 在 MyASTTransformation#visit 方法中找到要攔截的方法 ) 中 , 在MyASTTransformation#visit 方法中 , 首先獲取了 ModuleNode , 然后查找指定的 ClassNode , 從指定的 ClassNode 中查找對應(yīng)的 MethodNode 節(jié)點 ;

方法攔截主要是在 MethodNode 節(jié)點中執(zhí)行 ; 該 MethodNode 節(jié)點對應(yīng) Student 類的 hello 方法 ;

根據(jù) 【Groovy】編譯時元編程 ( 方法攔截時用到的 AST 語法樹節(jié)點 MethodNode 節(jié)點 | MethodNode 節(jié)點分析 | MethodNode 節(jié)點中的BlockStatement) 博客分析 , MethodNode 節(jié)點中封裝了 private Statement code; 成員 , 該 code 成員的實際類型是 BlockStatement 類型 ;

public class BlockStatement extends Statement

這里首先將 MethodNode 節(jié)點的 code 成員賦值給 BlockStatement , 相當(dāng)于將 Statement 對象轉(zhuǎn)為 BlockStatement 類型 ;

BlockStatement blockStatement = code

在 BlockStatement 對象中 , 封裝了 private List<Statement> statements = new ArrayList<Statement>(); 成員 , 這里面就是該 MethodNode 節(jié)點中的方法內(nèi)容 , 將該集合清空 , 就相當(dāng)于將方法體清空了 , 實現(xiàn)了方法攔截操作 ;

二、完整代碼示例及進(jìn)行編譯時處理的編譯過程

---

1、Groovy 腳本 Groovy.groovy

class Student{def namedef hello(){println "hello"} }def student = new Student() student.hello()

2、ASTTransformation 接口實現(xiàn) MyASTTransformation.groovy

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 抽象語法樹節(jié)點 , 是 ASTNode 數(shù)組類型* @param source 源單元 , 可以通過該對象拿到源文件*/@Overridevoid visit(ASTNode[] nodes, SourceUnit source) {println nodesprintln sourceprintln source.ASTprintln source.source.reader.text// 獲取 Groovy.groovy 腳本中的類集合 , 并進(jìn)行遍歷// 在 ModuleNode 中的類節(jié)點封裝在了如下成員中// List<ClassNode> classes = new LinkedList<ClassNode>();source.AST.classes.find {// 查找名稱為 Student 的類// it 是 ClassNode 節(jié)點it.name == "Student"}?.methods?.find {// 查找 Student 類下名稱為 hello 的方法// it 是 MethodNode 節(jié)點it.name == "hello"}?.with {// 找到了 Student 下的 hello 方法// 在 MethodNode 節(jié)點下調(diào)用// it 就是 MethodNode 節(jié)點BlockStatement blockStatement = code// 清空 BlockStatement 中的 List<Statement> statements 成員// 方法攔截清空 , 就不再執(zhí)行原本的方法// 方法注入不清空 , 會執(zhí)行原來的方法內(nèi)容blockStatement.statements.clear()}} }

3、配置 ASTTransformation

創(chuàng)建 D:\002_Project\012_Groovy\Groovy_Demo\src\main\groovy\resources\META-INF\servicesorg.codehaus.groovy.transform.ASTTransformation 目錄層級及文件 , 在文件中配置 ASTTransformation 實現(xiàn)類的全類名 :

MyASTTransformation

3、使用命令行進(jìn)行編譯時處理

首先 , 進(jìn)入 D:\002_Project\012_Groovy\Groovy_Demo\src\main\groovy 目錄 ,

cd D:\002_Project\012_Groovy\Groovy_Demo\src\main\groovy

然后 , 編譯 編譯時處理類 MyASTTransformation.groovy , 將編譯后的字節(jié)碼文件 MyASTTransformation.class 保存到 D:\002_Project\012_Groovy\Groovy_Demo\src\main\groovy\classes 目錄下 ,

groovyc -d classes MyASTTransformation.groovy

再后 , 打包上述編譯好的字節(jié)碼文件 , 存放在 D:\002_Project\012_Groovy\Groovy_Demo\src\main\groovy\test.jar 路徑 ;

jar -cf test.jar -C classes . -C resources .

最后 , 依賴 test.jar 執(zhí)行 Groovy.groovy 腳本

groovy -classpath test.jar Groovy.groovy

執(zhí)行結(jié)果為 :

[org.codehaus.groovy.ast.ModuleNode@7d7758be] org.codehaus.groovy.control.SourceUnit@2bdd8394 org.codehaus.groovy.ast.ModuleNode@7d7758be class Student{def namedef hello(){println "hello"} }def student = new Student() student.hello()

總結(jié)

以上是生活随笔為你收集整理的【Groovy】编译时元编程 ( 编译时方法拦截 | 在 MyASTTransformation#visit 方法中进行方法拦截 )的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。