怎么手写Vue3响应式系统

广告:宝塔Linux面板高效运维的服务器管理软件 点击【 https://www.bt.cn/p/uNLv1L 】立即购买

怎么手写Vue3响应式系统

响应式

首先,什么是响应式呢?

响应式就是被观察的数据变化的时候做一系列联动处理。就像一个社会热点事件,当它有消息更新的时候,各方媒体都会跟进做相关报道。这里社会热点事件就是被观察的目标。那在前端框架里,这个被观察的目标是什么呢?很明显,是状态。状态一般是多个,会通过对象的方式来组织。所以,我们观察状态对象的每个 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 activeEffectfunction 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 activeEffectfunction effect(fn) {  activeEffect = fn  fn()}
登录后复制

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

let activeEffectfunction 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 嵌套了,所以必须支持嵌套。

我们嵌套下试试:

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);0
登录后复制

按理说会打印一次 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 activeEffectconst 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 响应式的核心:

以上就是怎么手写Vue3响应式系统的详细内容,更多请关注9543建站博客其它相关文章!

广告:SSL证书一年128.66元起,点击购买~~~

9543建站博客
一个专注于网站开发、微信开发的技术类纯净博客。
作者头像
admin创始人

肥猫,知名SEO博客站长,14年SEO经验。

上一篇:uniapp隐藏标题栏只显示状态栏怎么办
下一篇:web前端是网页前端吗

发表评论

关闭广告
关闭广告