APT高級用法JavaPoet

1、什么是JavaPoet

JavaPort是square公司推出的開源java代碼生成框架,提供Java API生成.java源文件,這個框架功能非常實用,也是我們習慣的java面向對象OOP語法,可以很方便的使用它根據注解生成代碼,通過這種自動化生成代碼的方式,可以讓我們用更加簡潔優雅的方式替換繁瑣冗雜的重復工作。
項目主頁及源碼:https://github.com/square/javapoet

2、開發環境配置

3、JavaPoet8個常用的類

4、JavaPoet字符串格式化規則

4、JavaPoet的使用

這里我們借助APT的介紹與使用中的例子,使用javapoet生成我們想要的文件。

  • \color{red}{在compiler模塊中的build.gradle中添加javapoet的引用}
 // 幫助我們通過類調用的形式來生成Java代碼
implementation "com.squareup:javapoet:1.9.0"
  • \color{red}{新建一個注解處理器的類ARouterProcessorNew}
@AutoService(Processor.class)
@SupportedAnnotationTypes("com.migill.annotation.ARouter")
@SupportedSourceVersion(SourceVersion.RELEASE_7)
@SupportedOptions("content")
public class ARouterProcessorNew extends AbstractProcessor {
    private Elements elementUtils;
    private Types typeUtils;
    private Messager messager;
    private Filer filer;

    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        elementUtils = processingEnvironment.getElementUtils();
        typeUtils = processingEnvironment.getTypeUtils();
        messager = processingEnvironment.getMessager();
        filer = processingEnvironment.getFiler();
        String content = processingEnvironment.getOptions().get("content");
        messager.printMessage(Diagnostic.Kind.NOTE, content);
    }

    @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        if (set.isEmpty()) return false;
        Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(ARouter.class);
        for (Element element : elements) {
            String packageName = elementUtils.getPackageOf(element).getQualifiedName().toString();
            String className = element.getSimpleName().toString();
            String finalClassName = className + "$ARouter";
            messager.printMessage(Diagnostic.Kind.NOTE, "生成的文件名:" + finalClassName);
            try {
                // 高級寫法,javapoet構建工具
                ARouter aRouter = element.getAnnotation(ARouter.class);
                //構造方法體
                MethodSpec method = MethodSpec.methodBuilder("findTargetClass")//方法名
                        .addModifiers(Modifier.PUBLIC, Modifier.STATIC)
                        .returns(Class.class)
                        .addParameter(String.class, "path")
                        .addStatement("return path.equals($S) ? $T.class : null", aRouter.path(), ClassName.get((TypeElement) element))
                        .build();

                //構建類
                TypeSpec typeSpec = TypeSpec.classBuilder(finalClassName)
                        .addModifiers(Modifier.PUBLIC)
                        .addMethod(method)//添加方法體
                        .build();

                //在指定包名下,生成Java類文件
                JavaFile javaFile = JavaFile.builder(packageName, typeSpec).build();
                javaFile.writeTo(filer);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return false;
    }
}

現在編譯,就可以生成對應的文件了,如下圖,就是生成的文件與文件的內容

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。