Javac黑客指南
這篇文章英文的原文在:http://scg.unibe.ch/archive/projects/Erni08b.pdf. 做畢業(yè)設(shè)計報告時,老師要求必須翻譯一篇外文,于是很認(rèn)真的翻譯了一下,也算是為開源做一點小貢獻(xiàn)。翻譯如下:
Javac黑客指南
David Erni and Adrian Kuhn
University of Bern, March 2008
0. 摘要:
這篇文章是介紹修改java編譯器的。其中包含對Java編譯器的介紹,以及兩個例子的實現(xiàn):一個簡單的hello world 和一個重寫AST(抽象語法樹)的插件。
1. 介紹
隨著Java 6的發(fā)布,java編譯器已經(jīng)有了開源的版本了。開源的編譯器是OpenJDK項目的一部分,可以從Java編譯器小組的網(wǎng)站下載 http://www.openjdk.org/groups/compiler/ 。然而就這篇文檔的例子來說,任何Java 6的版本都是可用的,因為這些例子并不會重新編譯編譯器,他們只是擴展編譯器的功能。
這篇文章介紹了Java編譯器的內(nèi)在實現(xiàn)。首先我們給出java編譯器所包含的編譯步驟,然后我們在編寫兩個例子。兩個例子都使用到了編譯器里面的插件機制,也就是JSR269所描述的機制。然而,這兩個例子卻超出了JSR269的范圍。把JSR對象和編譯器對接,我們實現(xiàn)了AST的重寫。我們的例子里面,沒有使用assertion(斷言)語句,而使用了if-throw語句。
2. Java編譯器的內(nèi)核
這個部分概括了OpenJDK里面的java編譯器的編譯步驟和對應(yīng)的注釋。這個小節(jié)包含了一個簡短的介紹。
編譯的過程是由定義在com.sun.tools.javac.main里面的Java Compiler類來決定的。當(dāng)編譯器以默認(rèn)的編譯參數(shù)編譯時,它會執(zhí)行以下步驟:
a) Parse: 讀入一堆*.java源代碼,并且把讀進(jìn)來的符號(Token)映射到AST節(jié)點上去。
b) Enter: 把類的定義放到符號表(Symbol Table)中去。
c) Process annotations: 可選的。處理編譯單元(compilation units)里面所找到的標(biāo)記(annotation)。
d) Attribute: 為AST添加屬性。這一步包含名字解析(name resolution),類型檢測(type checking)和常數(shù)折疊(constant fold)。
e) Flow: 為前面得到的AST執(zhí)行流分析(Flow analysis)操作。這個步驟包含賦值(assignment)的檢查和可執(zhí)行性(reachability)的檢查。
f) Desugar: 重寫AST, 并且把一些復(fù)雜的語法轉(zhuǎn)化成一般的語法。
g) Generate: 生成源文件或者類文件。
wps_clip_image-12054_thumb
2.1 Parse
想要Parse文件,編譯器要用到com.sun.tools.javac.parser.*里面的類。作為***步,詞法分析器(lexical analyzer)把輸入的字符流(character sequence)映射成一個符號流(token sequence)。然后Parser再把生成的符號流映射成一個抽象語法樹(AST)
2.2 Enter
在這個步驟中,編譯器會找到當(dāng)前范圍(enclosing scope)中發(fā)現(xiàn)的所有的定義(definitions),并且把這些定義注冊成符號(symbols)。Enter這個步驟又分為以下兩個階段:
在***個階段,編譯器會注冊所有類的符號,并且把這寫符號和相應(yīng)的范圍(scope)聯(lián)系在一起。實現(xiàn)方法是使用一個Visitor(訪問者)類,由上而下的遍歷AST,訪問所有的類,包括類里面的內(nèi)部類。Enter給每一個類的符號都添加了一個MemberEnter對象,這個對象是由第二個階段來調(diào)用的
在第二個階段中,這些類被MemberEnter對象所完成(completed,即完成類的成員變量的Enter)。首先,MemberEnter決定一個類的參數(shù),父類和接口。然后這些符號被添加進(jìn)了類的范圍中。不像前一個步驟,這個步驟是懶惰執(zhí)行的。類的成員只有在被訪問時,才加入類的定義中的。這里的實現(xiàn),是通過安裝一個完成對象(member object)到類的符號中。這些對象可以在需要時調(diào)用member-enter
***,enter把所有的頂層類(top-level classes)放到一個todo-queue中,
2.3 Process Annotations
如果存在標(biāo)記處理器,并且編譯參數(shù)里面指定要處理標(biāo)記,那么這個過程就會處理在某個編譯單元里面的標(biāo)記。JSR269定義了一個接口,可以用來寫這種Annotation處理插件。然而,這個接口的功能非常有限,并且不能用Collective Behavior擴展這種語言。主要的限制是JSR269不提供子方法的反射調(diào)用。
2.4 Attribute
為Enter階段生成的所有AST添加屬性。應(yīng)當(dāng)注意,Attribte可能會需要額外的文件被解析(Parse),通過SourceCompleter加入到符號表中。
大多數(shù)的環(huán)境相關(guān)的分析都是發(fā)生在這個階段的。這些分析包括名稱解析,類型檢查,常數(shù)折疊。這些都是子任務(wù)。有些子任務(wù)調(diào)用下列的一些類,但也可能調(diào)用其他的。
l Check:這是用于類型檢查的類。當(dāng)有完成錯誤(completion error)或者類型錯誤時,它就會報錯。
l Resovle: 這是名字解析的類。如果解析失敗,就會報錯。
l ConstFold: 這是參數(shù)折疊類。常數(shù)折疊用于簡化在編譯時的常數(shù)表達(dá)式。
l Infer:類參數(shù)引用的類。
2.5 Flow
這個階段會對添加屬性后的類,執(zhí)行數(shù)據(jù)流的檢查。存活性分析(liveness analysis) 檢查是否每個語句都可以被執(zhí)行到。異常分析(Excepetion analysis) 檢查是豆每個被拋出的異常都是聲明過的,并且這些異常是否都會被捕獲。確定行賦值(definite assignment)分析保證每個變量在使用時已經(jīng)被賦值。而確定性不賦值(definite unassignment)分析保證final變量不會被多次賦值。
2.6 Desugar
除去多余的語法,像內(nèi)部類,類的常數(shù),assertion斷言語句,foreach循環(huán)等。
2.7 Generate
這是最終的階段。這個階段生成許多源文件或者類文件。到底是生成源文件還是類文件取決于編譯選項。
3. 什么是JSR 269
Annotation(標(biāo)記)是java 5里面引進(jìn)來的,用于在源代碼里面附加元信息(meta-information).Java 6則進(jìn)一步加強了標(biāo)記的處理功能,即JSR269. JSR269,即插入式標(biāo)記處理API,為java編譯器添加了一個插件機制。有了JSR269,就有能力為java編譯器寫一個特定的標(biāo)記處理器了。
JSR269有兩組基本API,一組用于對java語言的建模,一組用于編寫標(biāo)記處理器。這兩組API分別存在于javax.lang.model.* 和 javax.annotation.processing里面。JSR269的功能是通過以下的java編譯選項來調(diào)用的。
-proc:{none,only} 是否執(zhí)行Annotation處理或者編譯
-processor <classes> 指定標(biāo)記處理器的名字。這個選項將繞過默認(rèn)的標(biāo)記處理器查找過程
-processorpath <path> 指定標(biāo)記處理器的位置
標(biāo)記處理在javac中時默認(rèn)開啟的。如果要是只想處理標(biāo)記,而不想編譯生成類文件的話,用 –proc:only 選項既即可。
#p#
4. 如何用Javac打印出“Hello World!”
在這***個例子里面,我們些一個簡單的標(biāo)記處理器,用于在編譯的時候打印“Hello World!”.我們用編譯器的內(nèi)部消息機制來打印“hello world”。
首先,我們定義如下HelloWorld標(biāo)記。
- public @interface HelloWorld{
- }
添加一個Dummy類使用以上的標(biāo)記
- @HelloWorld
- public class Dummy{
- }
標(biāo)記處理可能會發(fā)生很輪。每一輪處理器只處理特定的一些標(biāo)記,并且生成的源文件或者類文件,交給下一輪來處理。如果處理器被要求只處理特定的某一輪,那么他也會處理后續(xù)的那些次,包括***一輪,就算***一輪沒有可以處理的標(biāo)記。處理器可能也會去處理被這個工具生成的文件。
后一個方法處理前一輪生成的標(biāo)記類型,并且返回是否這些標(biāo)記會聲明。如果返回是True,那么后續(xù)的處理器就不會去處理它們。如果返回是false,那么后續(xù)處理器會繼續(xù)處理它們。一個處理器可能總是返回同樣的邏輯值,或者是根據(jù)選項改變結(jié)果。為了要寫一個標(biāo)記處理器,我們用一個子類來繼承AbstractProcessor,并且用SupportedAnnotationTyps 和SupportedSourceVersion標(biāo)記這個子類。這個子類必須要復(fù)寫這兩個方法:
l public synchronized void init(ProcessingEnvironment processingEnv)
l public boolean process(Set<? extends TypeElement> annotations,
RoundEnvironment roundEnv)
這兩個方法都是在標(biāo)記處理過程中被java編譯器調(diào)用的。***個方法用來初始化插件,只被調(diào)用一次。而第二個方法每一輪標(biāo)記處理都會被調(diào)用,并且在所有處理都結(jié)束后還會調(diào)用一次。
我們的簡單的HelloWorldProcessors是這樣生成的:
- import javax.annotation.processing.*;
- import javax.lang.model.SourceVersion;
- import javax.lang.model.element.TypeElement;
- import javax.tools.Diagnostic;
- @SupportedAnnotationTypes("HelloWorld")
- @SupportedSourceVersion(SourceVersion.RELEASE_6)
- public class HelloWorldProcessor extends AbstractProcessor {
- @Override
- public synchronized void init(ProcessingEnvironment processingEnv) {
- super.init(processingEnv);
- }
- @Override
- public boolean process(Set<? extends TypeElement> annotations,
- RoundEnvironment roundEnv) {
- if (!roundEnv.processingOver()) {
- processingEnv.getMessager().printMessage(
- Diagnostic.Kind.NOTE, "Hello Worlds!");
- }
- return true;
- }
- }
第八行注冊了HelloWorld的標(biāo)記處理器。也就是說,當(dāng)標(biāo)記出現(xiàn)是,就會有一系列的程序被自動調(diào)用。第九行設(shè)置了標(biāo)記所支持的源代碼版本。
第12到15行復(fù)寫了初始化方法, 目前為止,我們只是調(diào)用父類的方法。
第17到24行復(fù)寫了處理方法。這個方法是由一些列被標(biāo)記的程序元素來調(diào)用的。這個方法在每一輪處理時,都會調(diào)用,并且在***會多出一輪,用于對空集合的元素的處理。這樣,我們可以由一個簡單的if語句,使得***多出的那一輪什么事情都不做。在其他輪中,我們只打印一個hello world消息。我們不用System.out.print,二十使用編譯器的消息框架來打印一個消息(note類型的)。其他可能的類型是警告(warning)或者錯誤(error)。
這個方法返回true,如果你想要聲明元素已經(jīng)被處理過了。
要運行這個例子,執(zhí)行:
javac HelloWorldProcessor.java
javac -processor HelloWorldProcessor *.java
這個應(yīng)該會輸出:
Note: Hello World!
5. 如何巧妙利用JSR269來重寫AST
在這個例子中,我們深入到編譯器自身的實現(xiàn)細(xì)節(jié)中去。我們利用JSR269做一些超出它本身的事情—重寫AST。這個處理器會把每一個Assertion語句替換成一個throw語句。也就是說,每當(dāng)有以下語句出現(xiàn)時
assert cond: detail;
會被替換成:
If(!cond) throw new AssertionError(detail);
后面的這個語句不會生成assert的字節(jié)碼,而是生成一個普通的if語句,帶有一個throw重句。結(jié)果就算你的虛擬機沒有激活assertions功能時,assertions的檢查還是會被執(zhí)行。這個功能對各種庫是非常有用的,因為你寫庫的時候,是沒有辦法控制用戶的VM設(shè)置的。
再次,我們還是先繼承AbstractProcessor。然而,這次我們不會針對某一個特殊的標(biāo)記,而是用“*”這個符號來表示對所有的源代碼都調(diào)用處理器。
- @SupportedAnnotationTypes("*")
- @SupportedSourceVersion(SourceVersion.RELEASE_6)
- public class ForceAssertions extends AbstractProcessor {
- }
初始化方法如下:
- private int tally;
- private Trees trees;
- private TreeMaker make;
- private Name.Table names;
- @Override
- public synchronized void init(ProcessingEnvironment env) {
- super.init(env);
- trees = Trees.instance(env);
- Context context = ((JavacProcessingEnvironment)
- env).getContext();
- make = TreeMaker.instance(context);
- names = Name.Table.instance(context);
- tally = 0;
- }
我們使用處理環(huán)境(ProcessingEnvironment)來獲得對編譯器一些組件的引用。在編譯器里面,在每次調(diào)用編譯器時都會有一個處理環(huán)境(ProcessingEnvironment)。在編譯器中,我們使用Component.instance(context)來獲得對組件的引用。
我們使用的組件如下:
l Trees – JSR269的一個工具類,用于聯(lián)系程序元素和樹節(jié)點。比如,對于一個方法元素,我們可以獲得這個元素對應(yīng)的AST樹節(jié)點。
l TreeMaker – 編譯器的內(nèi)部組件,是用于創(chuàng)建樹節(jié)點的工廠類。工廠類里面方法的命名方式跟Javac源代碼里面的方法是統(tǒng)一的。
l Name.Table – 另一個編譯器的內(nèi)部組件。Name類是編譯器內(nèi)部字符串的一個抽象。為了提高效率,Javac使用了哈希字符串。
請注意,在第39行,我們把處理環(huán)境(ProcessingEnvironment)強制轉(zhuǎn)換成了編譯器的內(nèi)部類型。
***,我們把一個計數(shù)器初始化成0.這個計數(shù)器是用來記錄發(fā)生替換的數(shù)量。
處理方法如下:
- @Override
- 46 public boolean process(Set<? extends TypeElement> annotations,
- RoundEnvironment roundEnv) {
- if (!roundEnv.processingOver()) {
- Set<? extends Element> elements = roundEnv.getRootElements();
- for (Element each : elements) {
- if (each.getKind() == ElementKind.CLASS) {
- JCTree tree = (JCTree) trees.getTree(each);
- TreeTranslator visitor = new Inliner();
- tree.accept(visitor);
- }
- }
- } else
- processingEnv.getMessager().printMessage(Diagnostic.Kind.NOTE,
- tally + " assertions inlined.");
- return false;
- }
我們遍歷所有的程序元素,為每一個類都重寫AST。在第51行,我們把JSR269的樹節(jié)點轉(zhuǎn)換成編譯器內(nèi)部的樹節(jié)點。這兩種樹節(jié)點的不同之處在于,JSR269節(jié)點是停留在方法層的(即方法method是最基本的元素,不會再細(xì)分下去),而內(nèi)部的AST節(jié)點,是所有元素(包括方法以下的)都可以訪問的。我們要訪問每一個語句,所以需要訪問到AST的所有節(jié)點。
#p#
樹的轉(zhuǎn)換是通過繼承TreeTranslator來完成的,TreeTranslator本身是繼承自TreeVisitor的。這些類都不是JSR269的一部分。所以,從這里開始,我們所寫的所有代碼都是在編譯器內(nèi)部工作的。
在第57行,是else部分,用于報告處理過的assertion語句數(shù)量。這個語句只有在***一輪處理才會執(zhí)行。
Inliner這個類實現(xiàn)了AST重寫。Inliner繼承了TreeTranslator,并且是標(biāo)記處理器的一個內(nèi)部類。注意,TreeTranslator本身是不會轉(zhuǎn)換任何節(jié)點的。
private class Inliner extends TreeTranslator {
}
為了轉(zhuǎn)換assertion語句,我們需要復(fù)寫默認(rèn)的TreeTranslator.visitAssert (JCAssert) 方法,如下所示:
- @Override
- public void visitAssert(JCAssert tree) {
- super.visitAssert(tree);
- JCStatement newNode = makeIfThrowException(tree);
- result = newNode;
- tally++;
- }
正在轉(zhuǎn)換的節(jié)點會被當(dāng)做參數(shù)傳入到方法中。在第67行,轉(zhuǎn)換的結(jié)果,通過賦值給變量TreeTranslator.result而返回。
按照慣例,一個轉(zhuǎn)換方法應(yīng)該這樣生成:
l 調(diào)用父類的轉(zhuǎn)換方法,以確保轉(zhuǎn)換可以被應(yīng)用到自己點上面去。
l 執(zhí)行真正的轉(zhuǎn)換
l 把轉(zhuǎn)換結(jié)果賦值給TreeTranslator.result。結(jié)果的類型不一定要和傳進(jìn)來的參數(shù)的類型一樣。相反,只要java編譯器允許,我們可以返回任何類型的節(jié)點。這里TreeTranslator本身沒有限制類型,但是如果返回了錯誤的類型,那么就很有在后續(xù)過程中產(chǎn)生災(zāi)難性后果。
我們寫一個私有函數(shù)來實現(xiàn)轉(zhuǎn)換,makeIfThrowException:
- private JCStatement makeIfThrowException(JCAssert node) {
- // make: if (!(condition) throw new AssertionError(detail);
- List<JCExpression> args = node.getDetail() == null
- ? List.<JCExpression> nil()
- : List.of(node.detail);
- JCExpression expr = make.NewClass(
- null,
- null,
- make.Ident(names.fromString("AssertionError")),
- args,
- null);
- return make.If(
- make.Unary(JCTree.NOT, node.cond),
- make.Throw(expr),
- null);
- }
這個方法傳入一個assertion語句,返回一個if語句。我們可以這樣做,事因為不管是assertion還是if,他們都是語句(statement),所以在java的語法中是等價的。Java中沒有明文規(guī)定,禁止用if語句來代替assertion語句。
makeIfThrowException是用于AST重寫的方法。我們使用TreeMaker來創(chuàng)建新的樹節(jié)點。如果有這樣的一個表達(dá)式:
assert cond:detail;
我們就可以替換成下面的形式:
If(!cond) throw new AssertionErrror(detal);
在第73到75行,我們考慮到了detail被省略的情況。在76到81行,我們創(chuàng)建了一個AST節(jié)點,這個節(jié)點的作用是創(chuàng)建AssertionError。在第79行,我們使用Name.Table來把字符串“AssertionError”變成編譯器內(nèi)部的字符串。在80行,我們再傳入73到75行創(chuàng)建的參數(shù)args。第77,78和81行傳入了null值,因為這個節(jié)點既沒有外部實例,也沒有類型參數(shù),也不是在匿名類內(nèi)部。
在第83行,我們對assertion的條件做了一個Not操作。84行,我們創(chuàng)建了一個throw表達(dá)式,***,在82到85行,我們把所有的東西都放到了if語句中。
注意:List類是java編譯器中另外一個令人印象深刻的實現(xiàn)。編譯器用了它自己的數(shù)據(jù)類型來實現(xiàn)List,而不是使用java集合框架(Java Collection Framework)。List和Pair數(shù)據(jù)類的實現(xiàn),都用到了Lisp語言里面所謂的cons。Pairs是這樣實現(xiàn)的:
- public class Pair<A, B> {
- public final A fst;
- public final B snd;
- public Pair(A fst, B snd) {
- this.fst = fst;
- this.snd = snd;
- }
- ...
- }
而List是這樣實現(xiàn)的:
- public class List<A> extends AbstractCollection<A> implements
- java.util.List<A> {
- public A head;
- public List<A> tail;
- public List(A head, List<A> tail) {
- this.tail = tail;
- this.head = head;
- }
- ...
- }
并且有許多靜態(tài)的方法,可以很方便的創(chuàng)建List:
l List.nil()
l List.of(A)
l List.of(A,A)
l List.of(A,A,A)
l List.of(A,A,A,A...)
Pair也是一樣:
l Pair.of(A,B)
同樣,非傳統(tǒng)的命名方式也帶來了更漂亮的代碼
不像傳統(tǒng)java中用的代碼:
List list = new List();
list.add(a);
list.add(b);
list.add(c);
而現(xiàn)在只需要寫:
List.of(a, b, c);
#p#
5.1 運行AST重寫
為了展示AST重寫,我們使用:
- public class Example {
- public static void main(String[] args) {
- String str = null;
- assert str != null : "Must not be null";
- }
- }
并且執(zhí)行:
javac ForceAssertions.java
javac -processor ForceAssertions Example.java
就會產(chǎn)生這樣的輸出:
Note: 1 assertions inlined
現(xiàn)在,我們我們我們禁用assertion,再執(zhí)行例子:
java -disableassertions Example
得到:
Exception in thread "main" java.lang.AssertionError: Must not be null at Example.main(Example.java:1)
利用編譯器的選項 –printsource,我們甚至可以得到重寫過后的AST,并且以Java源代碼的方式顯示出來。要注意的是,我們必須重定向輸出,否者原來的源文件會被覆蓋了。
執(zhí)行:
javac -processor ForceAssertions -printsource -d gen Example.java
產(chǎn)生結(jié)果:
- public class Example {
- public Example() {
- super();
- }
- public static void main(String[] args) {
- String str = null;
- if (!(str != null)) throw new AssertionError("Must not be null");
- }
- }
可以發(fā)現(xiàn),第9行已經(jīng)被重寫過了,第3到5行加入了一個默認(rèn)的構(gòu)造函數(shù)。
5.2 如何把標(biāo)記處理器注冊成服務(wù)
Java提供了一個注冊服務(wù)的機制。如果一個標(biāo)記處理器被注冊成了一個服務(wù),編譯器就會自動的去找到這個標(biāo)記處理器。注冊的方法是,在classpath中找到一個叫META-INF/services的文件夾,然后放入一個javax.annotation.processing.Processor的文件。文件格式是很明顯的,就是要包含要注冊的標(biāo)記處理器的完整名稱。每個名字都要占單獨的一行。
5.3 進(jìn)一步的閱讀
Erni在他的本科畢業(yè)設(shè)計中描述了一個更復(fù)雜的編譯器修改。他不是依賴JSR269,而是直接在編譯過程中的幾個點進(jìn)行直接修改。
參考
[1] David Erni. JAG - a Prototype for Collective Behavior in Java. Bachelors
Thesis, University of Bern. March 2008.
[2] Joseph D. Darcy. JSR-000269 Pluggable Annotation Processing API,
December 2006. http://jcp.org/en/jsr/detail?id=269.
A 在OSX下面安裝Java 6
默認(rèn)情況下,當(dāng)前的OSX只是內(nèi)置 了Java 5.0,為了安裝Java6,從以下地址下載安裝文件http://www.apple.com/support/downloads/javaformacosx105update1.html. 這個更新需要Mac OS X 10.5.2或者是更新的,而且要64位基于Intel的Mac。這個更新不會替換已經(jīng)存在的J2SE 5.0安裝,或者是改變Java的默認(rèn)版本。
B ForceAssertions.java的全部源代碼
- import java.util.Set;
- import javax.annotation.processing.AbstractProcessor;
- import javax.annotation.processing.ProcessingEnvironment;
- import javax.annotation.processing.RoundEnvironment;
- import javax.annotation.processing.SupportedAnnotationTypes;
- import javax.annotation.processing.SupportedSourceVersion;
- import javax.lang.model.SourceVersion;
- import javax.lang.model.element.Element;
- import javax.lang.model.element.ElementKind;
- import javax.lang.model.element.TypeElement;
- import javax.tools.Diagnostic;
- import com.sun.source.util.Trees;
- import com.sun.tools.javac.processing.JavacProcessingEnvironment;
- import com.sun.tools.javac.tree.JCTree;
- import com.sun.tools.javac.tree.TreeMaker;
- import com.sun.tools.javac.tree.TreeTranslator;
- import com.sun.tools.javac.tree.JCTree.JCAssert;
- import com.sun.tools.javac.tree.JCTree.JCExpression;
- import com.sun.tools.javac.tree.JCTree.JCStatement;
- import com.sun.tools.javac.util.Context;
- import com.sun.tools.javac.util.List;
- import com.sun.tools.javac.util.Name;
- @SupportedAnnotationTypes("*")
- @SupportedSourceVersion(SourceVersion.RELEASE_6)
- public class ForceAssertions extends AbstractProcessor {
- private int tally;
- private Trees trees;
- private TreeMaker make;
- private Name.Table names;
- @Override
- public synchronized void init(ProcessingEnvironment env) {
- super.init(env);
- trees = Trees.instance(env);
- Context context = ((JavacProcessingEnvironment)
- env).getContext();
- make = TreeMaker.instance(context);
- names = Name.Table.instance(context);
- tally = 0;
- }
- @Override
- public boolean process(Set<? extends TypeElement> annotations,
- RoundEnvironment roundEnv) {
- if (!roundEnv.processingOver()) {
- Set<? extends Element> elements =
- roundEnv.getRootElements();
- for (Element each : elements) {
- if (each.getKind() == ElementKind.CLASS) {
- JCTree tree = (JCTree) trees.getTree(each);
- TreeTranslator visitor = new Inliner();
- tree.accept(visitor);
- }
- }
- } else
- processingEnv.getMessager().printMessage(
- Diagnostic.Kind.NOTE, tally + " assertions
- inlined.");
- return false;
- }
- private class Inliner extends TreeTranslator {
- @Override
- public void visitAssert(JCAssert tree) {
- super.visitAssert(tree);
- JCStatement newNode = makeIfThrowException(tree);
- result = newNode;
- tally++;
- }
- private JCStatement makeIfThrowException(JCAssert node) {
- // make: if (!(condition) throw new AssertionError(detail);
- List<JCExpression> args = node.getDetail() == null
- ? List.<JCExpression> nil()
- : List.of(node.detail);
- JCExpression expr = make.NewClass(
- null,
- null,
- make.Ident(names.fromString("AssertionError")),
- args,
- null);
- return make.If(
- make.Unary(JCTree.NOT, node.cond),
- make.Throw(expr),
- null);
- }
- }
- }