手写ButterKnife
开发中使用注解框架可以极大地提高编码效率,注解框架用到的技术可以分为两种,运行时注解跟编译时注解。运行时注解一般配合反射机制使用,编译时注解则是用来生成模板代码。这里我们分别使用这两种方法实现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)有一个基本了解。
Annotation processing is a tool build in javac for scanning and processing annotations at compile time. You can register your own annotation processor for certain annotations.
在这里我们通过注解处理器自动生成模板代码,并且与项目代码一起编译。
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对象,然后调用其构造函数即可完成功能。虽然这里也用到了反射,但只是用来调用一次代理类的构造函数,所以不会对性能照成影响。