APT介绍与使用(含demo)

1 什么是APT

APT(Annotation Processing Toll)，是一种处理注解的工具，它对源代码文件进行检测找出其中的Annotation，根据注解自动生成代码，如果想要自定义的注解处理器能够正常运行，必须要通过APT工具进行处理。简单来说就是在编译期，通过注解生成 .java 文件。

APT也是AOP(Aspect-Oriented Programming)的一种实现方式。

使用到的APT的框架：DataBinding、Dagger2、ButterKnife、EventBus3、ARouter等

2 什么是Annotation

2.1 注解分为三类

  • 标准 Annotation
    包括 Override, Deprecated, SuppressWarnings，是java自带的几个注解，他们由编译器来识别，不会进行编译， 不影响代码运行

  • 元 Annotation
    @Retention, @Target, @Inherited, @Documented，它们是用来定义 Annotation 的 Annotation。

  • 自定义 Annotation
    使用@interface自定义注解时，自动继承了java.lang.annotation.Annotation接口，由编译程序自动完成其他细节。在定义注解时，不能继承其他的注解或接口。@interface用来声明一个注解，其中的每一个方法实际上是声明了一个配置参数。方法的名称就是参数的名称，返回值类型就是参数类型（只能是基本类型、Class、String、enum）。可以通过default来声明参数的默认值。

2.2 自定义的注解也分为三类

  • @Retention(RetentionPolicy.SOURCE)
    源码时注解，一般用来作为编译器标记。如Override, Deprecated, SuppressWarnings。

  • @Retention(RetentionPolicy.RUNTIME)
    运行时注解，在运行时通过反射去识别的注解。运行时注解的实质是，在代码中通过注解进行标记，运行时通过反射寻找标记进行某种处理。而运行时注解中使用到了反射，对性能会有一定的影响。

  • @Retention(RetentionPolicy.CLASS)
    编译时注解，在编译时被识别并处理的注解。编译时按照一定策略生成代码，避免编写重复代码，提高开发效率，且不影响性能。

3 APT相关API

3.1 AbstractProcessor

其中五个比较常用的方法

3.1.1 init(ProcessingEnvironment): void

该方法主要用于一些初始化的操作，通过该方法的参数ProcessingEnvironment可以获取一些有用的工具类，如

  • Elements： 操作Element工具类 (类、函数、属性都是Element)
  • Types：type(类信息)工具类，包含用于操作TypeMirror的工具方法
  • Messager：用来报告错误，警告和其他提示信息
  • Filer：文件生成器 类/资源，Filter用来创建新的源文件，class文件以及辅助文件

3.1.2 process(Set< ? extends TypeElement >,RoundEnvironment): boolean

注解处理器的核心方法，处理具体的注解，生成Java文件。
参数set：使用了支持处理的注解的节点集合（被注解标记的类的集合）。
参数roundEnvironment：当前或是之前的运行环境,可以通过该对象查找指定注解的集合。
返回值：true 表示已经处理完成，后续处理器不会再处理

3.1.3 getSupportedSourceVersion(): SourceVersion

指定JDK编译版本

3.1.4 getSupportedAnnotationTypes(): Set< String >

指定这个注解处理器允许/支持的注解类型

3.1.5 getSupporteOptions(): Set< String >

注解处理器接收的参数

3.1.6 后面的三个方法可以通过注解的形式完成

// 通过auto-service中的@AutoService可以自动生成AutoService注解处理器，用来注册
// 用来生成 META-INF/services/javax.annotation.processing.Processor 文件
@AutoService(Processor.class)
// 允许/支持的注解类型，让注解处理器处理
@SupportedAnnotationTypes({"com.jokerwan.annotation.JRouter"})
// 指定JDK编译版本
@SupportedSourceVersion(SourceVersion.RELEASE_8)
// 注解处理器接收的参数
@SupportedOptions("content")
public class ARouterProcessor extends AbstractProcessor {

}

4 APT练手demo实现ARouter生成路由辅助类

4.1 新建名为annotation的java library

定义标记类的路由地址的注解JRouter

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.CLASS)
public @interface JRouter {

    /**
     * 详细路由路径（必填），如："/app/MainActivity"
     */
    String path();

    /**
     * 路由组名（选填，如果开发者不填写，可以从path中截取出来）
     */
    String group() default "";
}

该注解是作用在类之上，所以Target是ElementType.TYPE，我们想要通过该注解在编译时生成一些辅助路由的文件，所以Retention是RetentionPolicy.CLASS，定义了注解的两个参数path和group，前者必填，后者选填，后者可截取一级路径，即第一个"/"和第二个"/"直接的路径名。

4.2 模拟APT生成后的代码XXXActivity$$ARouter 

public class XXXActivity$$ARouter {

    public static Class<?> findTargetClass(String path) {
        if (path.equals("/app/MainActivity")) {
            return MainActivity.class;
        }
        return null;
    }
}

此类是我们要通过APT生成的代码，类中有一个findTargetClass(String)方法，通过传入path获取到对应的Class。

4.3 新建名为compiler的java library

4.3.1 修改build.gradle文件如下

apply plugin: 'java-library'

dependencies {
    implementation fileTree(dir: 'libs', include: ['*.jar'])

    compileOnly'com.google.auto.service:auto-service:1.0-rc4'
    annotationProcessor'com.google.auto.service:auto-service:1.0-rc4'

    // 引入annotation，让注解处理器-处理注解
    implementation project(':annotation')
}

// java控制台输出中文乱码
tasks.withType(JavaCompile) {
    options.encoding = "UTF-8"
}

// jdk编译版本1.8
sourceCompatibility = "8"
targetCompatibility = "8"

这里引用 auto-service的目的是注册注解，并对其生成META-INF的配置信息，需要注意的是这里引用 auto-service 和 gradle 版本号有坑，编译的时候会出错，目前没有问题的两种版本号配置对应如下

  • As-3.2.1 + gradle4.10.1-all + auto-service:1.0-rc2
  • As-3.4.1 + gradle5.1.1-all + auto-service:1.0-rc4

4.3.2 实现注解处理器

两种方式

  • 手动生成java代码
  • 通过JavaPoet自动生成java代码（推荐，不易出错）
方式一：手动生成java代码
// 通过auto-service中的@AutoService可以自动生成AutoService注解处理器
// 用来生成 META-INF/services/javax.annotation.processing.Processor 文件
@AutoService(Processor.class)
// 允许/支持的注解类型，让注解处理器处理
@SupportedAnnotationTypes({"com.jokerwan.annotation.JRouter"})
// 指定JDK编译版本
@SupportedSourceVersion(SourceVersion.RELEASE_8)
// 注解处理器接收的参数
@SupportedOptions("content")
public class JRouterProcessor extends AbstractProcessor {

    // 操作Element工具类 (类、函数、属性都是Element)
    private Elements elementUtils;

    // type(类信息)工具类，包含用于操作TypeMirror的工具方法
    private Types typeUtils;

    // Messager用来报告错误，警告和其他提示信息
    private Messager messager;

    // 文件生成器 类/资源，Filter用来创建新的源文件，class文件以及辅助文件
    private Filer filer;

    /**
     * 初始化
     *
     * @param processingEnvironment 通过该方法的参数ProcessingEnvironment可以获取一些有用的工具类
     */
    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        elementUtils = processingEnvironment.getElementUtils();
        messager = processingEnvironment.getMessager();
        filer = processingEnvironment.getFiler();
        typeUtils = processingEnvironment.getTypeUtils();
        // 通过ProcessingEnvironment去获取build.gradle传过来的参数
        String content = processingEnvironment.getOptions().get("content");
        // 有坑：Diagnostic.Kind.ERROR，异常会自动结束，这里打印日志用Diagnostic.Kind.NOTE
        messager.printMessage(Diagnostic.Kind.NOTE, content);
    }

    /**
     * 处理具体的注解，生成Java文件
     *
     * @param set              使用了支持处理注解的节点集合（被注解标记的类的集合）
     * @param roundEnvironment 当前或是之前的运行环境,可以通过该对象查找找到的注解
     * @return true            表示已经处理完成，后续处理器不会再处理
     */
    @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        if (set.isEmpty()) {
            return false;
        }

        // 获取所有带JRouter注解的 类节点
        Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(JRouter.class);
        // 遍历所有类节点
        for (Element element : elements) {
            // 通过类节点获取包节点（全路径：com.jokerwan.xxx）
            String packageName = elementUtils.getPackageOf(element).getQualifiedName().toString();
            // 获取简单类名
            String className = element.getSimpleName().toString();
            messager.printMessage(Diagnostic.Kind.NOTE, "被注解的类有：" + className);
            // 最终想生成的类文件名
            String finalClassName = className + "$$ARouter";

            // 创建一个新的源文件（Class），并返回一个对象以允许写入它
            try {
                JavaFileObject sourceFile = filer.createSourceFile(packageName + "." + finalClassName);
                // 定义Writer对象，开启写入
                Writer writer = sourceFile.openWriter();
                writer.write("package " + packageName + ";\n");
                writer.write("public class " + finalClassName + " {\n");
                writer.write("public static Class<?> findTargetClass(String path) {\n");
                // 获取类之上的@JRouter注解
                JRouter jRouter = element.getAnnotation(JRouter.class);
                writer.write("if (path.equals(\""+ jRouter.path() +"\")) {\n");
                writer.write("return " + className + ".class;\n}\n");
                writer.write("return null;\n");
                writer.write("}\n}");

                writer.close();
            } catch (IOException e) {
                e.printStackTrace();
            }

        }
        return true;
    }
}

通过手动方式生成java文件需要严格按照java文件中的格式，注意空格、分号、换行，大部分编译不通过都是在写java文件的时候一些小细节出错了。

方式二：通过JavaPoet自动生成java代码

在compiler模块的build.gradle中添加依赖JavaPoet依赖

implementation "com.squareup:javapoet:1.10.0"

注解处理器代码如下

// 通过auto-service中的@AutoService可以自动生成AutoService注解处理器
// 用来生成 META-INF/services/javax.annotation.processing.Processor 文件
@AutoService(Processor.class)
// 允许/支持的注解类型，让注解处理器处理
@SupportedAnnotationTypes({"com.jokerwan.annotation.JRouter"})
// 指定JDK编译版本
@SupportedSourceVersion(SourceVersion.RELEASE_8)
// 注解处理器接收的参数
@SupportedOptions("content")
public class JRouterProcessor extends AbstractProcessor {

    // 操作Element工具类 (类、函数、属性都是Element)
    private Elements elementUtils;

    // type(类信息)工具类，包含用于操作TypeMirror的工具方法
    private Types typeUtils;

    // Messager用来报告错误，警告和其他提示信息
    private Messager messager;

    // 文件生成器 类/资源，Filter用来创建新的源文件，class文件以及辅助文件
    private Filer filer;

    /**
     * 初始化
     *
     * @param processingEnvironment 通过该方法的参数ProcessingEnvironment可以获取一些有用的工具类
     */
    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        elementUtils = processingEnvironment.getElementUtils();
        messager = processingEnvironment.getMessager();
        filer = processingEnvironment.getFiler();
        typeUtils = processingEnvironment.getTypeUtils();
        // 通过ProcessingEnvironment去获取build.gradle传过来的参数
        String content = processingEnvironment.getOptions().get("content");
        // 有坑：Diagnostic.Kind.ERROR，异常会自动结束，这里打印日志用Diagnostic.Kind.NOTE
        messager.printMessage(Diagnostic.Kind.NOTE, content);
    }

    /**
     * 处理具体的注解，生成Java文件
     *
     * @param set              使用了支持处理注解的节点集合（被注解标记的类的集合）
     * @param roundEnvironment 当前或是之前的运行环境,可以通过该对象查找找到的注解
     * @return true            表示已经处理完成，后续处理器不会再处理
     */
    @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        if (set.isEmpty()) {
            return false;
        }

        // 获取所有带JRouter注解的 类节点
        Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(JRouter.class);
        // 遍历所有类节点
        for (Element element : elements) {
            // 通过类节点获取包节点（全路径：com.jokerwan.xxx）
            String packageName = elementUtils.getPackageOf(element).getQualifiedName().toString();
            // 获取简单类名
            String className = element.getSimpleName().toString();
            messager.printMessage(Diagnostic.Kind.NOTE, "被注解的类有：" + className);
            // 最终想生成的类文件名
            String finalClassName = className + "$$ARouter";

            // 获取类之上@JRouter注解的path值
            JRouter jRouter = element.getAnnotation(JRouter.class);
            // 构建方法体
            MethodSpec methodSpec = MethodSpec.methodBuilder("findTargetClass")// 方法名
                    .addModifiers(Modifier.PUBLIC, Modifier.STATIC)// 修饰符
                    .addParameter(String.class, "path")// 参数
                    .addStatement("return path.equals($S) ? $T.class : null",
                            jRouter.path(), ClassName.get((TypeElement) element))// 代码语句拼接
                    .returns(Class.class)// 返回值类型
                    .build();
            // 构建类
            TypeSpec typeSpec = TypeSpec.classBuilder(finalClassName)
                    .addModifiers(Modifier.PUBLIC)
                    .addMethod(methodSpec)// 添加方法体
                    .build();
            // 在指定的包名下，生成Java类文件
            JavaFile javaFile = JavaFile.builder(packageName, typeSpec).build();
            try {
                javaFile.writeTo(filer);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return true;
    }
}

4.3.3 app模块分别依赖annotation和compiler模块

修改app模块build.gradle文件如下

apply plugin: 'com.android.application'

android {
    compileSdkVersion 29
    buildToolsVersion "29.0.2"
    defaultConfig {
        applicationId "com.jokerwan.apt_demo"
        minSdkVersion 21
        targetSdkVersion 29
        versionCode 1
        versionName "1.0"
        testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"

        // 在gradle文件中配置选项参数值（用于APT传参接收）
        // 切记：必须写在defaultConfig节点下
        javaCompileOptions {
            annotationProcessorOptions {
                arguments = [content : 'jokerwan apt']
            }
        }
    }
    buildTypes {
        release {
            minifyEnabled false
            proguardFiles getDefaultProguardFile('proguard-android-optimize.txt'), 'proguard-rules.pro'
        }
    }
}

dependencies {
    implementation fileTree(dir: 'libs', include: ['*.jar'])
    implementation 'androidx.appcompat:appcompat:1.1.0'
    implementation 'androidx.constraintlayout:constraintlayout:1.1.3'
    testImplementation 'junit:junit:4.12'
    androidTestImplementation 'androidx.test:runner:1.2.0'
    androidTestImplementation 'androidx.test.espresso:espresso-core:3.2.0'

    // 依赖注解
    implementation project(':annotation')
    // 依赖注解处理器
    annotationProcessor project(':compiler')

}

注意javaCompileOptions这里是给注解处理器JRouterProcessor上的注解@SupportedOptions("content")传递参数

rebuild一下工程，Build成功之后会打印如下信息


可以看到我们通过messager打印的信息已经出来了，一个是我们再app模块的build.gradle中传入的jokerwan apt，接着打印了被注解的类

看一下APT生成的类文件路径：

app/build/intermediates/javac/debug/compileDebugJavaWithJavac/classes/com/jokerwan/apt_demo/MainActivity$$ARouter.class

打开class文件，Android Studio已经帮我们反编译为java文件


可以看到，跟我们之前模拟的类一样，通过该类调用findTargetClass(String)方法传入标记在类上的注解中的path，就可以拿到该类的Class对象，由于是在编译器生成类，不会对性能产生影响。

demo地址如下
https://github.com/isJoker/APT_Demo

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
  • 人面猴
    序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
    沈念sama阅读 158,425评论 4赞 361
  • 死咒
    序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
    沈念sama阅读 67,058评论 1赞 291
  • 救了他两次的神仙让他今天三更去死
    文/潘晓璐 我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
    开封第一讲书人阅读 108,186评论 0赞 243
  • 道士缉凶录：失踪的卖姜人
    文/不坏的土叔 我叫张陵，是天一观的道长。 经常有香客问我，道长，这世上最难降的妖魔是什么？ 我笑而不...
    开封第一讲书人阅读 43,848评论 0赞 204
  • 港岛之恋（遗憾婚礼）
    正文 为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
    茶点故事阅读 52,249评论 3赞 286
  • 恶毒庶女顶嫁案：这布局不是一般人想出来的
    文/花漫 我一把揭开白布。 她就那样静静地躺着，像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上，一...
    开封第一讲书人阅读 40,554评论 1赞 216
  • 城市分裂传说
    那天，我揣着相机与录音，去河边找鬼。 笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。 我是一名探鬼主播，决...
    沈念sama阅读 31,830评论 2赞 312
  • 双鸳鸯连环套：你想象不到人心有多黑
    文/苍兰香墨 我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
    开封第一讲书人阅读 30,536评论 0赞 197
  • 万荣杀人案实录
    序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
    沈念sama阅读 34,239评论 1赞 241
  • 护林员之死
    正文 独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,505评论 2赞 244
  • 白月光启示录
    正文 我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,004评论 1赞 258
  • 活死人
    序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
    沈念sama阅读 28,346评论 2赞 253
  • 日本核电站爆炸内幕
    正文 年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
    茶点故事阅读 32,999评论 3赞 235
  • 男人毒药：我在死后第九天来索命
    文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,060评论 0赞 8
  • 一桩弑父案，背后竟有这般阴谋
    文/苍兰香墨 我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,821评论 0赞 194
  • 情欲美人皮
    我被黑心中介骗来泰国打工， 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。 一个月前我还...
    沈念sama阅读 35,574评论 2赞 271
  • 代替公主和亲
    正文 我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
    茶点故事阅读 35,480评论 2赞 267