APT实战,实现ButterKnife

APT

APT全称是Annotation Processing Tool即注解处理器,在ButterKnife中,我们使用ButterKnife提供的@BindView注解绑定一个控件后,ButterKnife会通过APT注解处理器,翻译解析这个注解,最后生成一个文件,达到无需手动findView的目的。下面来尝试一下如何是实现ButterKnife的效果。

ButterKnife使用demo

一 提供注解

新建一个Java Library用于提供注解

注解lib

BindView

@Target(ElementType.FIELD)
@Retention(RetentionPolicy.SOURCE)
public @interface BindView {
    //可以得到@BindView(R.id.btn)中间括号里面的内容
    int value();
}

二 处理注解

有了注解之后,就可以通过APT技术,获取注解的内容,新建一个Java Library用于处理注解。

注解处理lib

build.gradle

添加实现注解处理器依赖

     // 依赖注解提供库
     api project(path: ':annotaionlib')
    //添加APT依赖
    annotationProcessor 'com.google.auto.service:auto-service:1.0-rc6'
    compileOnly 'com.google.auto.service:auto-service:1.0-rc6'

AnnotationsCompiler.class

@AutoService(Processor.class)
public class AnnotationsCompiler extends AbstractProcessor {
    @Override
    public boolean process(Set set, RoundEnvironment roundEnvironment) {
        return false;
    }
}

实现注解处理的准备工作有如下

  1. 添加@AutoService(Processor.class) 和继承AbstractProcessor
  2. 返回支持的java版本
  3. 返回需要处理的注解
  4. 生成文件的对象
@AutoService(Processor.class)
public class AnnotationsCompiler extends AbstractProcessor {
    
    // 返回支持的java版本,这里是最新版本
    @Override
    public SourceVersion getSupportedSourceVersion() {
        return SourceVersion.latestSupported();
    }
    
     //返回需要处理的注解,这里只处理BindView注解
    @Override
    public Set getSupportedAnnotationTypes() {
        HashSet types = new HashSet<>();
        types.add(BindView.class.getCanonicalName());
        return types;
    }


    
    Filer filer;
    private Messager messager;
     //初始化工程,生成文件对象,后面生成文件的时候会用到
    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        filer = processingEnvironment.getFiler();
        messager = processingEnvironment.getMessager();
    }

    @Override
    public boolean process(Set set, RoundEnvironment roundEnvironment) {
        return false;
    }
}

准备好上面的步骤后,可以实现process方法了。

在实现process方法之前,回想一下如果我们不使用APT工具,那么该怎么实现该功能?APT的作用就是按照制定的规则实现文件,要制定规则,必须清楚要实现的文件应该是怎么样?

APT处理后,需要实现的文件

public class MainActivity extends AppCompatActivity {

    @BindView(R.id.btn)
    Button btn;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        ButterKnife.bind(this);
        new MainActivity_ViewBinding().bind(this);
        btn.setText("Rc在努力");
    }
}

class MainActivity_ViewBinding {
    public void bind(MainActivity target) {
        target.btn= (Button) target.findViewById(R.id.btn);
    }
}

上面是不使用APT,而是用代码实现APT的功能,现在只需把上面的功能改为APT自动实现就行了,有了思路就好解决。

    /**
     * 注解处理
     */
    @Override
    public boolean process(Set set, RoundEnvironment roundEnvironment) {
        //获取所有标记BindView注解的元素
        Set elementsAnnotatedWith
                = roundEnvironment.getElementsAnnotatedWith(BindView.class);

        //建立一个key value  key存的是标记了BindView的Activity, value是BindView注解集合
        Map> map = new HashMap<>();

        //遍历添加
        for (Element element : elementsAnnotatedWith) {
            VariableElement variableElement = (VariableElement) element;
            //取到activity的名字
            String activityName = variableElement.getEnclosingElement().getSimpleName().toString();
            List variableElements = map.get(activityName);
            if (variableElements == null) {
                variableElements = new ArrayList();
                map.put(activityName, variableElements);
            }
            variableElements.add(variableElement);

        }

        //有了注解之后,就可以生成对应的文件了
        if (map.size() > 0) {
            Writer writer = null;
            Iterator iterator = map.keySet().iterator();
            while(iterator.hasNext()) {
                String activityName = iterator.next();
                List variableElements = map.get(activityName);

                //得到包名
                TypeElement typeElement = (TypeElement)variableElements.get(0).getEnclosingElement();
                String packageName = processingEnv.getElementUtils().getPackageOf(typeElement).toString();

                //写入文件
                try{
                    JavaFileObject sourceFile = filer.createSourceFile(packageName + "." + activityName + "_ViewBinding");
                    writer = sourceFile.openWriter();
                    //package com.example.butterknife_framework_demo;
                    writer.write("package " + packageName + ";\n");
                    //import com.example.butterknife_framework_demo.IBinder;
                    //writer.write("import " + packageName + ".IBinder;\n");
                    writer.write("import " + "com.loveqrc.butterknife_api" + ".IBinder;\n");
                    //public class MainActivity_ViewBinding implements IBinder {
                    writer.write("public class " + activityName + "_ViewBinding implements IBinder<" + packageName + "." + activityName + ">{\n");
                    //@Override
                    writer.write("@Override\n");
                    //public void bind(com.example.butterknife_framework_demo.MainActivity target) {
                    writer.write("public void bind(" + packageName + "." + activityName + " target) {\n");
                    //target.textView = (android.widget.TextView) target.findViewById(2131165359);
                    for (VariableElement variableElement: variableElements) {

                        //得到名字
                        String variableName = variableElement.getSimpleName().toString();
                        //得到ID
                        int id = variableElement.getAnnotation(BindView.class).value();
                        //得到类型
                        TypeMirror typeMirror = variableElement.asType();
                        messager.printMessage(Diagnostic.Kind.NOTE, "HHHA:1.1.1====>" + variableName + " " + id);
                        //target.textView = (android.widget.TextView) target.findViewById(2131165359);
                        writer.write("\ttarget." + variableName + "=(" + typeMirror + ")target.findViewById(" + id + ");\n");
                    }
                    writer.write("\n}\n}");

                } catch (Exception e){
                    e.printStackTrace();
                } finally {
                    if(writer != null) {
                        try {
                            writer.close();
                        } catch (IOException e) {
                            e.printStackTrace();
                        }
                    }
                }


            }
        }

        return false;
    }

实例化生成后的文件

新建一个Android Libray用于实例化生成的文件。

IBinder

public interface IBinder {
    void bind(T target);
}

ButterKnife

public class ButterKnife {
    public static void bind(Activity activity) {
        String name = activity.getClass().getName() + "_ViewBinding";
        try {
            Class aClass = Class.forName(name);
            IBinder iBinder = (IBinder)aClass.newInstance();
            iBinder.bind(activity);
        }catch (Exception e) {
            e.printStackTrace();
        }
    }
}

IBinder用于给所有APT生成的类实现,这样做的好处是类似于面向接口编程,ButterKnifebind用于的绑定生成文件和Activity之间的关系。这里用到了一次反射。

源码

你可能感兴趣的:(APT实战,实现ButterKnife)