首页 > 代码库 > 手动实现IOC框架,与findViewById说拜拜

手动实现IOC框架,与findViewById说拜拜

手动实现IOC框架,与findViewById说拜拜

转载请标明出处:
http://blog.csdn.net/lisdye2/article/details/64233563
本文出自:【Alex_MaHao的博客】
项目中的源码已经共享到github,有需要者请移步【Alex_MaHao的github】

自序

在开发Android中,总要写许多的findViewById方法,这无疑是一件非常痛苦的事情。直到接触了Xutils框架,发现竟然可以使用注解的方式,优雅的干掉了findViewById方法,当时真是惊为天人。后来遇到了Butterknife之后,发现不仅能够通过实现findViewById方法,甚至连setOnClickListenergetString()getResource()方法,都能通过一行注解的方式快速的实现。

当时在使用过程中,感觉这种方式大大的提高了开发的效率,以及编码的舒畅度,自己很有必要实现以下。于是便有了这篇博客,该篇博客主要实现了findViewById方法,虽然广度不是很大,但他们的原理都是相同的。

如何使用

使用方式和Butterknife相似,通过注解@BindView标识控件,通过ViewFinder.inject(this)实现代码的注入。

public class MainActivity extends AppCompatActivity {
    // 通过注解绑定控件
    @BindView(R.id.text)
    TextView textView;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);
        setContentView(R.layout.activity_main);
        //代码注入
        ViewFinder.inject(this);
        textView.setText("123");
    }
}

实现原理

实现方式有两种

通过反射实现:如果对于反射有深入了解,则应该清楚,我们可以通过反射获取到该注解,并且获取到该注解的值等等一系列的必须量,通过反射我们实现对控件的注入。该方法虽然可以实现,但对效率有着一定的影响。毕竟反射很影响效率。

通过编译器生成代码实现:在我们运行java代码的时候,通常先通过javac将java文件编译成.class文件,然后运行class文件,那么我们能不能够在编译时期,根据我们的注解生成findViewById等方法,这样我们就能够在运行期查找控件。结果当然是可以的,本例就是使用这种方式。因为虽然其使用的是注解,但在运行期其实质仍是通过findViewById方法查找控件,相比于反射来说,大大的提高了性能。

通过编译器生成代码的大致流程如下:

  • 编写Modulioc-annotation,该工程主要定义注解@BindView用以修饰变量。
  • 编写Modul:ioc-compiler,该工程为最终会打成jar包,主要是在javac编译时期根据注解生成注入代码的相关类
  • 编写Modulioc-api,该工程主要提供注入的调用方法ViewFidder.inject(),调用代码注入的方法。
  • 编写Modulapp,测试工程。

代码实现

根据上面的流程,开始实现框架

编写ioc-annotation模块

该模块比较简单,就是定义一个注解。如下:

@Retention(RetentionPolicy.CLASS)
@Target(ElementType.FIELD)
public @interface BindView {
    int value();
}

该注解主要有两个功能:

  • Activity中修饰变量,用以标识需要findViewById的相关控件。
  • ioc-compiler模块中,用以检索和获取需要findViewById的控件。

编写ioc-compiler模块

在之前我们提到过,该框架的原理是在编译时期根据我们的要求生成注入的辅助代码,那么如何生成,以何种规则生成,肯定是由我们来定义的。

javac命令中,可以在其编译的指定目录放入一个.jar文件,当然这个.jar文件有特殊的要求(后面再说),这样运行javac命令之前,javac会调用jar,通过这种特性订制一些我们想实现的功能。

那么看一下该模块的关键类IocProcessor

@AutoService(Processor.class)
public class IocProcessor extends AbstractProcessor {

    // 文件相关的辅助类,生成JavaSouceCode
    private Filer mFileUtils;

    private Elements mElementUtils;
    // 日志相关
    private Messager mMessager;

    private Types mTypeUtils;

    @Override
    public synchronized void init(ProcessingEnvironment processingEnvironment) {
        super.init(processingEnvironment);
        mTypeUtils = processingEnvironment.getTypeUtils();
        mFileUtils = processingEnvironment.getFiler();
        mElementUtils = processingEnvironment.getElementUtils();
        mMessager = processingEnv.getMessager();
    }

    /**
     * 标示该处理器捕获处理的注解类型
     */
    @Override
    public Set<String> getSupportedAnnotationTypes() {
        Set<String> annotationTypes = new LinkedHashSet<>();
        // getCanonicalName 获取规范的名字
        annotationTypes.add(BindView.class.getCanonicalName());
        return annotationTypes;
    }
    /**
     *  指定使用的java版本,一般默认支持返回最新
     * @return
     */
    @Override
    public SourceVersion getSupportedSourceVersion() {
        return SourceVersion.latestSupported();
    }

    private Map<String, ProxyInfo> mProxyMap = new HashMap<>();
    /**
     * 相当于main 函数,处理扫描,评估和处理注解的代码以及生成java文件
     * @param set
     * @param roundEnvironment 用于查询包含特定注解的注解元素
     * @return
     */
    @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        //.....关键处理代码
        return true;
    }
}

为了方便浏览,暂时删去了一些代码。

首先在类的声明上,我们添加了一个注解@AutoService,该注解是由Google提供的用于生成.jar包的相关注解。使用该注解需要添加依赖com.google.auto.service:auto-service:1.0-rc3,在这里有一个疑问,为什么要使用该注解呢,生成的jar包邮什么特殊的地方吗?

因为该jar包是提供于注解时期的,所以其有特殊的目录形式:

技术分享

除了有最基本的类以外,多了META-INFservice目录,该目录中的文件命名是固定的,其内容很简单

com.example.IocProcessor

就一行内容,标明主要的处理类。而如果我们手动编写jar的目录很麻烦,所以Google提供了一个@AutoService,用以直接实现这种目录形式的jar包。

看完注解之后,看类声明,继承AbstractProcessor,实现三个方法:

  • getSupportedAnnotationTypes:获取注解支持的类型,在此,我们支持BindView,没有什么疑问
  • getSupportedSourceVersion:指定使用的java版本,一般默认支持返回最新,默认此写法,没什么问题。
  • process:处理注解的主要方法,主要在此方法中实现对相关类的处理。

那么看一下process的实现:

 private Map<String, ProxyInfo> mProxyMap = new HashMap<>();
    /**
     * 相当于main 函数,处理扫描,评估和处理注解的代码以及生成java文件
     * @param set
     * @param roundEnvironment 用于查询包含特定注解的注解元素
     * @return
     */
    @Override
    public boolean process(Set<? extends TypeElement> set, RoundEnvironment roundEnvironment) {
        mProxyMap.clear();
        // --------------------检索注解,保存包含注解的类以及需要注入的成员变量-------------------------
        // 该方法获取的element是被注解标注的所有元素,所以需要检查
        Set<? extends Element> elements = roundEnvironment.getElementsAnnotatedWith(BindView.class);
        for (Element element : elements) {
            // 检查element的类型
            if (!checkAnnotationValid(element, BindView.class)) {
                return false;
            }
            VariableElement variableElement = ((VariableElement) element);
            TypeElement typeElement = (TypeElement) variableElement.getEnclosingElement();
            // 全路径名
            String qualifiedName = typeElement.getQualifiedName().toString();
            // 类信息
            ProxyInfo proxyInfo = mProxyMap.get(qualifiedName);
            if (proxyInfo == null) {
                proxyInfo = new ProxyInfo(mElementUtils, typeElement);
                mProxyMap.put(qualifiedName, proxyInfo);
            }
            // 在类信息中保存需要注入的成员变量和id
            BindView annotation = variableElement.getAnnotation(BindView.class);
            int id = annotation.value();
            proxyInfo.injectVariables.put(id, variableElement);
        }
        // ---------------- 生成对应的类----------------------------
        for (String key : mProxyMap.keySet()) {
            ProxyInfo proxyInfo = mProxyMap.get(key);
            try {
                // processingEnv : 注解处理环境(工具类),提供很多有用的功能工具类
                JavaFileObject jfo = processingEnv.getFiler().createSourceFile(
                        proxyInfo.getProxyClassFullName(),
                        proxyInfo.getTypeElement());
                Writer writer = jfo.openWriter();
                writer.write(proxyInfo.generateJavaCode());
                writer.flush();
                writer.close();
                //----- 在生成类之后gradle会对代码进行优化
            } catch (IOException e) {
                error(proxyInfo.getTypeElement(),
                        "Unable to write injector for type %s: %s",
                        proxyInfo.getTypeElement(), e.getMessage());
            }

        }
        return true;
    }

自上而下分析:

  • 首先获取被注解@BindView修饰的元素,注意此时获取的元素可能是TypeElementVariableElementExecuteableElement等,不一定确定是成员变量,所以要做一层检查。

检查的方法如下

 /**
     * 检查查找的元素是否合法
     */
    private boolean checkAnnotationValid(Element annotatedElement, Class clazz) {
        if (annotatedElement.getKind() != ElementKind.FIELD) {
            error(annotatedElement, "%s must be declared on field.", clazz.getSimpleName());
            return false;
        }
        if (ClassValidator.isPrivate(annotatedElement)) {
            error(annotatedElement, "%s() must can not be private.", annotatedElement.getSimpleName());
            return false;
        }
        return true;
    }

    // 展示错误信息
    private void error(Element element, String message, Object... args) {
        if (args.length > 0) {
            message = String.format(message, args);
        }
        mMessager.printMessage(Diagnostic.Kind.NOTE, message, element);
    }
  • 其次:获取到需要生成辅助类的类(含有注解@BindView的类),并保存类信息到ProxyInfo中,同时保存被注解的成员变量到该ProxyInfo中,以Map的形式保存,键是注解的值,value是被注解的元素。到这里会好奇什么是辅助类,他是什么形式的,我们看一下最终生成的辅助类就会清晰了
// 依赖注入生成的辅助类,最终调用inject()方法,实现注入
public class MainActivity$$Finder implements Finder<MainActivity> {
    public MainActivity$$Finder() {
    }

    public void inject(MainActivity host, Object source, Provider provider) {
        // 依赖注入的具体实现
        host.textView = (TextView)((TextView)provider.findView(source, 2131492945));
    }
}

生成的辅助类很清晰,看一眼即可,此时多看会模糊,我们只要确定我们要生成如上格式的类就行了。

  • 保存了类以及注解的相关信息之后,便是生成代码,循环mProxyMap获取相关类,生成代码,其实就是通过流写文件,关键的实现writer.write(proxyInfo.generateJavaCode());,其中调用了proxyInfo.generateJavaCode()用以生成代码,看一下实现
 // 生成java 代码
    public String generateJavaCode() {
        StringBuilder builder = new StringBuilder();
        builder.append("// Generated code. Do not modify!\n");
        builder.append("package ").append(packageName).append(";\n\n");
        builder.append("import com.mahao.ioc_api.*;\n");
        builder.append(‘\n‘);
        builder.append("public class ").append(proxyClassName).append(" implements " + ProxyInfo.PROXY + "<" + typeElement.getQualifiedName() + ">");
        builder.append(" {\n");
        // 添加inject()方法
        generateMethods(builder);
        builder.append(‘\n‘);
        builder.append("}\n");
        return builder.toString();
    }

    // 生成注入的方法
    private void generateMethods(StringBuilder builder) {
        builder.append("@Override\n ");
        builder.append("public void inject(" + typeElement.getQualifiedName() + " host, Object source, Provider provider) {\n");
        for (int id : injectVariables.keySet()) {
            VariableElement element = injectVariables.get(id);
            String name = element.getSimpleName().toString();
            String type = element.asType().toString();
            builder.append("host." + name).append(" = ");
            builder.append("(" + type + ")(provider.findView( source," + id + "));\n");
        }
        builder.append("  }\n");
    }
  • 上面的两个方法可能有一点复杂,但逻辑不是很绕。
  • 最终我们编写完了在编译期需要用到的jar。可以看到其主要目的是为了生成xxx$$Finder的辅助类的。

编写ioc-api模块

有了标识,也有了编译时期的jar包,最终也生成了xxx$$Finder的辅助类,那么久剩下最后的调用了。

根据我们最开始的使用方法ViewFinder.inject()可知,肯定要从这里看起。

public class ViewFinder {

    private static final ActivityProvider PROVIDER_ACTIVITY = new ActivityProvider();
    private static final ViewProvider PROVIDER_VIEW = new ViewProvider();

    /**
     * 从activity 注入
     * @param activity
     */
    public static void inject(Activity activity) {
        inject(activity, activity, PROVIDER_ACTIVITY);
    }
    /**
     * 从View 中查找
     * @param host
     * @param view
     */
    public static void inject(Object host, View view) {
        // for fragment
        inject(host, view, PROVIDER_VIEW);
    }
    // 注入的最终调用
    public static void inject(Object host, Object source, Provider provider) {
        Class<?> clazz = host.getClass();
        String proxyClassFullName = clazz.getName()+"$$Finder";// 根据传入的对象获取辅助类的全路径名
        Class<?> proxyClazz = null;
        try {
            proxyClazz = Class.forName(proxyClassFullName);// 获取辅助类的class
            Finder viewInjector = (Finder) proxyClazz.newInstance(); // 生成辅助类的实例
            viewInjector.inject(host, source, provider); // 调用辅助类的inject()方法
        }catch (Exception e) {
            e.printStackTrace();
        }
    }
}

可以看到,最终的inject()实现方法。因为在生成辅助类的时候,我们是根据类名+$$finder的固定写法,所以很容易的通过class的形式生成辅助类的实例化,并调用inject方法。在这里出现了Finder对象,他实际是我们定义的接口

public interface Finder<T> {
    void inject(T host, Object source, Provider provider);
}

辅助类在生成代码时,统一继承该接口,便于调用inject()方法(多态).

注意,此时可以看一下viewInjector.inject(host, source, provider);方法,在看一下生成的辅助类

public class MainActivity$$Finder implements Finder<MainActivity> {

    // 看这里,有没有发现什么
    public void inject(MainActivity host, Object source, Provider provider) {
        host.textView = (TextView)((TextView)provider.findView(source, 2131492945));
    }
}

有没有感觉到一种抓住了什么的感觉,再仔细看一下ViewFinder类,提供了两个inject()方法,分别是从Activity中注入,一种是从View中注入,为什么从View中注入,我们需要多传入一个Object对象呢?

仔细观察发现:

  • host其实是我们成员变量所在的类
  • sourcefindViewById操作的类,即source.findViewById
  • provider: 对findViewById的一层封住,便于处理ActivityView两种情况。

最终,整个框架到这里就完全的串联起来了

最后的最后,我们需要添加ioc-compiler到我们的编译时期,这时需要借助gradle的一个脚本android-apt,用以达到在编译期添加工程依赖。详细过程不写了,可以看我的github上的源码。github地址已在文章顶部贴出。

总结

通过以上的流程可以发现,我们在编译期生成辅助类用以完成findViewById的操作。那么根据这个原理,实现setOnClickListener等,基本上和这些事类似的。甚至基于这种思路,我们完全可以做更多的事情。

<script type="text/javascript"> $(function () { $(‘pre.prettyprint code‘).each(function () { var lines = $(this).text().split(‘\n‘).length; var $numbering = $(‘
    ‘).addClass(‘pre-numbering‘).hide(); $(this).addClass(‘has-numbering‘).parent().append($numbering); for (i = 1; i <= lines; i++) { $numbering.append($(‘
  • ‘).text(i)); }; $numbering.fadeIn(1700); }); }); </script>

    手动实现IOC框架,与findViewById说拜拜