Vue.js设计与实现学习总结(第四章2)分支切换与清除

---

highlight: monokai-sublime

分支切换

const data = { ok: true, text: 'hello world' }
const obj = new Proxy(data, {/*....*/}

effect(function effectFn () {
    document.body.innerText = obj.ok ? obj.text : 'not'
})

如上代码, effectFn函数中存在三元表达式, 根据obj的ok属性的值执行响应的代码, ok属性改变时, 执行的代码也会改变, 就是所谓分支切换. 在effectFn中, 若ok的值为true, 会触发ok与text的读取操作, 因此该函数与这两个响应数据都建立了联系, 如下图:

但是当ok为false时, 不会对text进行读取操作, 因此无论text的值怎么改变, DOM都不会进行更新, 但是因为与辅助用函数进行了绑定, 函数依然会运行, 这是不应该的.

为要解决这个问题, 可以在每次副作用函数执行时, 可以先将其从所有与之关联的集合中删除. 因此就需要记录哪些集合中(就是上图中的Set)含有该副作用函数, 可以给副作用函数, 添加一个属性, 值为数组用于记录, 如下代码:

// 使用全局变量储存被注册的副作用函数
let activeEffect = undefined
function effect (fn) {
    const effectFn = () => {
        // 当 effectFn 执行时, 将其设置为当前激活的副作用函数
        activeEffect = effectFn
        fn()
    }
    
    // activeEffect.deps 用与储存所有包含该副作用函数(即关联该副作用函数)的集合 Set
    activeEffect.deps = []
    // 执行副作用函数
    effectFn()
 }

在副作用函数是Proxy中的拦截读取操作时绑定的, 因此可以在拦截读取操纵中收集包含该副作用函数的集合:

function track (target, key) {
    // 没有 activeEffect 直接结束
    if (activeEffect) return
    // 根据 target 从 WeakMap中获取 Map
    let depsMap = bucket.get(target)
    if (!depsMap) bucket.set(target, (depsMap = new Map()))
    
    let desp = depsMap.get(key)
    if (!deps) depsMap.set(key, (deps = new Set()))
    // 将当前激活的副作用函数添加到依赖集合中
    deps.add(activeEffect)
    
    // deps 就是存有当前副作用函数的集合, 即存在联系的依赖集合
    // 将其添加到 activeEffect.deps 中
    activeEffect.deps.push(deps)
}

关系如下图:

清除依赖

根据上图的联系, 就可以在副作用函数每次执行时, 根据 effectFn.deps 将副作用函数从依赖中(Set)删除

// 使用全局变量储存被注册的副作用函数
let activeEffect = undefined
function effect (fn) {
    const effectFn = () => {
        // 调用 cleanup 函数完成清理
        cleanup(effectFn)
        // 当 effectFn 执行时, 将其设置为当前激活的副作用函数
        activeEffect = effectFn
        fn()
    }
    
    // activeEffect.deps 用与储存所有包含该副作用函数(即关联该副作用函数)的集合 Set
    activeEffect.deps = []
    // 执行副作用函数
    effectFn()
 }
 
 function cleanup(effectFn) {
     // 遍历数组
     for (let item of effectFn.deps){
         // item 就是副作用函数集合(Set)
         item.delete(effectFn)
     }
     
     // 最后重置数组
     effectFn.deps.length = 0
 }

但此时会导致现在的响应式代码出现无限循环, 问题出在拦截设置操作中:

function trigger (target, key) {
    const depsMap = bucket.get(target)
    if (!depsMap) return
    const effects = depsMap.get(key)
    effects && effects.forEach(fn => fn()) // 问题出在这一行
}

上面代码中, 最后一行遍历的effects实际上就是当前key的副作用集合Set, 在遍历中副作用函数会运行, 此时会cleanup进行清除, 但是副作用函数的执行又会将其重新被收集到同一个集合中, 出现了一边删除该函数一边收集该函数导致死循环. 可以用另一个SSet进行遍历

function trigger (target, key) {
    const depsMap = bucket.get(target)
    if (!depsMap) return
    const effects = depsMap.get(key)
    
    const effectsToRun = new Set(effects)
    effectsToRun.forEach(fn => fn())
}