Android 曝光埋点组件 android 埋点方案_语法树

APT

APT 是 Annotation Processing Tool 的缩写,即注解处理器,是一种处理注解的工具。确切的说它是 javac 的一个工具, 它用来在编译时扫描和处理注解。注解处理器以 Java 代码(或者编译过的字节码)作为输入,生成 .java 文件作为输出。 简单来说就是在编译期,通过注解生成 .java 文件。权限控制、代码调试等。

Element

自定义注解处理器,需要继承 AbstractProcessor 类,而 AbstractProcessor 最终要的就是 process 方法。process 方法 处理的核心是 Element 对象。

Element 的源码源码如下:

Android 曝光埋点组件 android 埋点方案_android 埋点工具_02

Android 曝光埋点组件 android 埋点方案_android 埋点工具_03

从定义可以看出,Element 是一个接口,它定义了外部可以调用的方法。

Android 曝光埋点组件 android 埋点方案_语法树_04

Element 有 5 个直接子类,它们分别代表一种特定类型的元素。五个子类各有各的用处并且有各种独有的方法,在使用的 时候可以强制将 Element 对象转换成其中的任何一种,但是必须满足转换的条件,不然会抛出异常。

Android 曝光埋点组件 android 埋点方案_android 埋点工具_05

其中 TypeElement 和 VariableElement 是最核心的两个 Element,也是我们下文会使用到的。

ATP 实例

我们通过 APT 来实现一个功能,功能类似于 ButterKnife 中的 @BindView 注解。即:通过对 View 变量的注解,实现 View 的绑定(无需调用 findViewById)。

完整的项目源码后续会 release 给大家。

AST

AST,是 Abstract Syntax Tree 的缩写,即“抽象语法树”,是编辑器对代码的第一步加工之后的结果,是一个树形式表示 的源代码。源代码的每个元素映射到一个节点或子树。

Java 的编译过程可以分成三个阶段:

第一阶段:

所有的源文件会被解析成语法树;

第二阶段:

调用注解处理器,即 APT。如果注解处理器产生了新的源文件,新的源文件也要参与编译;

第三阶段:

语法树会被分析并转化成类文件。

Android 曝光埋点组件 android 埋点方案_Android 曝光埋点组件_06

原理概述

编辑器对代码处理的流程大概是:

JavaTXT-> 词语法分析 -> 生成 AST -> 语义分析 -> 编译字节码

通过操作 AST,可以达到修改源代码的功能。

可 以 在 注 解 处 理 器 的 process 函 数 里,通 过 roundEnvironment.getRootEle- ments() 方法可以拿到所有的 Element 对象,通过 trees.getTree(element) 方 法可以拿到对应的抽象语法树(AST),然后我们自定义一个 TreeTranslator, 在 visitMethodDef 里即可对方法进行判断。如果是目标处理方法,则通过 AST 的相关 API 插入埋点代码。

实现步骤

完整的项目源码后续会 release 给大家

知识点

• APT

• AST

缺点

• com.sun.tools.javac.tree 相关 API 语法晦涩,理解难度大,要求对编译原理 有一定的基础

• APT 无法扫描其他 module,导致 AST 无法处理其他 module

• 不支持 Lambda 语法

• 带有返回值的方法,很难把埋点代码插入到方法之后