Android 利用apt生成代码,实现butterKnife控件查找功能

了解了butterknife的实现原理后,研究了一下apt技术,接着自己查阅相关资料,撸了一遍apt的实现过程,因为看的资料比较老旧,实现过程颇为曲折,所以把自己的实现过程记录一下,方便新学习的小伙伴绕开这些坑。

ATP(Annotation processing tool)

Annotation processing tool也就是注解处理器了,原理是根据注解在代码编译的时候去生成相应的功能代码文件,打包的时候会跟着其他的源码一起打包成class文件,这样就避免了那些功能在运行时全部用反射去实现,从而提高了app的性能。

首先新建一个工程,然后新建一个java library module,取名binder_annotation,这里我们专门用来存放Annotation文件,接着自定义一个Annotation,取名BindView:

@Retention(RetentionPolicy.CLASS)
@Target(ElementType.FIELD)
public @interface BindView {
    int value();
}

接着新建另一个java library module,取名binder_compiler,我们在这里做注解处理的工作,和生成相应的java文件的操作,
在这个module的gradle文件里添加如下配置:

plugins {
    id 'java-library'
}

java {
    sourceCompatibility = JavaVersion.VERSION_1_7
    targetCompatibility = JavaVersion.VERSION_1_7
}
dependencies {
    implementation fileTree(dir: 'libs', include: ['*.jar'])
    //AutoService 主要的作用是注解 processor 类,并对其生成 META-INF 的配置信息。
    implementation 'com.google.auto.service:auto-service:1.0-rc6'
   //解决gradle的版本bug,不添加会导致我们的process类不被调用
    annotationProcessor 'com.google.auto.service:auto-service:1.0-rc6'
    //JavaPoet 这个库的主要作用就是帮助我们通过类调用的形式来生成代码。
    implementation 'com.squareup:javapoet:1.10.0'
    implementation project(':binder_annotation')
}

注意:Android Plugin for Gradle: >3.3.2的时候要添加 :

annotationProcessor 'com.google.auto.service:auto-service:1.0-rc6' 这行依赖,这是gradle的一个版本bug,高版本的gradle不会去调用我们编写好的process类,我在这里就陷进去好久。

app module下的gradle文件做如下配置:

dependencies {
    implementation fileTree(dir: 'libs', include: ['*.jar'])
    implementation 'androidx.appcompat:appcompat:1.3.0'
    implementation 'com.google.android.material:material:1.3.0'
    implementation 'androidx.constraintlayout:constraintlayout:2.0.4'
    testImplementation 'junit:junit:4.+'
    androidTestImplementation 'androidx.test.ext:junit:1.1.2'
    androidTestImplementation 'androidx.test.espresso:espresso-core:3.3.0'

    annotationProcessor project(':binder_compiler')
    implementation project(':binder_annotation')
}

接着新建BinderProcessor类,让它继承AbstractProcessor,并加上@AutoService(Processor.class)注解,这样它才会在代码编译期被执行:

@AutoService(Processor.class)
public class BinderProcessor extends AbstractProcessor {
    private Elements mElementUtils; ///处理Element的工具类
    private HashMap<String,BinderClassCreator>  mCreatorMap = new HashMap<>();//构造器工具的缓存map
}

重写相关方法:

 //初始化
  @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        //处理Element的工具类,用于获取程序的元素,例如包、类、方法。
        mElementUtils = processingEnvironment.getElementUtils();
    }
  //使用最新的版本
    @Override
    public SourceVersion getSupportedSourceVersion() {
        return SourceVersion.latestSupported();
    }
  //支持的注解类名集合,这里我们只做BindView的注解处理
    @Override
    public Set<String> getSupportedAnnotationTypes() {
        HashSet<String> supportTypes = new LinkedHashSet<>();
        supportTypes.add(BindView.class.getCanonicalName());
        return supportTypes;
    }

重点需要处理的方法是process(),相关注释在代码里,算是比较详细了,多看几遍应该看得懂:

@Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        //扫描整个工程里被BindView注解过的元素,会根据activity名来生成相应的工具类BinderClassCreator
        //BinderClassCreator里包含了生成相应的activity的_ViewBinding类,里面有做了findViewById的事情
        Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(BindView.class);
        for(Element element:elements){
            VariableElement variableElement = (VariableElement) element;//强转
            //返回此元素直接封装(非严格意义上)的元素。
            //类或接口被认为用于封装它直接声明的字段、方法、构造方法和成员类型
            //这里就是获取封装属性元素的类元素
            TypeElement classElement = (TypeElement) variableElement.getEnclosingElement();
            //获取简单类名
            String fullClassName = classElement.getQualifiedName().toString();
            //先在map缓存里取BinderClassCreator
            BinderClassCreator creator = mCreatorMap.get(fullClassName);
            if(creator == null){
                creator = new BinderClassCreator(mElementUtils.getPackageOf(classElement),classElement);
                //保存在map里
                mCreatorMap.put(fullClassName,creator);
            }
            //获取元素注解信息
            BindView bindAnnotation = variableElement.getAnnotation(BindView.class);
            int id = bindAnnotation.value();
            creator.putElement(id,variableElement);
        }
        //通过javaPoet构建生成java文件
        for(String key:mCreatorMap.keySet()){
            BinderClassCreator classCreator = mCreatorMap.get(key);
            JavaFile javaFile = JavaFile.builder(classCreator.getmPackageName(), classCreator.generateJavaCode()).build();
            try {
                javaFile.writeTo(processingEnv.getFiler());
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return false;
    }

这里用到一个BinderClassCreator类,用来帮助构建相应activity__ViewBinding类的工具:

/**
 * @author: lookey
 * @date: 2021/5/25
 * 用来生成_BinderView类的工具类
 */
public class BinderClassCreator {
    public static final String ParamName = "view";
    private TypeElement mTypeElement;//类元素
    private String mPackageName;
    private String mBinderClassName;
    private HashMap<Integer, VariableElement> mVariableElement = new HashMap<>();
    public BinderClassCreator(PackageElement mPackageElement, TypeElement mTypeElement) {
        this.mTypeElement = mTypeElement;
        this.mPackageName = mPackageElement.getQualifiedName().toString();
        this.mBinderClassName = mTypeElement.getSimpleName().toString()+"_ViewBinding";
    }
    public void putElement(int id,VariableElement variableElement){
        mVariableElement.put(id,variableElement);
    }

    public String getmPackageName() {
        return mPackageName;
    }
    //生成java类,及相应的方法
    public TypeSpec generateJavaCode(){
        return TypeSpec.classBuilder(mBinderClassName)
                .addModifiers(Modifier.PUBLIC) //public修饰
                .addMethod(generateMethod()) //添加方法
                .build();
    }
    private MethodSpec generateMethod(){
        //获取类名
        ClassName className = ClassName.bestGuess(mTypeElement.getQualifiedName().toString());
        return MethodSpec.methodBuilder("bindView")
                .addModifiers(Modifier.PUBLIC)
                .returns(void.class)
                .addParameter(className,ParamName)
                .addCode(generateMethodCode())
                .build();
    }
    private String generateMethodCode() {
        StringBuilder code = new StringBuilder();
        for (int id : mVariableElement.keySet()) {
            VariableElement variableElement = mVariableElement.get(id);
            //使用注解的属性的名称
            String name = variableElement.getSimpleName().toString();
            //使用注解的属性的类型
            String type = variableElement.asType().toString();
            //view.name = (type)view.findViewById(id)
            String findViewCode = ParamName + "." + name + "=(" + type + ")" + ParamName +
                    ".findViewById(" + id + ");\n";
            code.append(findViewCode);

        }
        return code.toString();
    }
}

再写一个工具类BinderViewTools 让我们的activity调用,类似ButterKnife.bind(),通篇下来也就在这里用到了反射:

/**
 * @author: lookey
 * @date: 2021/5/25
 */
public class BinderViewTools {
    public static void bind(Activity activity){
        Class aClass = activity.getClass();
        try {
            Class<?> bindClass = Class.forName(aClass.getCanonicalName() + "_ViewBinding");//找到生成的相应的bind类
            Method method = bindClass.getMethod("bindView", aClass);
            method.invoke(bindClass.newInstance(),activity);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

最后在activity使用一下,使用步骤类似butterknife:

package com.trendlab.aptex;

import androidx.appcompat.app.AppCompatActivity;

import android.os.Bundle;
import android.widget.TextView;

import com.trendlab.binder_annotation.BindView;

public class MainActivity extends AppCompatActivity {

    @BindView(R.id.tv111)
    public TextView tv111;
    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        BinderViewTools.bind(this); //运行时会去找到MainActivity_ViewBinding类,然后实例化一个对象,再调用findview()方法
        tv111.setText("hello binder");
    }
}

运行编译一下,一切正常的话会在app module下生成这个文件:


image.png

模拟器运行成功:


image.png

整体做完还是比较清晰的,重点是对javaPoet的熟练使用,和生成java文件的process()方法的构思,调试过程中遇到不能生成java文件,或者提示写入重复报错的情况可以尝试invalidate caches/restart 重启studio,如果调试还是不成功可以参考完整项目:https://github.com/gi13371/APTdemo
希望这篇文章对你有帮助!

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,761评论 5 460
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,953评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,998评论 0 320
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,248评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,130评论 4 356
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,145评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,550评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,236评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,510评论 1 291
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,601评论 2 310
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,376评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,247评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,613评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,911评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,191评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,532评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,739评论 2 335

推荐阅读更多精彩内容