插入式注解處理器在《深入理解Java虛擬機》一書中有一些介紹(前端編譯篇有提到),但一直沒有機會使用,直到碰到這個需求,覺得再合適不過了,就簡單用了一下,這里做個記錄。
了解過lombok底層原理的都知道其使用的就是的插入式注解,那么今天筆者就以真實場景演示一下插入式注解的使用。
需求
我們為公司提供了一套通用的JAVA基礎組件包,組件包內有不同的模塊,比如熔斷模塊、負載均模塊、rpc模塊等等,這些模塊均會被打成jar包,然后發布到公司的內部代碼倉庫中,供其他人引入使用。
這份代碼會不斷的迭代,我們希望可以通過promethus來監控現在公司內使用各版本代碼庫的比例,希望達到的效果圖如下:
我們希望看到每一個版本的使用率,這有利于我們做版本兼容,必要的時候可以對古早版本使用者溯源。
基于 Spring Boot + MyBatis Plus + Vue & Element 實現的后臺管理系統 + 用戶小程序,支持 RBAC 動態權限、多租戶、數據權限、工作流、三方登錄、支付、短信、商城等功能
問題
需求似乎很簡單,但真要獲取自身的jar版本號還是挺麻煩的,有個比較簡單但陰間的辦法,就是給每一個組件都加上當前的jar版本號,寫到配置文件里或者直接設置成常量,這樣上報promethus時就可以直接獲取到jar包版本號了,這個方法雖然可以解決問題,但每次迭代版本都要跟著改一遍所有組件包的版本號數據,過于麻煩。
有沒有更好的解決辦法呢?比如我們可不可以在gradle打包構建時拿到jar包的版本號,然后注入到每個組件中去呢?就像lombok那樣,不需要寫get、set方法,只需要加個注解標記就可以自動注入get、set方法。
比如我們可以給每個組件定義一個空常量,加上自定義的注解:
@TrisceliVersion publicstaticfinalStringversion="";
然后像lombok生成set/get方法那樣注入真正的版本號:
@TrisceliVersion publicstaticfinalStringversion="1.0.31-SNAPSHOT";
參考lombok的實現,這其實是可以做到的,下面來看解決方案。
基于 Spring Cloud Alibaba + Gateway + Nacos + RocketMQ + Vue & Element 實現的后臺管理系統 + 用戶小程序,支持 RBAC 動態權限、多租戶、數據權限、工作流、三方登錄、支付、短信、商城等功能
解決
java中解析一個注解的方式主要有兩種:編譯期掃描、運行期反射,這是lombok @Setter的實現:
@Target({ElementType.FIELD,ElementType.TYPE}) @Retention(RetentionPolicy.SOURCE) public@interfaceSetter{ //略... }
可以看到@Setter的Retention是SOURCE類型的,也就是說這個注解只在編譯期有效,它甚至不會被編入class文件,所以lombok無疑是第一種解析方式,那用什么方式可以在編譯期就讓注解被解析到并執行我們的解析代碼呢?答案就是定義插入式注解處理器(通過JSR-269提案定義的Pluggable Annotation Processing API實現)
插入式注解處理器的觸發點如下圖所示:
也就是說插入式注解處理器可以幫助我們在編譯期修改抽象語法樹(AST)!所以現在我們只需要自定義一個這樣的處理器,然后其內部拿到jar版本信息(因為是編譯期,可以找到源碼的path,源碼里隨便搞個文件存放版本號,然后用java io讀取進來即可),再將注解對應語法樹上的常量值設置成jar包版本號,語法樹變了,最終生成的字節碼也會跟著變,這樣就實現了我們想在編譯期給常量version注入值的愿望。
自定義一個插入式注解處理器也很簡單,首先要將自己的注解定義出來:
@Documented @Retention(RetentionPolicy.SOURCE)//只在編譯期有效,最終不會打進class文件中 @Target({ElementType.FIELD})//僅允許作用于類屬性之上 public@interfaceTrisceliVersion{ }
然后定義一個繼承了AbstractProcessor的處理器:
/** *{@linkAbstractProcessor}就屬于PluggableAnnotationProcessingAPI */ publicclassTrisceliVersionProcessorextendsAbstractProcessor{ privateJavacTreesjavacTrees; privateTreeMakertreeMaker; privateProcessingEnvironmentprocessingEnv; /** *初始化處理器 * *@paramprocessingEnv提供了一系列的實用工具 */ @SneakyThrows @Override publicsynchronizedvoidinit(ProcessingEnvironmentprocessingEnv){ super.init(processingEnv); this.processingEnv=processingEnv; this.javacTrees=JavacTrees.instance(processingEnv); Contextcontext=((JavacProcessingEnvironment)processingEnv).getContext(); this.treeMaker=TreeMaker.instance(context); } @Override publicSourceVersiongetSupportedSourceVersion(){ returnSourceVersion.latest(); } @Override publicSetgetSupportedAnnotationTypes(){ HashSet set=newHashSet<>(); set.add(TrisceliVersion.class.getName());//支持解析的注解 returnset; } @Override publicbooleanprocess(Set?extends?TypeElement>annotations,RoundEnvironmentroundEnv){ for(TypeElementt:annotations){ for(Elemente:roundEnv.getElementsAnnotatedWith(t)){//獲取到給定注解的element(element可以是一個類、方法、包等) //JCVariableDecl為字段/變量定義語法樹節點 JCTree.JCVariableDecljcv=(JCTree.JCVariableDecl)javacTrees.getTree(e); StringvarType=jcv.vartype.type.toString(); if(!"java.lang.String".equals(varType)){//限定變量類型必須是String類型,否則拋異常 printErrorMessage(e,"Type'"+varType+"'"+"isnotsupport."); } jcv.init=treeMaker.Literal(getVersion());//給這個字段賦值,也就是getVersion的返回值 } } returntrue; } /** *利用processingEnv內的Messager對象輸出一些日志 * *@parameelement *@parammerrormessage */ privatevoidprintErrorMessage(Elemente,Stringm){ processingEnv.getMessager().printMessage(Diagnostic.Kind.ERROR,m,e); } privateStringgetVersion(){ /** *獲取version,這里省略掉復雜的代碼,直接返回固定值 */ return"v1.0.1"; }
定義好的處理器需要SPI機制被發現,所以需要定義META.services:
測試
新建測試模塊,引入剛才寫好的代碼包:
這是Test類:
現在我們只需要讓gradle build一下,新得到的字節碼中該字段就有值了:
這只是插入式注解處理器 功能的冰山一角,既然它可以通過修改抽象語法樹來控制生成的字節碼,那么自然就有人能充分利用其特性來實現一些很酷的插件,比如lombok,我們再也不用寫諸如set/get這種模板式的代碼了,只要我們足夠有創意,就可以讓基于這一套API實現的插件在功能上有很大的發揮空間。
審核編輯:劉清
-
處理器
+關注
關注
68文章
19265瀏覽量
229671 -
JAVA
+關注
關注
19文章
2966瀏覽量
104704 -
SPI
+關注
關注
17文章
1706瀏覽量
91518 -
RPC
+關注
關注
0文章
111瀏覽量
11529
原文標題:項目終于用上了插入式注解,真香!
文章出處:【微信號:芋道源碼,微信公眾號:芋道源碼】歡迎添加關注!文章轉載請注明出處。
發布評論請先 登錄
相關推薦
評論