开发中使用注解框架可以极大地提高编码效率,注解框架用到的技术可以分为两种,运行时注解跟编译时注解。运行时注解一般配合反射机制使用,编译时注解则是用来生成模板代码。这里我们分别使用这两种方法实现ButterKnife的控件绑定功能。
1、运行时注解
运行时注解实现比较简单,但是由于完全依靠反射技术,所以运行效率较低。首先我们需要新建一个注解类,指定其保留时间为运行时,修饰对象为类成员变量,值为控件ID。
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface BindView {
@IdRes int value();
}
然后我们需要使用反射实现 bindView 方法。先是使用 class.getDeclaredFields 方法获取类中所有的成员变量,如果该成员变量被 @BindView 注解则根据注解中的控件ID调用 view.findViewById ,并将返回值赋给该成员变量。这种方法的一个好处是被 @BindView 注解的成员变量可以是私有的。
public class ButterKnife {
public static void bindView(Activity activity) {
bindView(activity, activity.getWindow().getDecorView());
}
public static void bindView(Object object, View view) {
Class clazz = object.getClass();
Field[] fields = clazz.getDeclaredFields();
for (Field f : fields) {
BindView binder = f.getAnnotation(BindView.class);
if (binder != null) {
f.setAccessible(true);
f.set(object, view.findViewById(binder.value()));
}
}
}
}
2、编译时注解
我们知道ButterKnife是基于编译时注解的,依赖注解生成模板代码从而实现控件绑定。使用编译时注解前需要对注解处理器(Annotation Processor)有一个基本了解。
在这里我们通过注解处理器自动生成模板代码,并且与项目代码一起编译。
2.1、创建库项目
在Android Studio中创建完项目后,我们需要在 File -> New -> New Module... 创建3个模块。
名称 | 类别 | 作用 |
test-annotations | Java Liabrary | 存放我们自己定义的注解类 |
test-api | Android Liabrary | 存放在主模块中使用的工具类 |
test-compiler | Java Liabrary | 存放注解处理器 |
然后需要在每个模块的build.gradle文件中配置它们之间的依赖关系以及需要用到的类库。
// test-annotations:
dependencies {
...
implementation 'com.android.support:support-annotations:27.1.0@jar'
} // test-api:
dependencies {
...
compile project(path: ':test-annotations')
} // test-compiler:
dependencies {
...
compile 'com.google.auto.service:auto-service:1.0-rc2'
compile project(path: ':test-annotations')
}
2.2、自定义注解
接下来我们需要在“test-annotations”模块中创建一个用于控件绑定的注解类,与第一节中类似,不过保留时间更改为了编译时。
@Retention(RetentionPolicy.CLASS)
@Target(ElementType.FIELD)
public @interface BindView {
@IdRes int value();
}
2.3、创建注解处理器
处理器的实现在“test-compiler”模块中,代码比较繁琐,大体上分为两步:收集信息然后生成代码。
@AutoService(Processor.class)
public class TestProcessor extends AbstractProcessor { private Filer mFileUtils;
private Elements mElementUtils; @Override
public synchronized void init(ProcessingEnvironment processingEnvironment) {
super.init(processingEnvironment);
mFileUtils = processingEnvironment.getFiler();
mElementUtils = processingEnvironment.getElementUtils();
} @Override
public Set<String> getSupportedAnnotationTypes() {
Set<String> annotationType = new HashSet<>();
annotationType.add(BindView.class.getCanonicalName());
return annotationType;
} @Override
public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
// 处理逻辑...
}
}
这里使用了谷歌提供的 @AutoService 对处理器进行注册,然后复写AbstractProcessor类的方法。在 getSupportedAnnotationTypes 函数中返回处理器支持的注解类型,在 init 函数中获取我们需要用到的工具类,处理器的主要逻辑则在 process 函数中。
@Override
public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
Map<String, ClassInfo> classMap = new HashMap<>();
// 收集信息:
Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(AnnotationValue.class);
for (Element element : elements) {
if (element.getKind() == ElementKind.FIELD) {
VariableElement variableElement = (VariableElement) element;
// 获取成员变量所在的类。
TypeElement typeElement = (TypeElement) variableElement.getEnclosingElement();
// 获取类的全限定名。
String qualifeiedName = typeElement.getQualifiedName().toString();
// 将同一个类的所有成员变量放到一个ClassInfo类中
ClassInfo classInfo = classMap.get(qualifeiedName);
if (classInfo == null) {
PackageElement packageElement = mElementUtils.getPackageOf(typeElement);
classInfo = new ClassInfo(packageElement, typeElement);
classMap.put(qualifeiedName, classInfo);
}
classInfo.addVariable(variableElement);
}
}
// 生成代码:
...
}
首先我们通过 roundEnvironment.getElementsAnnotatedWith 获取到被 @BindView 注解的所有元素,显然这里的 @BindView 希望注解在类的成员变量上,所以先对元素的类型进行判断,然后将所有的元素按照所在类的全限定名进行分类(通过ClassInfo类表示)。这里需要了解Element类的含义:
- VariableElement:一般代表成员变量。
- ExecutableElement:一般代表类中的方法。
- TypeElement:一般代表类。
- PackageElement:一般代表包。
@Override
public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
// 收集信息:
...
// 生成代码:
for (String key : classMap.keySet()) {
ClassInfo classInfo = classMap.get(key);
try {
JavaFileObject sourceFile = mFileUtils.createSourceFile(classInfo.getProxyClassFullName());
Writer writer = sourceFile.openWriter();
writer.write(classInfo.generateJavaCode());
writer.flush();
writer.close();
} catch (IOException e) {
e.printStackTrace();
}
}
return true;
}
代码生成阶段比较简单,先使用 mFileUtils.createSourceFile 函数创建文件对象,文件名通过 classInfo.getProxyClassFullName函数获得。然后向该文件对象中写入通过 classInfo.generateJavaCode 函数获取的源码。接下来看一下ClassInfo类的实现:
public class ClassInfo {
private String mPackageName;
private String mProxyClassName;
private ArrayList<VariableElement> mInjectVariables = new ArrayList<>();
public static final String PROXY = "ViewInject"; public ClassInfo(PackageElement packageElement, TypeElement classElement) {
// 获取包名
String packageName = packageElement.getQualifiedName().toString();
this.mPackageName = packageName;
// 获取全限定名后去除包名,将‘.’替换为‘$’是因为如果类A有个内部
// 类B,那么B在编译后名字会变为A.B
int packageLen = packageName.length() + 1;
String className = classElement.getQualifiedName().toString().substring(packageLen).replace('.', '$');
// 设置代理类的名字,之后我们需要根据这个名字获取到代理类。
this.mProxyClassName = className + "$$" + PROXY;
}
public String generateJavaCode() {
StringBuilder builder = new StringBuilder();
builder.append(String.format("package %s;\n", mPackageName));
builder.append(String.format("public final class %s {\n", mProxyClassName));
builder.append(String.format("public %s(%s obj) {\n", mProxyClassName, mClassName));
for (VariableElement element : mInjectVariables) {
AnnotationValue binder = element.getAnnotation(AnnotationValue.class);
String name = element.getSimpleName().toString();
builder.append(String.format("obj.%s=obj.findViewById(%d);\n", name, binder.value()));
}
builder.append("}\n }\n");
return builder.toString();
}
public String getProxyClassFullName() {
// 返回代理类全限定名
return mPackageName + "." + mProxyClassName;
}
public void addVariable(VariableElement element) {
mInjectVariables.add(element);
}
}
它的功能是为我们需要的代理类的生成源码,这里使用了字符串拼接的方式,也有许多第三方库可供使用,如Square公司的JavaPoet。
2.4、使用处理器
处理器编写完成后需要在主模块的build.gradle文件中进行配置才能生效,一是引入存放注解的“test-annotations”模块,二是指定“test-compiler”模块中的注解处理器。
dependencies {
...
annotationProcessor project(':test-complier')
compile project(path: ':test-api')
}
接着在MainActivity中用 @BindView 注解一个控件然后编译项目,编译完成后处理器自动生成的类就可以在主模块的 build\generated\source\apt\debug\包名\ 下找到。
package com.mmmmar.androidannotation; public final class MainActivity$$ViewInject {
public MainActivity$$ViewInject(MainActivity obj) {
obj.mTextView = obj.findViewById(2131165307);
}
}
2.5、完成控件绑定
生成代理类后我们可以在源码中进行手动调用,但是比较优雅的做法是像ButterKnife一样提供一个统一的调用工具,接下来我们在“test-api”模块中进行实现。
public class ButterKnife { public static void bindView(Activity activity) {
bindView(activity, activity.getWindow().getDecorView());
}
public static void bindView(Object object, View view) {
Class clazz = object.getClass();
String clazzName = clazz.getCanonicalName() + "$$ViewInject";
try {
Class proxy = clazz.getClassLoader().loadClass(clazzName);
Constructor constructor = proxy.getConstructor(clazz);
constructor.newInstance(object);
} catch (ClassNotFoundException e) {
e.printStackTrace();
} catch (NoSuchMethodException e) {
e.printStackTrace();
} catch (IllegalAccessException e) {
e.printStackTrace();
} catch (InstantiationException e) {
e.printStackTrace();
} catch (InvocationTargetException e) {
e.printStackTrace();
}
}
}
因为处理器生成的代理类的逻辑都在构造函数中,所以我们只需要通过类加载器根据类名获取到代理类的Class对象,然后调用其构造函数即可完成功能。虽然这里也用到了反射,但只是用来调用一次代理类的构造函数,所以不会对性能照成影响。