ButterKnife实现之Android注解处理器使用教程
1、新建一个注解
1.1、编译时注解
创建注解所需的元注解@Retention包含3个不同的值，RetentionPolicy.SOURCE、RetentionPolicy.CLASS、RetentionPolicy.RUNTIME。这3个值代表注解不同的保留策略。
使用RetentionPolicy.RUNTIME的注解为运行时注解，能在程序运行时通过反射获取注解的信息并进行逻辑处理；使用RetentionPolicy.CLASS的注解为编译时注解，能在程序编译时进行预处理操作，比如生成一些辅助代码；使用RetentionPolicy.SOURCE的注解能做一些检查性的操作，比如@Override和@SuppressWarning。
1.2、新建注解
编译时注解能够帮助我们生成辅助代码，能够满足在编译时获取注解信息生成带有findViewById的代码。所以我们新建一个编译时注解。新建注解前，我们新建一个名为annotation的Java Library类型的Module。然后在这个Module新建这个注解，命名为BindView，代码如下：
@Retention(RetentionPolicy.CLASS)
@Target(ElementType.FIELD)
public @interface BindView {
int value();
}
2、新建注解处理器
注解处理器是处理注解的类，处理编译时注解时我们需要编写一个注解处理器。注解处理器类需要继承AbstractProcessor类。本节我们来学习编写注解处理器，跟上一节一样我们再次新建一个Java Libary的module，这个module命名为processor，并依赖包含注解的annotation Module：在processor module的build.gradle添加如下代码：
dependencies{
implementation project(':annotation')
}
接着，在这个module中我们新建一个注解处理器-MainProcessor，它继承AbstractProcessor并实现AbstractProcess的4大方法，我们来学习这4大方法。
2.1、AbstractProcessor的4大方法
/**
* 注解处理器MainProcessor
*/
public class MainProcessor extends AbstractProcessor {
@Override
public synchronized void init(ProcessingEnvironment processingEnv) {
super.init(processingEnv);
}
@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
return false;
}
@Override
public Set<String> getSupportedAnnotationTypes() {
return super.getSupportedAnnotationTypes();
}
@Override
public SourceVersion getSupportedSourceVersion() {
return super.getSupportedSourceVersion();
}
继承AbstractProcessor需要重写上述代码段的4个方法，依次介绍它们的作用：
1、init方法：被注解处理工具调用，并输入ProcessingEnviroment参数。ProcessingEnviroment提供了很多有用的工具类，比如Elements、Types、Filer和Messager等。
2、process方法：相当于每个处理器的主函数main()，在这里编写扫描、评估和处理注解的代码以及生成Java文件。输入参数RoundEnviroment，可以让你查询包含特定注解的被注解元素。
3、getSupportedAnnotationTypes：这是必须指定的方法，指定这个注解处理器是注册给哪个注解的，注意，它的返回值是一个字符串的集合，包含该处理器想要处理的注解类型的合法全称。
4、getSupportedSourceVersion：用来指定你使用的Java版本，通常这里返回SourceVersion.latestSupported()。
可以将MainProcessor的getSupportAnnotationTypes方法和getSupportedSourceVersion方法更新成如下：
@Override
public Set<String> getSupportedAnnotationTypes() {
HashSet<String> typeSet = new HashSet<>();
typeSet.add(BindView.class.getCanonicalName());
return typeSet;
}
@Override
public SourceVersion getSupportedSourceVersion() {
return SourceVersion.latestSupported();
}
2.2、JavaPoet的使用
前面说到过，在程序编译时根据注解信息生成辅助文件，JavaPoet是一个可以生成Java代码的第三方框架，所以我们要利用它生成辅助文件。
1.添加JavaPoet依赖
implementation 'com.squareup:javapoet:1.7.0'
2.JavaPoet Api使用
1、生成方法
以ButterKnife的bind方法为例，初始化一个id为R.id.tv_hello的TextView，代码如下：
//这个MainActivity是个例子，实际上使用的是注解所对应的Activity
public void bind(MainActivity activity){
activity.tvHello = (TextView)(((android.app.Activity)activity).findViewById(R.id.tv_hello));
}
使用JavaPoet生成这个方法：
MethodSpec.Builder bindMethodBuilder = MethodSpec.methodBuilder("bind") //方法名为bind
.addModifiers(Modifier.PUBLIC) //方法修饰符：Public
.addParameter(MainActivity,"activity")//方法的参数：如MainActivity activity
.returns(void.class); //返回值：void
String code = String.format("activity.%s=(%s)(((android.app.Activity)activity).findViewById(%s));\n","tvHello","android.widget.TextView",R.id.tv_hello);
bindMethodBuilder.addCode(code);
2、生成类
以生成MainActivity的辅助类MainActivity_ViewBinding为例，类的内容：
public class MainActivity_ViewBinding{
//bind方法就是上面生成的方法
public void bind(MainActivity activity){
tvTest = (android.widget.TextView) ((android.app.Activity)activity).findViewById(R.id.tv_test);
}
}
使用JavaPoet生成该类：
TypeSpec.classBuilder("MainActivity_ViewBinding")
.addModifiers(Modifier.PUBLIC)
.addMethod(bindMethodBuilder.build())
.build();
这样的话，完整的类就使用JavaPoet生成出来了。还有更多的JavaPoet的用法，推荐看这篇文章：基于JavaPoet自动生成java代码文件
2.3、编写process方法
接下来就是注解处理器的核心部分了，我们通过process方法实现注解解析，生成源码的功能。process方法中需要用到ProcessingEnviroment参数，所以我们先处理init方法，保存变量：
public class MainProcessor extends AbstractProcessor {
private Elements elementUtils;
private ProcessingEnvironment processingEnvironment;
@Override
public synchronized void init(ProcessingEnvironment processingEnv) {
super.init(processingEnv);
elementUtils = processingEnv.getElementUtils();
processingEnvironment = processingEnv;
}
}
process方法的逻辑主要是解析注解和生成代码，我就直接上代码了：
@Override
public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
/*
*生成的代码
* 类：MainActivity_ViewBinding,包名：com.wei.annotation_processor_demo
* 内容：
* public class MainActivity_ViewBinding{
* public void bind(MainActivity activity){
* tvTest = (TextView) ((Activity)activity).findViewById(R.id.tv_test);
* }
* }
*/
Set<? extends Element> elementsAnnotatedWith = roundEnv.getElementsAnnotatedWith(BindView.class);
Map<VariableElement, Integer> elementMap = new HashMap<>();
for (Element element : elementsAnnotatedWith) {
//获取被注解的字段
VariableElement variableElement = (VariableElement) element;
//获取被注解的字段的类
TypeElement enclosingElement = (TypeElement) element.getEnclosingElement();
String className = enclosingElement.getSimpleName().toString();
//获取注解
BindView bindView = variableElement.getAnnotation(BindView.class);
int id = bindView.value();
//保存所有被注解的字段和注解的成员变量值，用于生成代码
elementMap.put(variableElement, id);
//获取被注解的字段所在类的包名
String packageName = elementUtils.getPackageOf(enclosingElement).getQualifiedName().toString();
//生成代码
TypeSpec typeSpec = generateCode(className, ClassName.bestGuess(enclosingElement.getQualifiedName().toString()), elementMap);
//生成javaFile
JavaFile javaFile = JavaFile.builder(packageName, typeSpec).build();
try {
//生成java代码
javaFile.writeTo(processingEnvironment.getFiler());
} catch (IOException e) {
e.printStackTrace();
}
}
return true;
}
private TypeSpec generateCode(String className, ClassName parameterClass, Map<VariableElement,Integer> elementMap){
//生成bind方法
MethodSpec.Builder bindMethodBuilder = MethodSpec.methodBuilder("bind")
.addModifiers(Modifier.PUBLIC)
.addParameter(parameterClass,"activity")
.returns(void.class);
for (Map.Entry<VariableElement, Integer> entry : elementMap.entrySet()) {
String fieldName = entry.getKey().getSimpleName().toString();
String fieldType = entry.getKey().asType().toString();
String code = String.format("activity.%s=(%s)(((android.app.Activity)activity).findViewById(%s));\n"
,fieldName,fieldType,String.valueOf(entry.getValue()));
processingEnvironment.getMessager().printMessage(Diagnostic.Kind.NOTE,"fieldName:"+fieldName+",fieldType:"+fieldType+",code:"+code);
bindMethodBuilder.addCode(code);
}
return TypeSpec.classBuilder(className+"_ViewBinding")
.addModifiers(Modifier.PUBLIC)
.addMethod(bindMethodBuilder.build())
.build();
}
2.4、注册注解处理器
为了能使用注解处理器，需要用一个服务文件来注册它。文件路径为：processor module的根目录/resources/META-INF.services/javax.annotation.processing.Processor。在javax.annotation.processing.Processor中添加内容：com.wei.processor.MainProcessor。这样就成功注册了注解处理器，同时需要注意2点：1.文件路径中的文件夹可能不存在，需要手动创建；2.文件内容是注解处理器的包名+类名，不要照抄我的。
AutoService
如果不想手动添加服务文件，就使用AutoService框架来生成服务文件。
使用步骤：
1、添加依赖
//google autoService
implementation 'com.google.auto.service:auto-service:1.0-rc4'
annotationProcessor "com.google.auto.service:auto-service:1.0-rc4"
2、使用
在注解处理器的类上添加@AutoService注解即可：
@AutoService(Processor.class)
public class MainProcessor extends AbstractProcessor {
//省略内容
//...
}
这样就实现了刚才我们手动创建服务文件同样的功能。
3、使用
注解处理器编写结束了，我们需要验证是否能够实现ButterKnife同样的效果。验证方法：我们在app module中添加annotation、processor两个库的依赖，在MainActivity中使用BindView注解，看看app module根目录/build/ap_generated_sources/debug/out/有无MainActivity_ViewBinding文件生成。
添加依赖：
implementation project(":annotation")
// implementation project(":processor")
annotationProcessor project(":processor")
使用annotationProcessor代替implementation有以下好处：
1、annotationProcessor引用的库只会在编译期间被依赖使用，不会打包进入apk，因为注册处理器是在编译期间使用的，打包进入apk会占用空间
2、为注解处理器生成的代码设置好路径，以便Android Studio能找到它
使用BindView注解
public class MainActivity extends AppCompatActivity {
@BindView(R.id.tv_hello)
TextView tvHello;
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
}
查看生成文件
使用
生成ViewBinding类后，可以通过反射执行该类bind方法，实现findViewById逻辑：
private void bind() throws ClassNotFoundException, NoSuchMethodException, IllegalAccessException, InstantiationException, InvocationTargetException {
Class<?> clazz = Class.forName(getClass().getName() + "_ViewBinding");
System.out.println(getClass().getName());
Method bind = clazz.getDeclaredMethod("bind", getClass());
bind.invoke(clazz.newInstance(), this);
}
调用这个方法也就实现了findViewById逻辑，最后：
@BindView(R.id.tv_hello)
TextView tvHello;
@Override
protected void onCreate(Bundle savedInstanceState) {
super.onCreate(savedInstanceState);
setContentView(R.layout.activity_main);
try {
bind();
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (NoSuchMethodException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
} catch (InstantiationException e) {
e.printStackTrace();
} catch (InvocationTargetException e) {
e.printStackTrace();
}
tvHello.setText("我成功了！");
}
4、参考文章
感谢一下文章提供的教程，万分感激：
1、Android APT技术学习
2、基于JavaPoet自动生成java代码文件
3、深入理解编译注解（二）annotationProcessor与android-apt
4099
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
