手写 Vue3 响应式系统:核心就一个数据结构

6,375 阅读7分钟

响应式是 Vue 的特色,如果你简历里写了 Vue 项目,那基本都会问响应式实现原理。

而且不只是 Vue,状态管理库 Mobx 也是基于响应式实现的。

那响应式是具体怎么实现的呢?

与其空谈原理,不如让我们来手写一个简易版吧。

响应式

首先,什么是响应式呢?

响应式就是被观察的数据变化的时候做一系列联动处理。

就像一个社会热点事件,当它有消息更新的时候,各方媒体都会跟进做相关报道。

这里社会热点事件就是被观察的目标。

那在前端框架里,这个被观察的目标是什么呢?

很明显,是状态。

状态一般是多个,会通过对象的方式来组织。所以,我们观察状态对象的每个 key 的变化,联动做一系列处理就可以了。

我们要维护这样的数据结构:

状态对象的每个 key 都有关联的一系列 effect 副作用函数,也就是变化的时候联动执行的逻辑,通过 Set 来组织。

每个 key 都是这样关联了一系列 effect 函数,那多个 key 就可以放到一个 Map 里维护。

这个 Map 是在对象存在的时候它就存在,对象销毁的时候它也要跟着销毁。(因为对象都没了自然也不需要维护每个 key 关联的 effect 了)

而 WeakMap 正好就有这样的特性,WeakMap 的 key 必须是一个对象,value 可以是任意数据,key 的对象销毁的时候,value 也会销毁。

所以,响应式的 Map 会用 WeakMap 来保存,key 为原对象。

这个数据结构就是响应式的核心数据结构了。

比如这样的状态对象:

const obj = {
    a: 1,
    b: 2
}

它的响应式数据结构就是这样的:

const depsMap = new Map();

const aDeps = new Set();
depsMap.set('a', aDeps);

const bDeps = new Set();
depsMap.set('b', bDeps);

const reactiveMap = new WeakMap()

reactiveMap.set(obj, depsMap);

创建出的数据结构就是图中的那个:

然后添加 deps 依赖,比如一个函数依赖了 a,那就要添加到 a 的 deps 集合里:

effect(() => {
    console.log(obj.a);
});

也就是这样:

const depsMap = reactiveMap.get(obj);

const aDeps = depsMap.get('a');

aDeps.add(该函数);

这样维护 deps 功能上没啥问题,但是难道要让用户手动添加 deps 么?

那不但会侵入业务代码,而且还容易遗漏。

所以肯定不会让用户手动维护 deps,而是要做自动的依赖收集。

那怎么自动收集依赖呢?

读取状态值的时候,就建立了和该状态的依赖关系,所以很容易想到可以代理状态的 get 来实现。

通过 Object.defineProperty 或者 Proxy 都可以:

const data = {
    a: 1,
    b: 2
}

let activeEffect
function effect(fn) {
  activeEffect = fn
  fn()
}

const reactiveMap = new WeakMap()

const obj = new Proxy(data, {
    get(targetObj, key) {
        let depsMap = reactiveMap.get(targetObj);
        
        if (!depsMap) {
          reactiveMap.set(targetObj, (depsMap = new Map()))
        }
        
        let deps = depsMap.get(key)
        
        if (!deps) {
          depsMap.set(key, (deps = new Set()))
        }
        
        deps.add(activeEffect)

        return targetObj[key]
   }
})

effect 会执行传入的回调函数 fn,当你在 fn 里读取 obj.a 的时候,就会触发 get,会拿到对象的响应式的 Map,从里面取出 a 对应的 deps 集合,往里面添加当前的 effect 函数。

这样就完成了一次依赖收集。

当你修改 obj.a 的时候,要通知所有的 deps,所以还要代理 set:

set(targetObj, key, newVal) {
    targetObj[key] = newVal

    const depsMap = reactiveMap.get(targetObj)

    if (!depsMap) return

    const effects = depsMap.get(key)
    effects && effects.forEach(fn => fn())
}

基本的响应式完成了,我们测试一下:

打印了两次,第一次是 1,第二次是 3。

effect 会先执行一次传入的回调函数,触发 get 来收集依赖,这时候打印的 obj.a 是 1

然后当 obj.a 赋值为 3 后,会触发 set,执行收集的依赖,这时候打印 obj.a 是 3

依赖也正确收集到了:

结果是对的,我们完成了基本的响应式!

当然,响应式不会只有这么点代码的,我们现在的实现还不完善,还有一些问题。

比如,如果代码里有分支切换,上次执行会依赖 obj.b 下次执行又不依赖了,这时候是不是就有了无效的依赖?

这样一段代码:

const obj = {
    a: 1,
    b: 2
}
effect(() => {
    console.log(obj.a ? obj.b : 'nothing');
});
obj.a = undefined;
obj.b = 3;

第一次执行 effect 函数,obj.a 是 1,这时候会走到第一个分支,又依赖了 obj.b。

把 obj.a 修改为 undefined,触发 set,执行所有的依赖函数,这时候走到分支二,不再依赖 obj.b。

把 obj.b 修改为 3,按理说这时候没有依赖 b 的函数了,我们执行试一下:

第一次打印 2 是对的,也就是走到了第一个分支,打印 obj.b

第二次打印 nothing 也是对的,这时候走到第二个分支。

但是第三次打印 nothing 就不对了,因为这时候 obj.b 已经没有依赖函数了,但是还是打印了。

打印看下 deps,会发现 obj.b 的 deps 没有清除

所以解决方案就是每次添加依赖前清空下上次的 deps。

怎么清空某个函数关联的所有 deps 呢?

记录下就好了。

我们改造下现有的 effect 函数:

let activeEffect
function effect(fn) {
  activeEffect = fn
  fn()
}

记录下这个 effect 函数被放到了哪些 deps 集合里。也就是:

let activeEffect
function effect(fn) {
  const effectFn = () => {
      activeEffect = effectFn
      fn()
  }
  effectFn.deps = []
  effectFn()
}

对之前的 fn 包一层,在函数上添加个 deps 数组来记录被添加到哪些依赖集合里。

get 收集依赖的时候,也记录一份到这里:

这样下次再执行这个 effect 函数的时候,就可以把这个 effect 函数从上次添加到的依赖集合里删掉:

cleanup 实现如下:

function cleanup(effectFn) {
    for (let i = 0; i < effectFn.deps.length; i++) {
        const deps = effectFn.deps[i]
        deps.delete(effectFn)
    }
    effectFn.deps.length = 0
}

effectFn.deps 数组记录了被添加到的 deps 集合,从中删掉自己。

全删完之后就把上次记录的 deps 数组置空。

我们再来测试下:

无限循环打印了,什么鬼?

问题出现在这里:

set 的时候会执行所有的当前 key 的 deps 集合里的 effect 函数。

而我们执行 effect 函数之前会把它从之前的 deps 集合中清掉:

执行的时候又被添加到了 deps 集合。

这样 delete 又 add,delete 又 add,所以就无限循环了。

解决的方式就是创建第二个 Set,只用于遍历:

这样就不会无限循环了。

再测试一次:

现在当 obj.a 赋值为 undefined 之后,再次执行 effect 函数,obj.b 的 deps 集合就被清空了,所以需改 obj.b 也不会打印啥。

看下现在的响应式数据结构:

确实,b 的 deps 集合被清空了。

那现在的响应式实现是完善的了么?

也不是,还有一个问题:

如果 effect 嵌套了,那依赖还能正确的收集么?

首先讲下为什么要支持 effect 嵌套,因为组件是可以嵌套的,而且组件里会写 effect,那也就是 effect 嵌套了,所以必须支持嵌套。

我们嵌套下试试:

effect(() => {
    console.log('effect1');
    effect(() => {
        console.log('effect2');
        obj.b;
    });
    obj.a;
});
obj.a = 3;

按理说会打印一次 effect1、一次 effect2,这是最开始的那次执行。

然后 obj.a 修改为 3 后,会触发一次 effect1 的打印,执行内层 effect,又触发一次 effect2 的打印。

也就是会打印 effect1、effect2、effect1、effect2。

我们测试下:

打印了 effect1、effet2 这是对的,但第三次打印的是 effect2,这说明 obj.a 修改后并没有执行外层函数,而是执行的内层函数。

为什么呢?

看下这段代码:

我们执行 effect 的时候,会把它赋值给一个全局变量 activeEffect,然后后面收集依赖就用的这个。

当嵌套 effect 的时候,内层函数执行后会修改 activeEffect 这样收集到的依赖就不对了。

怎么办呢?

嵌套的话加一个栈来记录 effect 不就行了?

也就是这样:

执行 effect 函数前把当前 effectFn 入栈,执行完以后出栈,修改 activeEffect 为栈顶的 effectFn。

这样就保证了收集到的依赖是正确的。

这种思想的应用还是很多的,需要保存和恢复上下文的时候,都是这样加一个栈。

我们再测试一下:

现在的打印就对了。

至此,我们的响应式系统就算比较完善了。

全部代码如下:

const data = {
    a: 1,
    b: 2
}

let activeEffect
const effectStack = [];

function effect(fn) {
  const effectFn = () => {
      cleanup(effectFn)
      
      activeEffect = effectFn
      effectStack.push(effectFn);
  
      fn()
  
      effectStack.pop();
      activeEffect = effectStack[effectStack.length - 1];
  }
  effectFn.deps = []
  effectFn()
}

function cleanup(effectFn) {
    for (let i = 0; i < effectFn.deps.length; i++) {
        const deps = effectFn.deps[i]
        deps.delete(effectFn)
    }
    effectFn.deps.length = 0
}

const reactiveMap = new WeakMap()

const obj = new Proxy(data, {
    get(targetObj, key) {
        let depsMap = reactiveMap.get(targetObj)
        
        if (!depsMap) {
          reactiveMap.set(targetObj, (depsMap = new Map()))
        }
        
        let deps = depsMap.get(key)
        
        if (!deps) {
          depsMap.set(key, (deps = new Set()))
        }
        
        deps.add(activeEffect)

        activeEffect.deps.push(deps);

        return targetObj[key]
   },
   set(targetObj, key, newVal) {
        targetObj[key] = newVal

        const depsMap = reactiveMap.get(targetObj)

        if (!depsMap) return

        const effects = depsMap.get(key)

        // effects && effects.forEach(fn => fn())
        const effectsToRun = new Set(effects);
        effectsToRun.forEach(effectFn => effectFn());
    }
})

总结

响应式就是数据变化的时候做一系列联动的处理。

核心是这样一个数据结构:

最外层是 WeakMap,key 为对象,value 为响应式的 Map。这样当对象销毁时,Map 也会销毁。

Map 里保存了每个 key 的依赖集合,用 Set 组织。

我们通过 Proxy 来完成自动的依赖收集,也就是添加 effect 到对应 key 的 deps 的集合里。 set 的时候触发所有的 effect 函数执行。

这就是基本的响应式系统。

但是还不够完善,每次执行 effect 前要从上次添加到的 deps 集合中删掉它,然后重新收集依赖。这样可以避免因为分支切换产生的无效依赖。

并且执行 deps 中的 effect 前要创建一个新的 Set 来执行,避免 add、delete 循环起来。

此外,为了支持嵌套 effect,需要在执行 effect 之前把它推到栈里,然后执行完出栈。

解决了这几个问题之后,就是一个完善的 Vue 响应式系统了。

当然,现在虽然功能是完善的,但是没有实现 computed、watch 等功能,之后再实现。

最后,再来看一下这个数据结构,理解了它就理解了 vue 响应式的核心: