深度解析：Vue3如何巧妙的实现强大的computed

前言

Vue中的computed是一个很是强大的功能，在computed函数中访问到的值改变了后，computed的值也会自动改变。

Vue2中的实现是利用了Watcher的嵌套收集，渲染watcher收集到computed watcher做为依赖，computed watcher又收集到响应式数据某个属性做为依赖，这样在响应式数据某个属性发生改变时，就会按照 响应式属性 -> computed值更新 -> 视图渲染这样的触发链触发过去，若是对Vue2中的原理感兴趣，能够看我这篇文章的解析：

手把手带你实现一个最精简的响应式系统来学习Vue的data、computed、watch源码

前置知识

阅读本文须要你先学习Vue3响应式的基本原理，能够先看个人这篇文章，原理和Vue3是一致的： 带你完全搞懂Vue3的Proxy响应式原理！TypeScript从零实现基于Proxy的响应式库。

在你拥有了一些前置知识之后，默认你应该知道的是：

1. effect其实就是一个依赖收集函数，在它内部访问了响应式数据，响应式数据就会把这个effect函数做为依赖收集起来，下次响应式数据改了就触发它从新执行。

2. reactive返回的就是个响应式数据，这玩意能够和effect搭配使用。

举个简单的栗子吧：

// 响应式数据
const data = reactive({ count: 0 })
// 依赖收集
effect(() => console.log(data.count))
// 触发上面的effect从新执行
data.count ++
复制代码

就这个例子来讲，data是一个响应式数据。

effect传入的函数由于内部访问到它上面的属性count了，

因此造成了一个count -> effect的依赖。

下次count改变了，这个effect就会从新执行，就这么简单。

computed

那么引入本文中的核心概念，computed来改写这个例子后呢：

// 1. 响应式数据
const data = reactive({ count: 0 })
// 2. 计算属性
const plusOne = computed(() => data.count + 1)
// 3. 依赖收集
effect(() => console.log(plusOne.value))
// 4. 触发上面的effect从新执行
data.count ++
复制代码

这样的例子也能跑通，为何data.count的改变能间接触发访问了计算属性的effect的从新执行呢？

咱们来配合单点调试一步步解析。

简化版源码

首先看一下简化版的computed的代码：

export function computed( getter ) {
  let dirty = true
  let value: T

  // 这里仍是利用了effect作依赖收集
  const runner = effect(getter, {
    // 这里保证初始化的时候不去执行getter
    lazy: true,
    computed: true,
    scheduler: () => {
      // 在触发更新时 只是把dirty置为true 
      // 而不去马上计算值 因此计算属性有lazy的特性
      dirty = true
    }
  })
  return {
    get value() {
      if (dirty) {
        // 在真正的去获取计算属性的value的时候
        // 依据dirty的值决定去不去从新执行getter 获取最新值
        value = runner()
        dirty = false
      }
      // 这里是关键 后续讲解
      trackChildRun(runner)
      return value
    },
    set value(newValue: T) {
      setter(newValue)
    }
  }
}
复制代码

能够看到，computed其实也是一个effect。这里对闭包进行了巧妙的运用，注释里的几个关键点决定了计算属性拥有懒加载的特征，你不去读取value的时候，它是不会去真正的求值的。

前置准备

首先要知道，effect函数会当即开始执行，再执行以前，先把effect自身变成全局的activeEffect，以供响应式数据收集依赖。

而且activeEffect的记录是用栈的方式，随着函数的开始执行入栈，随着函数的执行结束出栈，这样就能够维护嵌套的effect关系。

先起几个别名便于讲解

// 计算effect
computed(() => data.count + 1)
// 日志effect
effect(() => console.log(plusOne.value))
复制代码

从依赖关系来看，
日志effect读取了计算effect
计算effect读取了响应式属性count
因此更新的顺序也应该是：
count改变 -> 计算effect更新 -> 日志effect更新

那么这个关系链是如何造成的呢

单步解读

在日志effect开始执行的时候，

⭐⭐
此时activeEffect是日志effect

此时的effectStack是[ 日志effect ]
⭐⭐

plusOne.value的读取，触发了

get value() {
      if (dirty) {
        // 在真正的去获取计算属性的value的时候
        // 依据dirty的值决定去不去从新执行getter 获取最新值
        value = runner()
        dirty = false
      }
      // 这里是关键 后续讲解
      trackChildRun(runner)
      return value
},
复制代码

runner就是计算effect，进入了runner之后
⭐⭐
此时activeEffect是计算effect

此时的effectStack是[ 日志effect, 计算effect ]
⭐⭐
computed(() => data.count + 1)日志effect会去读取count，触发了响应式数据的get拦截：

此时count会收集计算effect做为本身的依赖。

而且计算effect会收集count的依赖集合，保存在本身身上。(经过effect.deps属性)

dep.add(activeEffect)
activeEffect.deps.push(dep)
复制代码

也就是造成了一个双向收集的关系，

计算effect存了count的全部依赖，count也存了计算effect的依赖。

而后在runner运行结束后，计算effect出栈了，此时activeEffect变成了栈顶的日志effect

⭐⭐
此时activeEffect是日志effect

此时的effectStack是[ 日志effect ]
⭐⭐

接下来进入关键的步骤：trackChildRun

trackChildRun(runner)  

function trackChildRun(childRunner: ReactiveEffect) {
  for (let i = 0; i < childRunner.deps.length; i++) {
    const dep = childRunner.deps[i]
    dep.add(activeEffect)
  }
}
复制代码

这个runner就是计算effect，它的deps上此时挂着count的依赖集合，

在trackChildRun中，它把当前的acctiveEffect也就是日志effect也加入到了count的依赖集合中。

此时count的依赖集合是这样的：[ 计算effect, 日志effect ]

这样下次count更新的时候，会把两个effect都从新触发，而因为触发的顺序是先触发computed effect 后触发普通effect，所以就完成了

1. 计算effect的dirty置为true，标志着下次读取须要从新求值。
2. 日志effect读取计算effect的value，得到最新的值并打印出来。

总结

不得不认可，computed这个强大功能的实现果真少不了内部很是复杂的实现，这个双向依赖收集的套路相信也会给各位小伙伴带来很大的启发。跟着尤大学习，果真有肉吃！

另外因为@vue/reactivity的框架无关性，我把它整合进了React，作了一个状态管理库，能够完整的使用上述的computed等强大的Vue3能力。

react-composition-api

有兴趣的小伙伴也能够看一下，star一下！