angular源码阅读3:真的,依赖注入的原理

前面已经提到了:

如何注册一个module。

如何获取一个module。

injector与module以及provider的关系。

那么已经剩下最后一部分了,就是关于依赖是如何被注入的。

且看下面这段代码。

//如你所见,注册了一个moduleA
//又如你所见,给moduleA上面注册了两个全局变量,a和b
angular.module('moduleA',[]);
angular.module('moduleA').constant('a',1);
angular.module('moduleA').constant('b',2);

//生成一个注射器
var injector = createInjector('moduleA');


//定义一个函数,接收两个数作为参数,打印两个数的和
function add(a,b){
    console.log(a+b);
}

函数声明好了,moduleA上面也有值了,那么怎么把moduleA上面的值作为函数的参数传进来呢?熟悉angular的小伙伴肯定知道有如下几种办法:

//第一种,给函数加$inject属性
add.$inject=['a','b'];
add(a,b)//3

//第二种,数组形式
['a','b',add(a,b)];

//第三种,直接用
add(a,b);

这次主要看第一种和第二种的原理,第三种的原理有点复杂,留着下一次说。

还记得之前的那个createInjector函数么?这个函数接收一些module名称作为参数,返回一个对象就是injector对象。当createInjector运行的时候,会把所有module都遍历一次,把module的_invokeQueue里面的任务都找到相应的provider执行一下,然后把各个module里面注册的东西都保存一份。

而依赖注入的本质就是——当你执行函数A的时候,根据函数A的$injector属性,或者函数A的数组元素(比如 ['a','b',add(a,b)]这种),去自己保存的那堆东西里面找到相应的数据,把这个数据传给函数进行调用。

来,配上源代码,别看代码长,其实都是上一篇博客的代码,这次只往里面加了一小部分。

//createInjector(['app1','app2'])
//参数是一个字符串或者一个数组,内容是module名
function createInjector(modulesToLoad) {
    //cache用来缓存一些一直可以用到的值
    var cache = {};

    //所有加载过的module都记录在这个对象里面
    var loadedModules = {};

    $provide = {
        constant: function(key, value) {
            cache[key] = value;
        }
    }

    //这里的foreach方法并不是一个真正能运行的foreach,能看懂就行了
    //每次APP启动的时候,injector都会按照传入的module名来遍历所有module
    //这样就可以得到所有module发布的任务,并且一一执行这些任务
    $.forEach(modulesToLoad, function loadModule(moduleName) {
        //查询一下记录,如果没有加载过,再去执行
        if (!loadedModules[moduleName]) {
            var module = window.angular.module(moduleName);
            //每加载一个模块,就操作一下loadedModules,把这个模块信息放进去
            loadedModules[moduleName]=true;
            //每次先把所有依赖的模块加载进去
            $.forEach(module.requires,loadModule);
            //然后再去调用这个模块的invokeQueue
            $.forEach(module._invokeQueue, function(invokeArgs) {
                var method = invokeArgs[0];
                var args = invokeArgs[1];
                $provide[method].apply($provide, args);
            })
        }
    })

    //invoke方法用来调用函数
    function invoke(fn){
        //把fn.$inject这个数组里面的元素拿出来,args是一个新的数组
        var args=$.map(fn.$inject,function(token){
            return cache[token];
        })
        fn.apply(null,args);
    }


    return {
        has: function(key) {
            return cache.hasOwnProperty(key)
        },
        get: function(key) {
            return cache[key]
        },
        invoke:invoke
    }
}

看到那个invoke方法了么?来先顺一下思路,当createInjector函数加载了moduleA之后,返回了injector的时候,在自己的内部存储的cache里面已经是这样的:

cache={
    a:1,
    b:2
}

然后给返回的injector增加了一个invoke方法,这个方法是用来调用函数的。代码里看的应该很清楚,invoke方法接收一个函数作为参数,然后去寻找这个函数的$injector属性,这个属性里发现了a,b两个值,于是就去cache里面找key值为a,b的两个值,然后成为一个数组args,在用fn.apply(null,args)调用函数,这样这两个之就成为函数的参数传进去了。  

关于apply这个方法,简单说一句,第一个参数是函数的this值,第二个数组是函数的参数。有兴趣看源码的童鞋应该对这个不陌生,我就不啰嗦了。

总之,这样一来,就把module里面的值注入给函数了。注意哦,调用函数是Injector这个对象调用的:

injector.invoke(add)//调用add函数

那么问题来了,既然给函数增加$inject属性可以注入依赖了,那么数组形式的调用怎么实现呢?其实这个也很简单。

回想一下,其实依赖注入的本质是通过一个key值,从cache对象里面拿到相应的数据,对吧?

在injector里面还有一个方法,叫做 annotate,中文意思是解剖,这个方法的作用就是把函数的依赖给解剖出来:比如这篇文章的例子,我们是通过$inject属性,把a,b两个key值拿出来的。看代码:

//数组形式依赖注入
['a','b',function add(a,b){}];

//直接调用的依赖注入
function (a,b){}

//annonate方法的作用就是把上面两种形式里面的a,b这两个键值拿出来

其实对于数组形式依赖注入来说,键值很好拿,拆一下数组就行了。我就贴一下annotate的简单代码,很好懂的:

function annotate(fn){
        //如果是一个数组 ['a','b',function(a,b){}]
        //返回数组最后一个函数之前的所有部分
        if(isArray(fn)){
            return fn.slice(0,fn.length-1);
        }else{
            //如果不是数组,就返回这个函数的$inject属性
            return fn.$inject;
        }
    }

就是这么简单!

好了,其实依赖注入更牛逼的是那个直接调用的依赖注入,这个篇幅比较长,我打算明天再更。

晚安了兄弟们。欢迎留言和我交流呀 :)