Android模块开发之APT技术

APT,就是Annotation Processing Tool 的简称,就是可以在代码编译期间对注解进行处理,并且生成Java文件,减少手动的代码输入。注解我们平时用到的比较多的可能会是运行时注解,比如大名鼎鼎的retrofit就是用运行时注解,通过动态代理来生成网络请求。编译时注解平时开发中可能会涉及的比较少,但并不是说不常用,比如我们经常用的轮子Dagger2, ButterKnife, EventBus3 都在用,所以要紧跟潮流来看看APT技术的来龙去脉。

关于注解的大概分类,不清楚的同学可以参考我前面的博客,反射注解与动态代理综合使用。

1.实现目标

我们还是通过一个栗子来进行分析。平时一般启动Activity都是这样通过startActivity或者startActivityForResult等等balabala。今天我们通过给Activity添加一个注解@RouteAnnotation(name = "RouteName_ActivityTwo"),然后通过注解来启动Activity,AnnocationRouter.getSingleton().navigat("RouteName_ActivityTwo")。

 

mContext.startActivity(intent)或者startActivityForResult

下面是演示的图片:

 

Android模块开发之APT技术_第1张图片

1.png

Android模块开发之APT技术_第2张图片

2.png

2.自定义注解

首先就是定义注解RouteAnnotation,作用对象就是类,作用范围就是编译时。然后就是接受一个参数,也就是Activity的别名。这个不清楚请自行参考我上面提到的博客。

 

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.CLASS)
public @interface RouteAnnotation {
    String name();
}

3.创建注解处理器

注解处理器一般会是一个项目比较底层的模块,因此我们创建一个Java Library,annotationprocess模块,要依赖前面的注解。自定义的处理器需要继承AbstractProcessor,需要自己实现process方法,一般我们会实现其中的4个方法:

 

public class AnnotationProcessor extends AbstractProcessor {
    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
    }

    @Override
    public SourceVersion getSupportedSourceVersion() {
    }

    @Override
    public Set getSupportedAnnotationTypes() {
    }

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

init()方法可以初始化拿到一些使用的工具,比如文件相关的辅助类 Filer;元素相关的辅助类Elements;日志相关的辅助类Messager;
getSupportedSourceVersion()方法返回 Java 版本;
getSupportedAnnotationTypes()方法返回要处理的注解的结合;
上面几个方法写法基本都是固定的,重头戏是process()方法。

在看具体的代码之前需要先说下Elements这个类,因为后面生成Java文件会经常用到。通过Element可以拿到特定的元素类型。

TypeElement :类或者接口类型
VariableElement:成员变量
ExecuteableElement:成员方法

下面看下我们这个栗子中AnnotationProcessor具体代码:

 

public class AnnotationProcessor extends AbstractProcessor {


    private Filer mFiler;

    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        mFiler = processingEnvironment.getFiler();
    }

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

    @Override
    public Set getSupportedAnnotationTypes() {
        LinkedHashSet types = new LinkedHashSet<>();
        types.add(RouteAnnotation.class.getCanonicalName());
        return types;
    }

    @Override
    public boolean process(Set set, RoundEnvironment roundEnvironment) {
        HashMap nameMap = new HashMap<>();

        Set annotationElements = roundEnvironment.getElementsAnnotatedWith(RouteAnnotation.class);

        for (Element element : annotationElements) {
            RouteAnnotation annotation = element.getAnnotation(RouteAnnotation.class);
            String name = annotation.name();
            nameMap.put(name, element.getSimpleName().toString());
            //nameMap.put(element.getSimpleName().toString(), name);//MainActiviy-RouteName_MainActivity
        }

        //generate Java File
        generateJavaFile(nameMap);

        return true;
    }

    private void generateJavaFile(Map nameMap) {
        //generate constructor
        MethodSpec.Builder constructorBuidler = MethodSpec.constructorBuilder()
                .addModifiers(Modifier.PUBLIC)
                .addStatement("routeMap = new $T<>()", HashMap.class);
        for (String key : nameMap.keySet()) {
            String name = nameMap.get(key);
            constructorBuidler.addStatement("routeMap.put(\"$N\", \"$N\")", key, name);
        }
        MethodSpec constructorName = constructorBuidler.build();

        //generate getActivityRouteName method
        MethodSpec routeName = MethodSpec.methodBuilder("getActivityName")
                .addModifiers(Modifier.PUBLIC)
                .returns(String.class)
                .addParameter(String.class, "routeName")
                .beginControlFlow("if (null != routeMap && !routeMap.isEmpty())")
                .addStatement("return (String)routeMap.get(routeName)")
                .endControlFlow()
                .addStatement("return \"\"")
                .build();

        //generate class
        TypeSpec typeSpec = TypeSpec.classBuilder("AnnotationRoute$Finder")
                .addModifiers(Modifier.PUBLIC)
                .addMethod(constructorName)
                .addMethod(routeName)
                .addSuperinterface(Provider.class)
                .addField(HashMap.class, "routeMap", Modifier.PRIVATE)
                .build();


        JavaFile javaFile = JavaFile.builder("com.example.juexingzhe.annotaioncompiletest", typeSpec).build();
        try {
            javaFile.writeTo(mFiler);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

下面分别看下每个方法:
1.init方法中主要是拿到File文件工具对象,后面生成Java文件需要用到;
2.getSupportedSourceVersion就是返回Java版本,这个写法基本固定;
3.getSupportedAnnotationTypes就是返回需要处理的注解,我们这个栗子中就是RouteAnnotation;

4.重点就是process方法,方法里面主要工作就是生成Java文件。我们具体看下步骤:

1.roundEnvironment.getElementsAnnotatedWith(RouteAnnotation.class)拿到所有RouteAnnotation注解标注的类
2.循环取出注解的name属性和被标注的类名并缓存,其实就是:
put("RouteName_ActivityTwo", "ActivityTwo");
3.通过javapoet库生成Java类,javapoet是square公司良心出品,让我们脱离手动凭借字符串来生成Java类的痛苦,可以通过各种姿势来生成Java类,这里不多做介绍,有需要的可以看官方文档,很详细。
4.最后通过JavaFile.builder("包名",TypeSpec)生成Java文件,包名可以随意取,最后生成的文件都是在主程序模块app.build.generated.source.apt目录下

最后生成的文件就是下面这样,对着生成的文件看上面构造类文件的过程会比较清晰。

 

public class AnnotationRoute$Finder implements Provider {
  private HashMap routeMap;

  public AnnotationRoute$Finder() {
    routeMap = new HashMap<>();
    routeMap.put("RouteName_ActivityTwo", "ActivityTwo");
    routeMap.put("RouteName_MainActivity", "MainActivity");
    routeMap.put("RouteName_ActivityThree", "ActivityThree");
  }

  public String getActivityName(String routeName) {
    if (null != routeMap && !routeMap.isEmpty()) {
      return (String)routeMap.get(routeName);
    }
    return "";
  }
}

4.注册注解处理器

编译的时候JVM怎么找到我们自定义的注解处理器?这个时候就要用到Java SPI机制,这个可以参考我们的上一篇博客Android模块开发之SPI。就是在annotationprocess模块的resources目录下新建META-INF/services,然后新建File,名称javax.annotation.processing.Processor,文件内容就是我们自定义注解处理器的全限定名com.example.AnnotationProcessor
谷歌官方也出品了一个开源库Auto-Service,通过注解@AutoService(Processor.class)可以省略上面配置的步骤,这个后面有机会我们再专门扒一扒。到这里我们就可以使用自定义的注解了。

5.使用自定义注解

在ActivityTwo上面添加注解
@RouteAnnotation(name = "RouteName_ActivityTwo"),然后在MainActivity中就可以通过name查找到ActivityTwo,navigat方法中其实也是封装了intent,然后通过startActivity启动。

 

AnnocationRouter.getSingleton().navigat("RouteName_ActivityTwo");

那么既然也是startActivity进行启动,为什么要自定义注解,然后通过navigat进行启动呢?设想一个场景,我们可以将所有Activity通过自定义注解设置别名name,然后外界就可以通过路径来启动Activity,比如构造url路径来启动。这方面也有很多的开源库这样做,比如阿里出品的路由框架ARouter等。

6.总结

APT技术其实就是自定义注解和注解处理器,在编译期间生成Java文件,类似于IOC控制反转,可以方便的进行解耦,在多模块开发时可以基于APT技术构造一套路由框架,去除startActivity等造成的类依赖,也可以通过url方式实现和H5的混合开发。
文字栗子的项目工程和具体跳转封装没有详细说明,本文只专注于APT技术的说明,后面会有专门的文章来说明。

谢谢!

欢迎关注公众号:JueCode



作者:juexingzhe
链接:https://www.jianshu.com/p/9616f4a462bd
来源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

你可能感兴趣的:(android组件化)