Vue2.0 $nextTick源码理解

Posted 2021-10-04 登楼痕

$nextTick的原理其实就是Js执行机制的问题，即微任务和宏任务那套东西。

同步视图更新：每次属性值改变，渲染函数的watcher会重新求值完成重新渲染，所以多个属性值修改会导致对此重新渲染。

异步视图更新：将执行更新操作的watcher放进一个队列中，并通过id避免重复放入，然后当所有突变完成后，再一次性执行队列里的观察者的更新方法，同时清空队列。

所以再wacther对象的update方法里，可以看到：

update () {
  /* istanbul ignore else */
  if (this.computed) {
    // 省略...
  } else if (this.sync) {
    this.run()
  } else {
    queueWatcher(this)
  }}

queueWatcher函数做的就是：

1、首先判断进来的watcher是否已有，没有就标记一下它的id是true；

2、当队列没有执行update时，就把新来的watcher丢到队列尾部，如果正在更新，就插入队列，保证执行顺序；

3、执行nextTick(flushSchedulerQueue)。

nextTick如何实现

Vue里的$nextTick其实就是对nextTick(fn, this)函数的封装。js的事件循环是在调用栈里拿一次宏任务执行，然后再对对应的微任务对应执行完毕，再去执行下一次宏任务。宏任务之间可能会穿插UI重渲染，所以最好是要在微任务中把要更新的数据更新了，这样只需一次UI渲染就能得到新DOM。因此使用Promise是最佳选择，讲数据更新放到微任务中，而setTimeout是宏任务。

if (typeof Promise !== 'undefined' && isNative(Promise)) {
  const p = Promise.resolve()
  microTimerFunc = () => {
    p.then(flushCallbacks)
    // in problematic UIWebViews, Promise.then doesn't completely break, but
    // it can get stuck in a weird state where callbacks are pushed into the
    // microtask queue but the queue isn't being flushed, until the browser
    // needs to do some other work, e.g. handle a timer. Therefore we can
    // "force" the microtask queue to be flushed by adding an empty timer.
    if (isios) setTimeout(noop)
  }} else {
  // fallback to macro
  microTimerFunc = macroTimerFunc
}

其实都是退而求其次的兼容写法，对于macroTimerFunc：

if (typeof setImmediate !== 'undefined' && isNative(setImmediate)) {
  macroTimerFunc = () => {
    setImmediate(flushCallbacks)
  }} else if (typeof MessageChannel !== 'undefined' && (
  isNative(MessageChannel) ||
  // PhantomJS
  MessageChannel.toString() === '[object MessageChannelConstructor]')) {
  const channel = new MessageChannel()
  const port = channel.port2
  channel.port1.onmessage = flushCallbacks
  macroTimerFunc = () => {
    port.postMessage(1)
  }} else {
  /* istanbul ignore next */
  macroTimerFunc = () => {
    setTimeout(flushCallbacks, 0)
  }}

setImmediate和MessageChannel都是优于setTimeout的，因为不需要做任何超时检测工作。

具体看一下nextTick函数的入口：

export function nextTick (cb?: Function, ctx?: Object) {
  // 省略...}

在用法上分为传cb和不传cb。

1、当传了回调函数时：

export function nextTick (cb?: Function, ctx?: Object) {
  let _resolve
  callbacks.push(() => {
    if (cb) {
      try {
        cb.call(ctx)
      } catch (e) {
        handleError(e, ctx, 'nextTick')
      }
    } else if (_resolve) {
      _resolve(ctx)
    }
  })
  if (!pending) {
    pending = true
    if (useMacroTask) {
      macroTimerFunc()
    } else {
      microTimerFunc()
    }
  }
  // 省略...
}

此时会将cb包裹到一个新的函数里，放入callbacks数组。

接着判断pending真假，表示回调队列是否处于等待刷新的状态，初始值为false，当外部调用了$nextTick后，就要刷新回调队列，将flushCallbacks函数分别注册为宏任务和微任务，都要等待调用栈清空后才会执行。

function flushCallbacks () {
  pending = false
  const copies = callbacks.slice(0)
  callbacks.length = 0
  for (let i = 0; i < copies.length; i++) {
    copies[i]()
  }}

总体来说就是把nextTick的回调函数包装起来放到callbacks里，然后通过将flushCallbacks转换成微任务或者宏任务（always优先微任务）来执行回调队列，在这个过程中，避免重复的watcher.update执行，来优化性能。如果没传cb，就返回一个Promise.resolve()来支持.then写法。