文章目錄

• 一、在 MyASTTransformation#visit 方法中進行方法注入
• • 1、使用 new AstBuilder().buildFromSpec 進行方法注入
  • 2、使用 new AstBuilder().buildFromString 進行方法注入
  • 3、使用 new AstBuilder().buildFromCode 進行方法注入
• 二、完整代碼示例及進行編譯時處理的編譯過程
• • 1、Groovy 腳本 Groovy.groovy
  • 2、ASTTransformation 接口實現 MyASTTransformation.groovy
  • 3、配置 ASTTransformation
  • 3、使用命令行進行編譯時處理

一、在 MyASTTransformation#visit 方法中進行方法注入

---

在 【Groovy】編譯時元編程 ( 編譯時方法攔截 | 在 MyASTTransformation#visit 方法中進行方法攔截 ) 博客中的 方法攔截的基礎上進行方法注入 ;

首先對 MethodNode 進行處理

// 找到了 Student 下的 hello 方法// 在 MethodNode 節點下調用// it 就是 MethodNode 節點BlockStatement blockStatement = code// 清空 BlockStatement 中的 List<Statement> statements 成員// 方法攔截清空 , 就不再執行原本的方法// 方法注入不清空 , 會執行原來的方法內容blockStatement.statements.clear()

如果將 blockStatement.statements 清理了 , 就不會再執行 Student#hello 原本的方法內容 ;

保留 blockStatement.statements 原來的集合元素 , 繼續向其中添加其它元素 , 可以在原方法的基礎上執行其它內容 ;

1、使用 new AstBuilder().buildFromSpec 進行方法注入

先創建方法節點 ,

// 創建方法節點def methods = new AstBuilder().buildFromSpec {expression {methodCall {variable('this')constant('println')argumentList {constant('hello buildFromSpec')}}}}

然后將方法節點 , 添加到 blockStatement.statements 集合中 ;

// 將方法節點添加到 hello 方法中blockStatement.statements.addAll(methods)

2、使用 new AstBuilder().buildFromString 進行方法注入

// 創建方法節點def methods2 = new AstBuilder().buildFromString('println "hello buildFromString"')// 將方法節點添加到 hello 方法中blockStatement.statements.addAll(methods2)

3、使用 new AstBuilder().buildFromCode 進行方法注入

// 創建方法節點, 注意此處拿到的是def methods3 = new AstBuilder().buildFromCode {println "hello buildFromCode"}// 將方法節點添加到 hello 方法中blockStatement.statements.addAll(methods3[0].statements)

二、完整代碼示例及進行編譯時處理的編譯過程

---

1、Groovy 腳本 Groovy.groovy

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

2、ASTTransformation 接口實現 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 抽象語法樹節點 , 是 ASTNode 數組類型* @param source 源單元 , 可以通過該對象拿到源文件*/@Overridevoid visit(ASTNode[] nodes, SourceUnit source) {println nodesprintln sourceprintln source.ASTprintln source.source.reader.text// 獲取 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、配置 ASTTransformation

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

MyASTTransformation

3、使用命令行進行編譯時處理

首先 , 進入 D:\002_Project\012_Groovy\Groovy_Demo\src\main\groovy 目錄 ,

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

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

groovyc -d classes MyASTTransformation.groovy

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

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

最后 , 依賴 test.jar 執行 Groovy.groovy 腳本

groovy -classpath test.jar Groovy.groovy

執行結果為 :

[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() [org.codehaus.groovy.ast.ModuleNode@16ce702d] org.codehaus.groovy.control.SourceUnit@7b94089b org.codehaus.groovy.ast.ModuleNode@16ce702d println "hello buildFromString" [org.codehaus.groovy.ast.ModuleNode@72c28d64] org.codehaus.groovy.control.SourceUnit@6492fab5 org.codehaus.groovy.ast.ModuleNode@72c28d64 __synthesized__label__1644323893072__:{println "hello buildFromCode"} hello buildFromCode

總結

以上是生活随笔為你收集整理的【Groovy】编译时元编程 ( 编译时方法注入 | 使用 buildFromSpec、buildFromString、buildFromCode 进行方法注入 )的全部內容，希望文章能夠幫你解決所遇到的問題。

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