深入解析 Vue 的热更新原理，尤大是如何巧用源码中的细节？

本文转载自微信公众号「前端从进阶到入院 」，作者ssh前端 。转载本文请联系前端从进阶到入院公众号。 

大家都用过 Vue-CLI 创建 vue 应用，在开发的时候我们修改了 vue 文件，保存了文件，浏览器上就自动更新出我们写的组件内容，非常的顺滑流畅，大大提高了开发效率。想知道这背后是怎么实现的吗，其实代码并不复杂。

这个功能的实现底层用了vue-hot-load-api[1]这个库，得益于 vue 的良好设计，热更新的实现总共就一个 js 文件，200 行代码，绰绰有余。

而在这个库里涉及到的技巧又非常适合我们去深入了解 vue 内部的一些机制，所以赶快来和我一起学习吧。

提要

本文单纯的从vue-hot-load-api这个库出发，在浏览器的环境运行 Vue 的热更新示例，主要测试的组件是普通的 vue 组件而不是 functional 等特殊组件，以最简单的流程搞懂热更新的原理。

在源码解析中贴出的代码会省略掉一些不太相关的流程，更便于理解。

解析

从 github 仓库示例入手

进入了这个 github 仓库以后，最先开始看的肯定是 Readme 的里的示例，在看示例的时候作者给出的注释就非常重要了，他会标注出每一个重要的环节。并且我们要结合自己的一些经验排除掉和这个库无关的代码。(在这个示例中，webpack 的相关代码就可以先不去过多关注)

第一步需要调用install方法，传入 Vue 构造函数，根据注释来看，这一步是要知道这个库与 Vue 版本之间是否兼容。

// make the API aware of the Vue that you are using. 
// also checks compatibility. 
api.install(Vue); 

接下来的这段注释告诉我们，每个需要热更新的组件选项对象，我们都需要为它建立一个独一无二的 id，并且这段代码需要在初始化的时候完成。

if (初始化) { 
  // for each component option object to be hot-reloaded, 
  // you need to create a record for it with a unique id. 
  // do this once on startup. 
  api.createRecord('very-unique-id', myComponentOptions); 
} 

最后就是激动人心的热更新时间了，

根据注释来看，这个库的使用分为两种情况。

• rerender 只有 template 或者 render 函改变的情况下使用。
• reload 如果 template 或者 render 未改变，则这个函数需要调用 reload 方法先销毁然后重新创建(包括它的子组件)。

// if a component has only its template or render function changed, 
// you can force a re-render for all its active instances without 
// destroying/re-creating them. This keeps all current app state intact. 
api.rerender('very-unique-id', myComponentOptions); 
 
// --- OR --- 
 
// if a component has non-template/render options changed, 
// it needs to be fully reloaded. This will destroy and re-create all its 
// active instances (and their children). 
api.reload('very-unique-id', myComponentOptions); 

从这个简单的示例里面可以看出，这个库的核心流程就是：

1. api.install 检测兼容性。
2. api.createRecord 为组件对象通过一个独一无二的 id 建立一个记录。
3. api.rerender 或api.reload 进行组件的热更新。

什么，Readme 的示例到此就结束了?这个 very-unique-id 到底是个什么东西，myComponentOptions 又是什么样的。

因为这个仓库可能并不是面向广大开发者的，所以它的文档写的非常的简略。其实看完了这个简短的示例，大家肯定还是一脸懵逼的。

在看一个你没有熟练使用的库的源码的时候，其实还有一个很关键的步骤，那就是看测试用例。

探索测试用例

测试用例[2]

上面我们总结出两个关键 api rerender 和 reload 之后，就带着目的性的去看测试用例。

const Vue = require('vue'); 
const api = require('../src'); 
 
// 初始化 
api.install(Vue); 
 
// 这个方法接受id和组件选项对象， 
// 通过createRecord去记录组件 
// 然后返回一个vue组件实例。 
function prepare(id, Comp) { 
  api.createRecord(id, Comp); 
  return new Vue({ 
    render: h => h(Comp), 
  }); 
} 

rerender 用例

const id0 = 'rerender: mounted'; 
test(id0, done => { 
  // 用'rerender: mounted'作为这个组件对象的id， 
  // 这个组件的内容应该是 <div>foo</div> 
  // 调用 $mount生成dom节点 
  const app = prepare(id0, { 
    render: h => h('div', 'foo'), 
  }).$mount(); 
 
  // $el就是组件生成的dom元素，期望textContent文字内容为foo 
  expect(app.$el.textContent).toBe('foo'); 
 
  // rerender 后dom节点变成 <div>bar</div> 
  api.rerender(id0, { 
    render: h => h('div', 'bar'), 
  }); 
 
  // 通过nextTick保证dom节点已经更新 
  // 期望textContent文字内容为bar 
  Vue.nextTick(() => { 
    expect(app.$el.textContent).toBe('bar'); 
    done(); 
  }); 
}); 

const id1 = 'reload: mounted'; 
test(id1, done => { 
  // 通过一个count来计数 
  let count = 0; 
 
  // app组件会在created的时候让count + 1 
  // destroyed的时候让count - 1 
  const app = prepare(id1, { 
    created() { 
      count++; 
    }, 
    destroyed() { 
      count--; 
    }, 
    data: () => ({ msg: 'foo' }), 
    render(h) { 
      return h('div', this.msg); 
    }, 
  }).$mount(); 
  // 确保内容正确 
  expect(app.$el.textContent).toBe('foo'); 
  // 确保created周期执行 此时的count是1 
  expect(count).toBe(1); 
 
  // 调用created 传入新组件的created时 count会-1 
  api.reload(id1, { 
    created() { 
      count--; 
    }, 
    data: () => ({ msg: 'bar' }), 
    render(h) { 
      return h('div', this.msg); 
    }, 
  }); 
 
  Vue.nextTick(() => { 
    // 确保内容正确 
    expect(app.$el.textContent).toBe('bar'); 
    // 在reload之前 count是1 
    // 调用reload之后 会先调用前一个组件的destory生命周期 此时count是0 
    // 接下来调用新组建的created生命周期 此时count是-1 
    expect(count).toBe(-1); 
    done(); 
  }); 
}); 

具体流程已经在注释里分析了，果然和示例代码的注释里写的一样，而且现在我们也更清楚这个 api 到底该怎么用了。

总结一个最简单的可用 demo

import api from 'vue-hot-reload-api'; 
import Vue from 'vue'; 
 
// 初始化 
api.install(Vue, true); 
 
const appOptions = { 
  render: h => h('div', 'foo'), 
}; 
 
api.createRecord('my-app', appOptions); 
 
new Vue(appOptions).$mount('#app'); 
 
setTimeout(() => { 
  api.rerender('my-app', { 
    render: h => h('div', 'bar'), 
  }); 
}, 2000); 

这个 demo(源码[3])是直接在浏览器可用的，效果如下：

源码分析

源码地址[4]

全局变量

进入 js 文件的入口，首先定义了一些变量

// Vue构造函数 
let Vue; // late bind 
// Vue版本 
let version; 
// createRecord方法保存id -> 组件映射关系的对象 
const map = Object.create(null); 
if (typeof window !== 'undefined') { 
  // 将map对象存储在window上 
  window.__VUE_HOT_MAP__ = map; 
} 
// 是否已经安装过 
let installed = false; 
// 这个变量暂时没用 
let isBrowserify = false; 
// 初始化生命周期的名字 默认是Vue的beforeCreate生命周期 
let initHookName = 'beforeCreate'; 

其实看到 window 对象的出现，我们就已经可以确定这个 api 可以在浏览器端调用。

install

exports.install = function(vue, browserify) { 
  // 如果安装过了就不再重复安装 
  if (installed) { 
    return; 
  } 
  installed = true; 
 
  // 兼容es modules模块 
  Vue = vue.__esModule ? vue.default : vue; 
  // 把vue的版本如2.6.3分隔成[2, 6, 3] 这样的数组 
  version = Vue.version.split('.').map(Number); 
  isBrowserify = browserify; 
 
  // compat with < 2.0.0-alpha.7 
  // 兼容2.0.0-alpha.7以下版本 
  if (Vue.config._lifecycleHooks.indexOf('init') > -1) { 
    initHookName = 'init'; 
  } 
 
  // 只有Vue在2.0以上的版本才支持这个库。 
  exports.compatible = version[0] >= 2; 
  if (!exports.compatible) { 
    console.warn( 
      '[HMR] You are using a version of vue-hot-reload-api that is ' + 
        'only compatible with Vue.js core ^2.0.0.' 
    ); 
    return; 
  } 
}; 

可以看出 install 方法很简单，就是帮你看一下 Vue 的版本是否在 2.0 以上，确认一下兼容性，关于初始化生命周期，在这篇文章里我们就不考虑 2.0.0-alpha.7 以下版本，可以认为这个库的初始化工作就是在 beforeCreate 这个生命周期进行。

createRecord

/** 
 * Create a record for a hot module, which keeps track of its constructor 
 * and instances 
 * 
 * @param {String} id 
 * @param {Object} options 
 */ 
 
exports.createRecord = function(id, options) { 
  // 如果已经存储过了就return 
  if (map[id]) { 
    return; 
  } 
 
  // 关键流程 下一步解析 
  makeOptionsHot(id, options); 
 
  // 将记录存储在map中 
  // instances变量应该不难猜出是vue的实例对象。 
  map[id] = { 
    options: options, 
    instances: [], 
  }; 
}; 

这一步在把 id 和对应的 options 对象存进 map 后，就没做啥了，关键步骤肯定在于makeOptionsHot这个方法。

/** 
 * Make a Component options object hot. 
 * 让一个组件对象变得性感...哦不，是支持热更新。 
 * 
 * @param {String} id 
 * @param {Object} options 
 */ 
 
function makeOptionsHot(id, options) { 
  // options 就是我们传入的组件对象 
  // initHookName 就是'beforeCreate' 
  injectHook(options, initHookName, function() { 
    // 这个函数会在beforeCreate声明周期执行 
    const record = map[id]; 
    if (!record.Ctor) { 
      // 此时this已经是vue的实例对象了 
      // 把组件实例的构造函数赋值给record的Ctor属性。 
      record.Ctor = this.constructor; 
    } 
    // 在instances里存储这个实例。 
    record.instances.push(this); 
  }); 
  // 在组件销毁的时候把上面存储的instance删除掉。 
  injectHook(options, 'beforeDestroy', function() { 
    const instances = map[id].instances; 
    instances.splice(instances.indexOf(this), 1); 
  }); 
} 
 
// 往生命周期里注入某个方法 
function injectHook(options, name, hook) { 
  const existing = options[name]; 
  options[name] = existing 
    ? Array.isArray(existing) 
      ? existing.concat(hook) 
      : [existing, hook] 
    : [hook]; 
} 

看完了这几个函数以后，我们对 createRecord 应该有个清晰的认识了。

比如上面我们的例子中这段代码

const appOptions = { 
  render: h => h('div', 'foo'), 
}; 
 
api.createRecord('my-app', appOptions); 

在 map 中创建一个记录，这个记录有options字段也就是上面传入的组件对象，还有instances用于记录活动组件的实例，Ctor用来记录组件的构造函数。

// map 
{ 
    my-app: { 
        options: appOptions, 
        instances: [], 
        Ctor: null 
    } 
} 

在 appOptions 中，混入生命周期方法 beforeCreate，在组件的这个生命周期中，把组件自身的示例 push 到 map 里对应 instances 数组中，并且记录自己的构造函数在 Ctor 字段上。 beforeCreate 执行完了以后的 map 对象长这样。

接下来进入关键的 rerender 函数。

rerender

exports.rerender = (id, options) => { 
  const record = map[id]; 
  if (!options) { 
    // 如果没传第二个参数 就把所有实例调用 $forceUpdate 
    record.instances.slice().forEach(instance => { 
      instance.$forceUpdate(); 
    }); 
    return; 
  } 
  record.instances.slice().forEach(instance => { 
    // 将实例上的 $options上的render直接替换为新传入的render函数 
    instance.$options.render = options.render; 
    // 执行 $forceUpdate更新视图 
    instance.$forceUpdate(); 
  }); 
}; 

其实这个原函数很长，但是简化以后核心的更改视图的方法就是这些，平常我们在写 vue 单文件组件的时候都会像下面这样写：

<template> 
    <span>{{ msg }}</span> 
</template> 
 
<script> 
export default { 
  data() { 
      return { 
          msg: 'Hello World' 
      } 
  } 
} 
</script> 

这样的.vue 文件，会被 vue-loader 编译成单个的组件选项对象，template 中的部分会被编译成 render 函数挂到组件上，最终生成的对象是类似于：

export default { 
  data() { 
    return { 
      msg: 'Hello World', 
    }; 
  }, 
  render(h) { 
    return h('span', this.msg); 
  }, 
}; 

而在运行时，组件实例(也就是生命周期或者 methods 中访问的 this 对象)会通过option.render 去实现的。我们可以去 vue 的源码里验证一下我们的猜想。

_render

而在options.render 给替换成新的 render 方法了，这个时候再调用$forceUpdate，不就渲染新传入的 render 了吗?这个运行时的偷天换日我不得不佩服~

reload

reload 的讲解我们基于这样一个示例：

一开始会显示 foo 的文本，一秒以后会显示成 bar。

function prepare(id, Comp) { 
  api.createRecord(id, Comp); 
  return new Vue({ 
    render: h => h(Comp), 
  }); 
} 
 
const id1 = 'reload: mounted'; 
const app = prepare(id1, { 
  data: () => ({ msg: 'foo' }), 
  render(h) { 
    return h('div', this.msg); 
  }, 
}).$mount('#app'); 
 
// reload 
setTimeout(() => { 
  api.reload(id1, { 
    data: () => ({ msg: 'bar' }), 
    render(h) { 
      return h('div', this.msg); 
    }, 
  }); 
}, 1000); 

reload 的情况会更加复杂，涉及到很多 Vue 内部的运行原理，这里只能简单的描述一下。

exports.reload = function(id, options) { 
  const record = map[id]; 
  if (options) { 
    // reload的情况下 传入的options会当做一个新的组件 
    // 所以要用makeOptionsHot重新做一下记录 
    makeOptionsHot(id, options); 
    const newCtor = record.Ctor.super.extend(options); 
 
    newCtor.options._Ctor = record.options._Ctor; 
    record.Ctor.options = newCtor.options; 
    record.Ctor.cid = newCtor.cid; 
    record.Ctor.prototype = newCtor.prototype; 
  } 
  record.instances.slice().forEach(function(instance) { 
    instance.$vnode.context.$forceUpdate(); 
  }); 
}; 

这段代码关键的点开始于

const newCtor = record.Ctor.super.extend(options); 

利用新传入的配置生成了一个新的组件构造函数 然后对 record 上的 Ctor 进行了一系列的赋值

newCtor.options._Ctor = record.options._Ctor; 
record.Ctor.options = newCtor.options; 
record.Ctor.cid = newCtor.cid; 
record.Ctor.prototype = newCtor.prototype; 

注意第一次调用 reload 时，这里的 record.Ctor 还是最初传入的 Ctor，是由

const app = prepare(id1, { 
  data: () => ({ msg: 'foo' }), 
  render(h) { 
    return h('div', this.msg); 
  }, 
}).$mount('#app'); 

这个配置对象所生成的构造函数，但是构造函数的 options、cid 和 prototype 被替换成了由

api.reload(id1, { 
  data: () => ({ msg: 'bar' }), 
  render(h) { 
    return h('div', this.msg); 
  }, 
}); 

这个配置对象所生成的构造函数上的 options、cid 和 prototype，此时的 cid 肯定是不同的。

也就是说，构造函数的 cid 变了!，这个点记住后面要考!

继续看源码

record.instances.slice().forEach(function(instance) { 
  instance.$vnode.context.$forceUpdate(); 
}); 

此时的 instance 只有一个，就是在 reload 之前运行的那个 msg 为 foo 的实例，它的$vnode.context 是什么呢?

直接在放上控制台打印出来的截图，这个 context 是一个 vue 实例，注意这个 options 里的 render 函数，是不是非常眼熟，没错，这个 vue 实例其实就是我们的 prepare 函数中

new Vue({ 
  render: h => h(Comp), 
}); 

返回的 vm 实例。

那么这个函数的 $forceUpdate必然会触发 render: h => h(Comp) 这个函数，看到此时我们似乎还是没看出来这些操作为什么会销毁旧组件，创建新组件。那么此时只能探究一下这个h到底做了什么，这个h就是对应着 $createElement 方法。

$createElement方法

$createElement 在创建 vnode 的时候，最底层会调用一个 createComponent 方法，

这个方法把 Comp 对象当做 Ctor，然后调用 Vue.extend 这个 api 创造出构造函数，

默认情况下第一次 h(Comp) 会生成类似于 vue-component-${cid}作为组件的 tag，

在本例中最开始渲染 msg 为 foo 的组件时，tag 为 vue-component-1，

并且会把这个构造函数缓存在_Ctor 这个变量上，这样下次渲染再执行到 createComponent 的时候就不需要重新生成一次构造函数了，

Vue 在选择更新策略时调用一个sameVnode方法

来决定是要进行打补丁，还是彻底销毁重建，这个sameVnode如下：

function sameVnode(a, b) { 
  return ( 
    // 省略其他... 
    a.tag === b.tag 
  ); 
} 

其中很关键的一个对比就是a.tag === b.tag

但是 reload 方法偷梁换柱把 Ctor 的 cid 换成了 2，

生成的 vnode 的 tag 是就 vue-component-2

后续再调用 context.$forceUpdate 的时候，会发现两个组件的 tag 不一样，所以就走了销毁 -> 重新创建的流程。

总结

这个库里面还是能看出很多尤大的编程风格，很适合进行学习，只是 reload 方法必须要深入了解 Vue 源码才有可能搞懂生效的原理。