浅读vue源码(2.x)之diff算法

206 阅读11分钟

背景

  在小米实习的空闲时间,探索一下vue源码中diff的实现。

正文

在我响应式原理那片博客中,我们已经了解到,当vue实例被检测的属性改变时,会发生视图更新,即调用updateComponent函数对视图进行重新渲染。

updateComponent = () => {
  vm._update(vm._render(), hydrating)
}

在该函数中,执行vm._render()时,会去获取相关属性最新情况,从而得到一个新的Vnode。

而在vm._update()函数中,会将新节点挂载到真实DOM元素上。需要注意的是,该函数并不是简单粗暴地把旧的Vnode节点删除,再直接挂上新的Vnode节点,而是会调用diff算法,这也是vue高效更新视图的核心。

进入vm._update()函数(core/instance/lifecycle.js)

Vue.prototype._update = function (vnode: VNode, hydrating?: boolean) {
  const vm: Component = this
  const prevEl = vm.$el// vue实例挂载的真实dom节点
  const prevVnode = vm._vnode// 旧的vnode节点
  const prevActiveInstance = activeInstance
  activeInstance = vm
  vm._vnode = vnode// 新的vnode节点
  // Vue.prototype.__patch__ is injected in entry points
  // based on the rendering backend used.
  if (!prevVnode) {// 第一次将vue实例挂载到dom节点
    // initial render
    vm.$el = vm.__patch__(vm.$el, vnode, hydrating, false /* removeOnly */)
  } else {// 节点的更新,diff的起点
    // updates
    vm.$el = vm.__patch__(prevVnode, vnode)
  }
  activeInstance = prevActiveInstance
  // update __vue__ reference
  if (prevEl) {
    prevEl.__vue__ = null
  }
  if (vm.$el) {
    vm.$el.__vue__ = vm
  }
  // if parent is an HOC, update its $el as well
  if (vm.$vnode && vm.$parent && vm.$vnode === vm.$parent._vnode) {
    vm.$parent.$el = vm.$el
  }
  // updated hook is called by the scheduler to ensure that children are
  // updated in a parent's updated hook.
}

抓住问题的核心,我们只关注vm.patch()函数。

在core/platform/web/runtime/index.js,已经将Vue.prototype.patch指向patch()函数。

所以我们进入core/vdom/patch.js

return function patch (oldVnode, vnode, hydrating, removeOnly) {
  if (isUndef(vnode)) {// 新节点不存在
    if (isDef(oldVnode)) invokeDestroyHook(oldVnode)
    return
  }
  let isInitialPatch = false
  const insertedVnodeQueue = []
  if (isUndef(oldVnode)) {// 旧节点不存在
    // empty mount (likely as component), create new root element
    isInitialPatch = true
    createElm(vnode, insertedVnodeQueue)
  } else {
    const isRealElement = isDef(oldVnode.nodeType)// 只有真实dom元素才有nodeType属性
    if (!isRealElement && sameVnode(oldVnode, vnode)) {// 都为虚拟节点且同类型
      // patch existing root node
      patchVnode(oldVnode, vnode, insertedVnodeQueue, removeOnly)
    } else {...// oldVnode为真实dom节点,或两个虚拟节点不为同类型
    }
  }
  invokeInsertHook(vnode, insertedVnodeQueue, isInitialPatch)
  return vnode.elm
}

输入:新旧Vnode节点。(首次渲染oldVnode为真实dom元素)

输出:节点已挂载到的真实dom节点。

先看一下Vnode节点有哪些属性。(core/vdom/vnode.js)

constructor (
  tag?: string,
  data?: VNodeData,
  children?: ?Array<VNode>,
  text?: string,
  elm?: Node,
  context?: Component,
  componentOptions?: VNodeComponentOptions,
  asyncFactory?: Function
) {
  this.tag = tag// 标签名
  this.data = data// 数据信息
  this.children = children// 子节点
  this.text = text// 文本
  this.elm = elm// 虚拟节点挂载到的真实dom节点
  this.ns = undefined
  this.context = context
  this.fnContext = undefined
  this.fnOptions = undefined
  this.fnScopeId = undefined
  this.key = data && data.key
  this.componentOptions = componentOptions
  this.componentInstance = undefined
  this.parent = undefined
  this.raw = false
  this.isStatic = false// 是否为静态节点
  this.isRootInsert = true
  this.isComment = false// 是否为注释节点
  this.isCloned = false
  this.isOnce = false
  this.asyncFactory = asyncFactory
  this.asyncMeta = undefined
  this.isAsyncPlaceholder = false
}

然后回到patch(preVnode,vnode)函数

return function patch (oldVnode, vnode, hydrating, removeOnly) {
  if (isUndef(vnode)) {// 新节点不存在
    if (isDef(oldVnode)) invokeDestroyHook(oldVnode)
    return
  }
  let isInitialPatch = false
  const insertedVnodeQueue = []
  if (isUndef(oldVnode)) {// 旧节点不存在
    // empty mount (likely as component), create new root element
    isInitialPatch = true
    createElm(vnode, insertedVnodeQueue)
  } else {
    const isRealElement = isDef(oldVnode.nodeType)// 只有真实dom元素才有nodeType属性
    if (!isRealElement && sameVnode(oldVnode, vnode)) {// 都为虚拟节点且同类型
      // patch existing root node
      patchVnode(oldVnode, vnode, insertedVnodeQueue, removeOnly)
    } else {...// oldVnode为真实dom节点,或两个虚拟节点不为同类型
    }
  }
  invokeInsertHook(vnode, insertedVnodeQueue, isInitialPatch)
  return vnode.elm
}

整个函数,针对新旧节点,分为四种情况:

1.当新节点不存在:

若旧节点存在,则销毁旧节点,否则直接返回。

2.当旧节点不存在:

直接插入新节点。

3.当新旧节点都为虚拟节点且同类型:

调用parchVnode()函数,更新节点。

4.oldVnode为真实dom节点,或两个虚拟节点不为同类型:

直接挂载新节点,或销毁旧节点,插入新节点。

而diff算法的高效,在于第三种情况,即节点的更新

function sameVnode (a, b) {
  return (
    a.key === b.key && (
      (
        a.tag === b.tag &&
        a.isComment === b.isComment &&
        isDef(a.data) === isDef(b.data) &&
        sameInputType(a, b)
      ) || (
        isTrue(a.isAsyncPlaceholder) &&
        a.asyncFactory === b.asyncFactory &&
        isUndef(b.asyncFactory.error)
      )
    )
  )
}
function sameInputType (a, b) {
  if (a.tag !== 'input') return true
  let i
  const typeA = isDef(i = a.data) && isDef(i = i.attrs) && i.type
  const typeB = isDef(i = b.data) && isDef(i = i.attrs) && i.type
  return typeA === typeB || isTextInputType(typeA) && isTextInputType(typeB)
}

通过 sameVnode() 函数来判断是否是同一类型:即只有当 key、 tag、 isComment(是否为注释节点)、 data 同时定义(或不定义),同时满足当标签类型为 input 的时候 type 相同的情况。

接下来看看关键的patchVnode()函数。

function patchVnode (oldVnode, vnode, insertedVnodeQueue, removeOnly) {
  if (oldVnode === vnode) {// 新旧节点完全相同
    return
  }
  const elm = vnode.elm = oldVnode.elm// 要被挂载到的DOM节点
  if (isTrue(oldVnode.isAsyncPlaceholder)) {// 异步占位
    if (isDef(vnode.asyncFactory.resolved)) {
      hydrate(oldVnode.elm, vnode, insertedVnodeQueue)
    } else {
      vnode.isAsyncPlaceholder = true
    }
    return
  }
  // reuse element for static trees.
  // note we only do this if the vnode is cloned -
  // if the new node is not cloned it means the render functions have been
  // reset by the hot-reload-api and we need to do a proper re-render.
  if (isTrue(vnode.isStatic) &&// 新节点为静态节点
    isTrue(oldVnode.isStatic) &&// 旧节点为静态节点
    vnode.key === oldVnode.key &&// 两者key相同
    (isTrue(vnode.isCloned) || isTrue(vnode.isOnce))// 新节点为克隆节点或有v-once属性
  ) {
    vnode.componentInstance = oldVnode.componentInstance
    return
  }
  let i
  const data = vnode.data
  if (isDef(data) && isDef(i = data.hook) && isDef(i = i.prepatch)) {
    i(oldVnode, vnode)
  }
  const oldCh = oldVnode.children
  const ch = vnode.children
  if (isDef(data) && isPatchable(vnode)) {
    for (i = 0; i < cbs.update.length; ++i) cbs.update[i](oldVnode, vnode)
    if (isDef(i = data.hook) && isDef(i = i.update)) i(oldVnode, vnode)
  }
  if (isUndef(vnode.text)) // 新节点不为文本节点
    if (isDef(oldCh) && isDef(ch)) {// 新旧节点的子节点都存在
      if (oldCh !== ch) updateChildren(elm, oldCh, ch, insertedVnodeQueue, removeOnly)
    } else if (isDef(ch)) {// 只有新节点的子节点存在
      if (isDef(oldVnode.text)) nodeOps.setTextContent(elm, '')
      addVnodes(elm, null, ch, 0, ch.length - 1, insertedVnodeQueue)
    } else if (isDef(oldCh)) {// 只有旧节点的子节点存在
      removeVnodes(elm, oldCh, 0, oldCh.length - 1)
    } else if (isDef(oldVnode.text)) {
      nodeOps.setTextContent(elm, '')
    }
  } else if (oldVnode.text !== vnode.text) {// 新节点为本文节点
    nodeOps.setTextContent(elm, vnode.text)
  }
  if (isDef(data)) {
    if (isDef(i = data.hook) && isDef(i = i.postpatch)) i(oldVnode, vnode)
  }
}

看下来,该函数针对新旧节点的情况分为以下几个处理分支:

1.新旧节点完全相同,不处理。

2.旧节点为异步占位,不处理。

3.新旧节点为静态节点,且两者的key完全相同,且新节点为克隆节点或包含v-once属性,将旧节点的componentInstance函数赋给新节点。

4.当新节点为文本节点时,直接把文本挂载到dom元素上。

5.当新节点不为文本节点时:

(1)新旧节点的子节点都不存在,若旧节点为文本节点,则清空dom元素的文本。

(2)只有旧节点的子节点存在,则清除旧节点的所有子节点。

(3)只有新节点的子节点存在,清空旧节点的文本,插入新节点的子节点。

(4)当新旧节点的子节点都存在且不等,调用updateChildren()函数。

接下来看看updateChildren()函数,也是整个更新节点操作的精彩部分。

function updateChildren (parentElm, oldCh, newCh, insertedVnodeQueue, removeOnly) {
  let oldStartIdx = 0
  let newStartIdx = 0
  let oldEndIdx = oldCh.length - 1
  let oldStartVnode = oldCh[0]
  let oldEndVnode = oldCh[oldEndIdx]
  let newEndIdx = newCh.length - 1
  let newStartVnode = newCh[0]
  let newEndVnode = newCh[newEndIdx]
  let oldKeyToIdx, idxInOld, vnodeToMove, refElm
  // removeOnly is a special flag used only by <transition-group>
  // to ensure removed elements stay in correct relative positions
  // during leaving transitions
  const canMove = !removeOnly
  if (process.env.NODE_ENV !== 'production') {
    checkDuplicateKeys(newCh)
  }
  while (oldStartIdx <= oldEndIdx && newStartIdx <= newEndIdx) {
    if (isUndef(oldStartVnode)) {
      oldStartVnode = oldCh[++oldStartIdx] // Vnode has been moved left
    } else if (isUndef(oldEndVnode)) {
      oldEndVnode = oldCh[--oldEndIdx]
    } else if (sameVnode(oldStartVnode, newStartVnode)) {
      patchVnode(oldStartVnode, newStartVnode, insertedVnodeQueue)
      oldStartVnode = oldCh[++oldStartIdx]
      newStartVnode = newCh[++newStartIdx]
    } else if (sameVnode(oldEndVnode, newEndVnode)) {
      patchVnode(oldEndVnode, newEndVnode, insertedVnodeQueue)
      oldEndVnode = oldCh[--oldEndIdx]
      newEndVnode = newCh[--newEndIdx]
    } else if (sameVnode(oldStartVnode, newEndVnode)) { // Vnode moved right
      patchVnode(oldStartVnode, newEndVnode, insertedVnodeQueue)
      canMove && nodeOps.insertBefore(parentElm, oldStartVnode.elm, nodeOps.nextSibling(oldEndVnode.elm))
      oldStartVnode = oldCh[++oldStartIdx]
      newEndVnode = newCh[--newEndIdx]
    } else if (sameVnode(oldEndVnode, newStartVnode)) { // Vnode moved left
      patchVnode(oldEndVnode, newStartVnode, insertedVnodeQueue)
      canMove && nodeOps.insertBefore(parentElm, oldEndVnode.elm, oldStartVnode.elm)
      oldEndVnode = oldCh[--oldEndIdx]
      newStartVnode = newCh[++newStartIdx]
    } else {
      if (isUndef(oldKeyToIdx)) oldKeyToIdx = createKeyToOldIdx(oldCh, oldStartIdx, oldEndIdx)
      idxInOld = isDef(newStartVnode.key)
        ? oldKeyToIdx[newStartVnode.key]
        : findIdxInOld(newStartVnode, oldCh, oldStartIdx, oldEndIdx)
      if (isUndef(idxInOld)) { // New element
        createElm(newStartVnode, insertedVnodeQueue, parentElm, oldStartVnode.elm, false, newCh, newStartIdx)
      } else {
        vnodeToMove = oldCh[idxInOld]
        if (sameVnode(vnodeToMove, newStartVnode)) {
          patchVnode(vnodeToMove, newStartVnode, insertedVnodeQueue)
          oldCh[idxInOld] = undefined
          canMove && nodeOps.insertBefore(parentElm, vnodeToMove.elm, oldStartVnode.elm)
        } else {
          // same key but different element. treat as new element
          createElm(newStartVnode, insertedVnodeQueue, parentElm, oldStartVnode.elm, false, newCh, newStartIdx)
        }
      }
      newStartVnode = newCh[++newStartIdx]
    }
  }
  if (oldStartIdx > oldEndIdx) {
    refElm = isUndef(newCh[newEndIdx + 1]) ? null : newCh[newEndIdx + 1].elm
    addVnodes(parentElm, refElm, newCh, newStartIdx, newEndIdx, insertedVnodeQueue)
  } else if (newStartIdx > newEndIdx) {
    removeVnodes(parentElm, oldCh, oldStartIdx, oldEndIdx)
  }
}

很长,但我们逐步分析。

let oldStartIdx = 0
let newStartIdx = 0
let oldEndIdx = oldCh.length - 1
let oldStartVnode = oldCh[0]
let oldEndVnode = oldCh[oldEndIdx]
let newEndIdx = newCh.length - 1
let newStartVnode = newCh[0]
let newEndVnode = newCh[newEndIdx]
let oldKeyToIdx, idxInOld, vnodeToMove, refElm

首先我们定义 oldStartIdx,oldEndIdx,newStartIdx,newEndIdx 为新旧子节点序列的首尾索引,oldStartVnode,oldEndVnode,newStartVnode,newEndVnode 为上四个索引对应的真实节点。

image.png

接着,进入while循环,终止条件是oldStartIdx > oldEndIdx 或 newStartIdx > newEndIdx。

if (isUndef(oldStartVnode)) {
  oldStartVnode = oldCh[++oldStartIdx] // Vnode has been moved left
} else if (isUndef(oldEndVnode)) {
  oldEndVnode = oldCh[--oldEndIdx]


image.png

当 oldStartVnode 或者 oldEndVnode 不存在的时候,oldStartIdx 与 oldEndIdx 继续向中间靠拢,并更新对应的 oldStartVnode 与 oldEndVnode 的指向。

else if (sameVnode(oldStartVnode, newStartVnode)) {
  patchVnode(oldStartVnode, newStartVnode, insertedVnodeQueue)
  oldStartVnode = oldCh[++oldStartIdx]
  newStartVnode = newCh[++newStartIdx]
} else if (sameVnode(oldEndVnode, newEndVnode)) {
  patchVnode(oldEndVnode, newEndVnode, insertedVnodeQueue)
  oldEndVnode = oldCh[--oldEndIdx]
  newEndVnode = newCh[--newEndIdx]


image.png

image.png

头节点相同,再次调用patchVnode()函数,对应索引后移。

尾节点相同,再次调用patchVnode()函数,对应索引前移。

else if (sameVnode(oldStartVnode, newEndVnode)) { // Vnode moved right
  patchVnode(oldStartVnode, newEndVnode, insertedVnodeQueue)
  canMove && nodeOps.insertBefore(parentElm, oldStartVnode.elm, nodeOps.nextSibling(oldEndVnode.elm))
  oldStartVnode = oldCh[++oldStartIdx]
  newEndVnode = newCh[--newEndIdx]


image.png

当旧节点的子节点序列头节点与新节点的子节点序列尾节点相同时,首先再次调用 patchVnode() 函数,接着将当前的 oldStartVnode 插入到 oldEndVnode 后面,最后移动相关的两个索引。

else if (sameVnode(oldEndVnode, newStartVnode)) { // Vnode moved left
  patchVnode(oldEndVnode, newStartVnode, insertedVnodeQueue)
  canMove && nodeOps.insertBefore(parentElm, oldEndVnode.elm, oldStartVnode.elm)
  oldEndVnode = oldCh[--oldEndIdx]
  newStartVnode = newCh[++newStartIdx]


image.png

当旧节点的子节点序列尾节点与新节点的子节点序列首节点相同时,首先再次调用 patchVnode() 函数,接着将当前的 oldEndVnode 插入到oldStartVnode 前面,最后移动相关的两个索引。

} else {
  if (isUndef(oldKeyToIdx)) oldKeyToIdx = createKeyToOldIdx(oldCh, oldStartIdx, oldEndIdx)
  idxInOld = isDef(newStartVnode.key)
    ? oldKeyToIdx[newStartVnode.key]
    : findIdxInOld(newStartVnode, oldCh, oldStartIdx, oldEndIdx)
  if (isUndef(idxInOld)) { // New element
    createElm(newStartVnode, insertedVnodeQueue, parentElm, oldStartVnode.elm, false, newCh, newStartIdx)
  } else {
    vnodeToMove = oldCh[idxInOld]
    if (sameVnode(vnodeToMove, newStartVnode)) {
      patchVnode(vnodeToMove, newStartVnode, insertedVnodeQueue)
      oldCh[idxInOld] = undefined
      canMove && nodeOps.insertBefore(parentElm, vnodeToMove.elm, oldStartVnode.elm)
    } else {
      // same key but different element. treat as new element
      createElm(newStartVnode, insertedVnodeQueue, parentElm, oldStartVnode.elm, false, newCh, newStartIdx)
    }
  }
  newStartVnode = newCh[++newStartIdx]
}

当以上情况都不满足时,会进入该 else 分支。

首先进入 createKeyToOldIdx() 函数:

function createKeyToOldIdx (children, beginIdx, endIdx) {
  let i, key
  const map = {}
  for (i = beginIdx; i <= endIdx; ++i) {
    key = children[i].key
    if (isDef(key)) map[key] = i
  }
  return map
}

返回一个关于当前旧节点子节点序列的,key 与 index 索引对应的一个 map 表。

例如:

[
  {xx: xx, key: 'key0'},
  {xx: xx, key: 'key1'},
  {xx: xx, key: 'key2'}
]

在经过 createKeyToOldIdx 转化以后会变成:

{
  key0: 0,
  key1: 1,
  key2: 2
}

现在,我们就可以先分析以下代码了:

if (isUndef(oldKeyToIdx)) oldKeyToIdx = createKeyToOldIdx(oldCh, oldStartIdx, oldEndIdx)// 提供一个map结构
idxInOld = isDef(newStartVnode.key)// 找到当前newStartVnode的key在oldCh中对应的索引
  ? oldKeyToIdx[newStartVnode.key]
  : findIdxInOld(newStartVnode, oldCh, oldStartIdx, oldEndIdx)// 对应newStartVnode的key不存在的情况

先创建一个key与index索引的 map 表,找到当前 newStartVnode 的 key 在旧子节点序列 oldCh 上对应的索引。

注:findIdxInold()函数用来应对 newStartVnode 的 key 不存在的情况,此函数中会根据 newStartVnode 对当前旧节点的子节点序列再做一次遍历。代码如下:

function findIdxInOld (node, oldCh, start, end) {
  for (let i = start; i < end; i++) {
    const c = oldCh[i]
    if (isDef(c) && sameVnode(node, c)) return i
  }
}


if (isUndef(idxInOld)) { // New element
  createElm(newStartVnode, insertedVnodeQueue, parentElm, oldStartVnode.elm, false, newCh, newStartIdx)
}

如果 idxInOld 不存在,说明在当前的 newStartVnode 是新增的,此时根据 newStartVnode 创建一个新节点插入到 oldStartVnode 前面。(最后会将 newStartIdx 索引后移)

} else {
  vnodeToMove = oldCh[idxInOld]
  if (sameVnode(vnodeToMove, newStartVnode)) {
    patchVnode(vnodeToMove, newStartVnode, insertedVnodeQueue)
    oldCh[idxInOld] = undefined
    canMove && nodeOps.insertBefore(parentElm, vnodeToMove.elm, oldStartVnode.elm)
  } else {
    // same key but different element. treat as new element
    createElm(newStartVnode, insertedVnodeQueue, parentElm, oldStartVnode.elm, false, newCh, newStartIdx)
  }
}
newStartVnode = newCh[++newStartIdx]

如果 idxInOld 存在,比较新旧节点:

(1)当两者是同类型节点:

image.png

首先,再次调用 patchVnode(),接着,销毁 idxInOld 指向的节点即 vnodeToMove ,最后,将 vnodeToMove 插入到 oldStartVnode 前面。(最后会将 newStartIdx 索引后移)

(2)当两者不是同类型节点:

根据newStartVnode创建一个新节点插入到oldStartVnode前面。(最后会将 newStartIdx 索引后移)

最后,我们来到while循环结束后的收尾阶段。

if (oldStartIdx > oldEndIdx) {
  refElm = isUndef(newCh[newEndIdx + 1]) ? null : newCh[newEndIdx + 1].elm
  addVnodes(parentElm, refElm, newCh, newStartIdx, newEndIdx, insertedVnodeQueue)


image.png

如果 oldStartIdx > oldEndIdx,说明老节点比对完了,但是新节点还有多的,需要将新节点插入到真实 DOM 中去,调用 addVnodes 将这些节点插入即可。

} else if (newStartIdx > newEndIdx) {
  removeVnodes(parentElm, oldCh, oldStartIdx, oldEndIdx)
}


image.png

如果 newStartIdx > newEndIdx 条件,说明新节点比对完了,老节点还有多,将这些无用的老

节点通过 removeVnodes 批量删除即可。

至此,我们完成了对diff算法的解析,再从头梳理一下:

首先,diff算法的起点是 patch() 函数,用于比较新旧节点。针对比对的不同情况,只有当节点为同一类型时,会触发节点的更新,即patchVnode() 函数(其他情况为粗暴的增删)。

接着,在 patchVnode() 函数中,针对节点的比对情况,只有当新节点不为文本节点且包含子节点时,会触发对两者子节点的比对,即updateChildren() 函数(其他情况为不处理,修改元素文本和清除元素文本)

最后,在 updateChildren() 函数中,针对新旧节点的子节点序列的比对情况,做出不同的处理。例如,旧节点的移动,删除和新节点的插入。

到此为止,我们已经研究了vue三大核心中的两个,即响应式原理和diff算法,而剩下的模板编译,将会涉及到编译原理的知识,着实复杂,等有时间再去慢慢研究。但对框架的探索,不应该只停留在源码阅读,我将会实现一个简易版的vue放到github上,为这两个星期对vue源码的探索画上句号。