Android-Apt 注解处理器（二）

本篇是apt文章的第二篇，不太了解注解的童鞋可以先去第一篇学习一下注解，然后再看第二篇。
Android-Apt 注解处理器（一）
在学完注解以后我们的注解处理器就能很快的上手了。

一、创建annotation 注解module ，注意：该module是java modle不是Android module

1.build.gradle配置

apply plugin: 'java-library'

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

sourceCompatibility = "1.7"
targetCompatibility = "1.7"

2.编写我们自己的注解

@Retention(RetentionPolicy.CLASS)
@Documented
@Target({ElementType.TYPE, ElementType.FIELD})
public @interface Bubble {
    String name() default "123";
}

接口（根据实际情况设计）

public interface IBubble {
    String getBubble();
}

二、创建annotationProcessor module 该module也是java module

1.build.gradle配置

apply plugin: 'java-library'

dependencies {
    annotationProcessor 'com.google.auto.service:auto-service:1.0-rc4'
    implementation 'com.google.auto.service:auto-service:1.0-rc4'
    // 用于生成java类
    implementation 'com.squareup:javapoet:1.10.0'
    // 依赖我们的注解module
    implementation project(':lib-annotation')
}

sourceCompatibility = "1.8"
targetCompatibility = "1.8  "

2.AbstractProcessor 介绍

void init(ProcessingEnvironment processingEnvironment) ；
初始化、提供一些工具类。
2.Set<String> getSupportedAnnotationTypes()；
支持的注解类型，返回一个Set<String>,将我们要处理的注解全路径添加进去。
3.SourceVersion getSupportedSourceVersion()；
支持的编译版本可以指定位 SourceVersion.RELEASE_8、SourceVersion.RELEASE_7等等。
4.Set<String> getSupportedOptions()；
获取支持的选项、可以返回一下参数。
5.boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment);
这是我们要重点注意的方法，所有的注解处理以及我们要生成文件都会在这个方法中进行。

3.创建一个BubbleProcessor 继承AbstractProcessor，在init里面获取到processingEnvironment提供的各种工具

@AutoService(Processor.class)
@SupportedAnnotationTypes("com.gzgxinfo.lib_annotation.Bubble")
@SupportedSourceVersion(SourceVersion.RELEASE_8)
public class BubbleProcessor extends AbstractProcessor {
    private Filer mFiler;
    private Messager mMessager;
    private Elements mElements;

    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        mFiler = processingEnvironment.getFiler();
        mMessager = processingEnvironment.getMessager();
        mElements = processingEnvironment.getElementUtils();
        mMessager.printMessage(Diagnostic.Kind.WARNING, "初始化");
    }
    @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        mMessager.printMessage(Diagnostic.Kind.WARNING, "================================================");
        return true;
    }
}

4、实现getSupportedAnnotationTypes 方法返回我们要处理的注解

    @Override
    public Set<String> getSupportedAnnotationTypes() {
        Set<String> set = new LinkedHashSet<>();
        set.add(Bubble.class.getCanonicalName());
        mMessager.printMessage(Diagnostic.Kind.WARNING, Bubble.class.getName());
        return set;
    }

5.实现SourceVersion getSupportedSourceVersion()

  @Override
    public SourceVersion getSupportedSourceVersion() {
        return SourceVersion.latestSupported();
    }

6.实现Set<String> getSupportedOptions()


    @Override
    public Set<String> getSupportedOptions() {
        Set<String> set = new LinkedHashSet<>();
        set.add("BUBBLE");
        return set;
    }

在使用注解处理器的module的build.gradle中

defaultConfig {
        applicationId "com.test.demo2"
        minSdkVersion 16
        targetSdkVersion 30
        versionCode 1
        versionName "1.0"

        testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"
        javaCompileOptions {
            annotationProcessorOptions {
                arguments = [BUBBLE: "我是option"]
            }
        }
    }

在init中获取到我们传的参数

        Map<String, String> options = processingEnvironment.getOptions();
        mMessager.printMessage(Diagnostic.Kind.WARNING, "BUBBLE:"+options.get("BUBBLE"));

输出：

image.png

6.下面是最重要的方法，process

 @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        mMessager.printMessage(Diagnostic.Kind.WARNING, "================================================");


        List<TypeElement> bubbles = new ArrayList<>();
        Set<? extends Element> rootElements = roundEnvironment.getRootElements();
        for (Element element : rootElements) {
            // 不是类 跳过
            if (!(element instanceof TypeElement)) {
                continue;
            }
            TypeElement typeElement = (TypeElement) element;
            // 寻找到带有Bubble 注解的类 
            Bubble bubble = typeElement.getAnnotation(Bubble.class);
            mMessager.printMessage(Diagnostic.Kind.WARNING, bubble + "      " + (bubble == null));
            if (bubble == null) {
                continue;
            }
            mMessager.printMessage(Diagnostic.Kind.WARNING, "找到的类**********   " + (typeElement.getInterfaces().contains(IBubble.class)));
            mMessager.printMessage(Diagnostic.Kind.WARNING, typeElement.getQualifiedName());
            // 添加到集合
            bubbles.add(typeElement);
        }

        for (TypeElement bubble : bubbles) {
            mMessager.printMessage(Diagnostic.Kind.WARNING, "\n\n\n找到的所有类**********:   " + bubble);
        }
        if (!bubbles.isEmpty()) {

            ClassName list = ClassName.get("java.util", "List");
            ClassName arrayList = ClassName.get("java.util", "ArrayList");
            ParameterizedTypeName typeName = ParameterizedTypeName.get(list, ClassName.get(IBubble.class));
            // 创建一个IBubble 的List集合
            FieldSpec fieldSpec = FieldSpec.builder(typeName, "mList", Modifier.PRIVATE)
                    .initializer("new $T()", arrayList)
                    .build();
            // 创建个Field
            FieldSpec str = FieldSpec.builder(String.class, "str", Modifier.PRIVATE)
                    .build();

            // 把我们获取到的类 在我们要生成的java 类中new出来
            CodeBlock.Builder codeblock = CodeBlock.builder();
            for (TypeElement bubble : bubbles) {
                codeblock.addStatement("$N.add(new $T())", fieldSpec.name, ClassName.get(bubble));
            }
            // 创建方法
            MethodSpec init = MethodSpec.constructorBuilder()
                    .addModifiers(Modifier.PUBLIC)
                    .addStatement("str=$S", "555555")
                    .addCode(codeblock.build())
                    .build();
            // 创建一个方法 获取到我们的集合
            MethodSpec getList = MethodSpec.methodBuilder("getList")
                    .addModifiers(Modifier.PUBLIC)
                    .returns(typeName)
                    .addStatement("return $N", fieldSpec.name)
                    .build();
            MethodSpec setList = MethodSpec.methodBuilder("setList")
                    .addParameter(ClassName.get(IBubble.class), "item")
                    .addModifiers(Modifier.PUBLIC)
                    .addStatement("$N.add(item)", fieldSpec.name)
                    .build();

            // 创建一个类 叫 BubbleClass
            TypeSpec typeSpec = TypeSpec
                    .classBuilder("BubbleClass")
                    .addModifiers(Modifier.PUBLIC)
                    .addMethod(setList)
                    .addMethod(getList)
                    .addMethod(init)
                    .addField(fieldSpec)
                    .addField(str)
                    .build();

            // 创建java文件 并写入
            JavaFile file = JavaFile.builder("com.bubble.apt", typeSpec).build();
            try {
                file.writeTo(mFiler);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return true;
    }

三、使用，创建app module

在build.gradle 中添加我们的注解处理器还有注解的依赖

    implementation project(':lib-annotation')
    annotationProcessor project(":lib-apt")

编写一个类实现IBubble 并使用我们的注解

@Bubble
public class Bubble1 implements IBubble {
    @Override
    public String getBubble() {
        return "Bubble1";
    }
}

编译过后生成

image.png

到此我们的注解处理器编写完成，特别注意的就是在使用我们的注解处理器的时候

要使用

   annotationProcessor project(":lib-apt")

而不是

   implementation project(":lib-apt")

下面贴出完整代码

package com.bubble.lib_apt;

import com.bubble.lib_annotation.Bubble;
import com.bubble.lib_annotation.IBubble;
import com.google.auto.service.AutoService;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.CodeBlock;
import com.squareup.javapoet.FieldSpec;
import com.squareup.javapoet.JavaFile;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.ParameterizedTypeName;
import com.squareup.javapoet.TypeSpec;

import java.io.IOException;
import java.util.ArrayList;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import javax.annotation.processing.AbstractProcessor;
import javax.annotation.processing.Filer;
import javax.annotation.processing.Messager;
import javax.annotation.processing.ProcessingEnvironment;
import javax.annotation.processing.Processor;
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.Modifier;
import javax.lang.model.element.TypeElement;
import javax.lang.model.util.Elements;
import javax.tools.Diagnostic;

@AutoService(Processor.class)
@SupportedAnnotationTypes("com.bubbble.lib_annotation.Bubble")
@SupportedSourceVersion(SourceVersion.RELEASE_8)
public class BubbleProcessor extends AbstractProcessor {
    private Filer mFiler;
    private Messager mMessager;
    private Elements mElements;

    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        mFiler = processingEnvironment.getFiler();
        mMessager = processingEnvironment.getMessager();
        mElements = processingEnvironment.getElementUtils();
        mMessager.printMessage(Diagnostic.Kind.WARNING, "初始化");

        Map<String, String> options = processingEnvironment.getOptions();
        mMessager.printMessage(Diagnostic.Kind.WARNING, "BUBBLE:" + options.get("BUBBLE"));
    }

    @Override
    public Set<String> getSupportedAnnotationTypes() {
        Set<String> set = new LinkedHashSet<>();
        set.add(Bubble.class.getCanonicalName());
        mMessager.printMessage(Diagnostic.Kind.WARNING, Bubble.class.getCanonicalName());
        return set;
    }

    @Override
    public Set<String> getSupportedOptions() {
        Set<String> set = new LinkedHashSet<>();
        set.add("BUBBLE");
        return set;
    }

    @Override
    public SourceVersion getSupportedSourceVersion() {
        return SourceVersion.latestSupported();
    }

    @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        mMessager.printMessage(Diagnostic.Kind.WARNING, "================================================");


        List<TypeElement> bubbles = new ArrayList<>();
        Set<? extends Element> rootElements = roundEnvironment.getRootElements();
        for (Element element : rootElements) {
            // 不是类 跳过
            if (!(element instanceof TypeElement)) {
                continue;
            }
            TypeElement typeElement = (TypeElement) element;
            // 寻找到带有Bubble 注解的类
            Bubble bubble = typeElement.getAnnotation(Bubble.class);
            mMessager.printMessage(Diagnostic.Kind.WARNING, bubble + "      " + (bubble == null));
            if (bubble == null) {
                continue;
            }
            mMessager.printMessage(Diagnostic.Kind.WARNING, "找到的类**********   " + (typeElement.getInterfaces().contains(IBubble.class)));
            mMessager.printMessage(Diagnostic.Kind.WARNING, typeElement.getQualifiedName());
            // 添加到集合
            bubbles.add(typeElement);
        }

        for (TypeElement bubble : bubbles) {
            mMessager.printMessage(Diagnostic.Kind.WARNING, "\n\n\n找到的所有类**********:   " + bubble);
        }
        if (!bubbles.isEmpty()) {

            ClassName list = ClassName.get("java.util", "List");
            ClassName arrayList = ClassName.get("java.util", "ArrayList");
            ParameterizedTypeName typeName = ParameterizedTypeName.get(list, ClassName.get(IBubble.class));
            // 创建一个IBubble 的List集合
            FieldSpec fieldSpec = FieldSpec.builder(typeName, "mList", Modifier.PRIVATE)
                    .initializer("new $T()", arrayList)
                    .build();
            // 创建个Field
            FieldSpec str = FieldSpec.builder(String.class, "str", Modifier.PRIVATE)
                    .build();

            // 把我们获取到的类 在我们要生成的java 类中new出来
            CodeBlock.Builder codeblock = CodeBlock.builder();
            for (TypeElement bubble : bubbles) {
                codeblock.addStatement("$N.add(new $T())", fieldSpec.name, ClassName.get(bubble));
            }
            // 创建方法
            MethodSpec init = MethodSpec.constructorBuilder()
                    .addModifiers(Modifier.PUBLIC)
                    .addStatement("str=$S", "555555")
                    .addCode(codeblock.build())
                    .build();
            // 创建一个方法 获取到我们的集合
            MethodSpec getList = MethodSpec.methodBuilder("getList")
                    .addModifiers(Modifier.PUBLIC)
                    .returns(typeName)
                    .addStatement("return $N", fieldSpec.name)
                    .build();
            MethodSpec setList = MethodSpec.methodBuilder("setList")
                    .addParameter(ClassName.get(IBubble.class), "item")
                    .addModifiers(Modifier.PUBLIC)
                    .addStatement("$N.add(item)", fieldSpec.name)
                    .build();

            // 创建一个类 叫 BubbleClass
            TypeSpec typeSpec = TypeSpec
                    .classBuilder("BubbleClass")
                    .addModifiers(Modifier.PUBLIC)
                    .addMethod(setList)
                    .addMethod(getList)
                    .addMethod(init)
                    .addField(fieldSpec)
                    .addField(str)
                    .build();

            // 创建java文件 并写入
            JavaFile file = JavaFile.builder("com.bubble.apt", typeSpec).build();
            try {
                file.writeTo(mFiler);
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return true;
    }
}

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

Android-Apt 注解处理器（二）

一、创建annotation 注解module ，注意：该module是java modle不是Android module

1.build.gradle配置

2.编写我们自己的注解

二、创建annotationProcessor module 该module也是java module

1.build.gradle配置

2.AbstractProcessor 介绍

3.创建一个BubbleProcessor 继承AbstractProcessor，在init里面获取到processingEnvironment提供的各种工具

4、实现getSupportedAnnotationTypes 方法返回我们要处理的注解

5.实现SourceVersion getSupportedSourceVersion()

6.实现Set<String> getSupportedOptions()

6.下面是最重要的方法，process

三、使用，创建app module

到此 我们的注解处理器编写完成，特别注意的就是在使用我们的注解处理器的时候

下面贴出完整代码

推荐阅读更多精彩内容

到此我们的注解处理器编写完成，特别注意的就是在使用我们的注解处理器的时候