APT的使用

1.声明注解

创建一个名字为apt-annotation的javalib,并在lib里创建一个BindView注解

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

2.继承AbstractProcessor实现自己的注解处理类

创建另外一个名为apt-processor的javalib,在lib中创建MyProcessor 来根据注解动态生成代码

package com.example.apt_processor;

import com.example.apt_annotation.BindView;
import com.google.auto.service.AutoService;
import com.squareup.javapoet.ClassName;
import com.squareup.javapoet.JavaFile;
import com.squareup.javapoet.MethodSpec;
import com.squareup.javapoet.TypeSpec;

import java.io.IOException;
import java.lang.annotation.ElementType;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
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.SupportedSourceVersion;
import javax.lang.model.SourceVersion;
import javax.lang.model.element.Element;
import javax.lang.model.element.Modifier;
import javax.lang.model.element.Name;
import javax.lang.model.element.PackageElement;
import javax.lang.model.element.TypeElement;
import javax.lang.model.element.VariableElement;
import javax.lang.model.util.Elements;
import javax.swing.JList;
import javax.tools.Diagnostic;
//必须写的,用来声明这个类是注解处理类
@AutoService(Processor.class)
@SupportedSourceVersion(SourceVersion.RELEASE_8)
//必须在javalib中创建否则无法继承AbstractProcessor类
public class MyProcessor extends AbstractProcessor {

    private Elements elementUtils;
    private Map<TypeElement, List<VariableElement>> map = new HashMap<>();
    private Messager messager;
    private Filer filer;


    //返回要处理的注解 可以添加多个
    @Override
    public Set<String> getSupportedAnnotationTypes() {
        Set<String> set = new HashSet<>();
        set.add(BindView.class.getCanonicalName());
        return set;
    }

    @Override
    public synchronized void init(ProcessingEnvironment processingEnv) {
        super.init(processingEnv);
        processingEnv.getMessager().printMessage(Diagnostic.Kind.NOTE,"init()");
        //日志打印工具
        messager = processingEnv.getMessager();
        elementUtils = processingEnv.getElementUtils();
        //用于生成类文件
        filer = processingEnv.getFiler();

    }
    
    @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        //拿到所有添加BindView注解的成员变量
        Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(BindView.class);
        messager.printMessage(Diagnostic.Kind.NOTE,"111111111111111");
        //如果set的长度为0则不处理返回false,这里这样处理是因为process方法并不是只调用一次,所以要进行过滤
        if (set.size()<=0){
            return false;
        }
        
        //for循环把每个activity中的控件放入统一list中
        for (Element element : elements) {
            messager.printMessage(Diagnostic.Kind.NOTE,element.getKind().toString());
            PackageElement packageElement = elementUtils.getPackageOf(element);
            messager.printMessage(Diagnostic.Kind.NOTE," getQualifiedName = "+packageElement.getQualifiedName().toString());
            VariableElement variableElement = (VariableElement) element;
            TypeElement  typeElement = (TypeElement) variableElement.getEnclosingElement();
            List<VariableElement> list = map.get(typeElement);
            if (list == null){
                list = new ArrayList<>();
                map.put(typeElement,list);
            }
            Name simpleName = variableElement.getSimpleName();
            BindView annotation = variableElement.getAnnotation(BindView.class);
            messager.printMessage(Diagnostic.Kind.NOTE," simpleName = "+simpleName.toString());
            list.add(variableElement);
        }


        //for循环生成每个activity对应的view绑定工具类
        for (TypeElement typeElement : map.keySet()) {
            String className = typeElement.getSimpleName().toString();
            String packageName = elementUtils.getPackageOf(typeElement).getQualifiedName().toString();
            String finalClassName = className+"Binding";
            messager.printMessage(Diagnostic.Kind.NOTE,"类名 = "+finalClassName);

            ClassName classType = ClassName.bestGuess(className);

            String paramName = "activity";
            MethodSpec.Builder builder = MethodSpec.methodBuilder("bind")
                    .addModifiers(Modifier.PUBLIC,Modifier.STATIC)
                    .returns(void.class)
                    .addParameter(classType,paramName);


            List<VariableElement> variableElements = map.get(typeElement);
            for (int i = 0; i < variableElements.size(); i++) {
                VariableElement variableElement = variableElements.get(i);
                String type = variableElement.asType().toString();
                int viewId = variableElement.getAnnotation(BindView.class).value();
                String name = variableElement.getSimpleName().toString();
                builder.addStatement("$L.$L = ($L)$L.findViewById($L)",paramName,name,type,paramName,viewId);
            }
            MethodSpec methodSpec = builder.build();

            TypeSpec typeSpec = TypeSpec.classBuilder(finalClassName)
                    .addModifiers(Modifier.PUBLIC)
                    .addMethod(methodSpec)
                    .build();
            JavaFile javaFile = JavaFile.builder(packageName,typeSpec).build();
            try {
                javaFile.writeTo(filer);
                messager.printMessage(Diagnostic.Kind.NOTE,"3333");
            } catch (IOException e) {
                e.printStackTrace();
            }

        }
        return true;
    }
}
plugins {
    id 'java-library'
}

java {
    sourceCompatibility = JavaVersion.VERSION_1_8
    targetCompatibility = JavaVersion.VERSION_1_8
}


dependencies{
    //自动注册,动态生成 META-INF/...文件
    implementation 'com.google.auto.service:auto-service:1.0-rc6'
    annotationProcessor 'com.google.auto.service:auto-service:1.0-rc6'
    //依赖apt-annotation
    implementation project(':apt-annotation')
    //帮助生成代码的工具
    implementation 'com.squareup:javapoet:1.13.0'
}

3.app中使用

package com.example.apttest;

import androidx.appcompat.app.AppCompatActivity;

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

import com.example.apt_annotation.BindView;

public class MainActivity extends AppCompatActivity {

    @BindView(R.id.contentTv)
    public TextView textView;
    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        MainActivityBinding.bind(this);
        textView.setText("77777777777");
    }
}
dependencies {

    implementation 'androidx.appcompat:appcompat:1.5.1'
    implementation 'com.google.android.material:material:1.4.0'
    implementation 'androidx.constraintlayout:constraintlayout:2.0.4'
    testImplementation 'junit:junit:4.13.2'
    androidTestImplementation 'androidx.test.ext:junit:1.1.3'
    androidTestImplementation 'androidx.test.espresso:espresso-core:3.4.0'

    implementation project(':apt-annotation')
    annotationProcessor project(':apt-processor')

}

生成的工具类代码

package com.example.apttest;

public class MainActivityBinding {
  public static void bind(MainActivity activity) {
    activity.textView = (android.widget.TextView)activity.findViewById(2131230834);
  }
}

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

推荐阅读更多精彩内容