Vue 响应式核心 observer 源码详解 + 实践

笔者日前学习了 Vue 的 Observer 部分,简单地谷歌了一下,因为没有找到解释地十分彻底的中文资源,记下自己对其的理解并分享。

本文引用的 Vue 版本为 v2.5.17-beta.0 。
不过 Vue 的 Observer 部分自2017年以来至今没什么大变化,v2.5.16 到 v2.5.17-beta.0 对 Observer 有个小小的 bugfix。

内容

本文介绍 Vue 响应式原理的实现过程,并试图以之为参照改造出一个便于移植的库。这里笔者把 Vue 的 observer 部分提出来独立地讲,读者不需要对 Vue 其他部分十分熟悉。

Vue 的响应式模型十分完善,实现地足够巧妙,私以为有学习的必要。本文准备从写一个简单的模型出发,一步步填充功能,演化成 Vue 源码的形态,所以文章看起来似乎巨长,但代码多有重复;我认为这样写,读者看起来会比较轻松,所以请不必长文恐惧。卢瑟福说,“只有你能将一个理论讲得连女仆都懂了,你才算真正懂了”。虽然读者可能不是女仆(?),我也会写得尽量明白的。

本文对 Observer 介绍地很完全,对象和数组的不同处理,deep watching,以及异步队列都会讲解。当然,也不会完全整成源码那么麻烦,一些只和 Vue 有关的代码删除了,此外计算属性(computed property)的部分只说明原理,省略了实现。

但一般的 JS 技巧,ECMAScript 6,闭包的知识,Object.defineProperty 的知识还是需要具备的。

Vue 源码是用 Flow 写的,本文改成 TypeScript 了(同为类型注解,毕竟后者更流行),未学习过的同学只要把文中不像 JS 的部分去掉,当 JS 就行了。

JS 中数组是对象的一种,因为 Observer 部分对数组与普通对象的对待区别很大,所以下文说到对象,都是指 constructor 为 Object 的普通对象。

准备

可以先git clone [email protected]:vuejs/vue.git一份源码备看。observer 的部分在源码的 src/core/observer 目录下。

本文代码已经放在 https://github.com/xyzingh/le... ,运行 npm i && npm run test 可以测试。

新建文件夹 learn-vue-observer,创建几个文件。

util.ts

/* 一些常用函数的简写 */

export function def(obj: any, key: string, value: any, enumerable: boolean = false) {
  Object.defineProperty(obj, key, {
    value,
    enumerable,
    writable: true,
    configurable: true,
  });
}

export function isObject(obj: any) {
  return obj !== null && typeof obj === "object";
}

export function hasOwn(obj: any, key: string): boolean {
  return Object.prototype.hasOwnProperty.call(obj, key);
}

export function isPlainObject(obj: any): boolean {
  return Object.prototype.toString.call(obj) === "[object Object]";
}

export function isNative(ctor: any): boolean {
  return typeof ctor === "function" && /native code/.test(ctor.toString());
}

export function remove(arr: any[], item: any): any[] | void {
  if (arr.length) {
    const index = arr.indexOf(item);
    if (index > -1) {
      return arr.splice(index, 1);
    }
  }
}

版本 0.1

假设我们要把下面这个对象转变成响应式的。

let obj = {
  a: {
    aa: {
      aaa: 123,
      bbb: 456,
    },
    bb: "obj.a.bb",
  },
  b: "obj.b",
};

怎样算作是响应式的呢?如果将 obj 的任意键的值改变,都能执行一个相应的函数进行相关操作(比如更新DOM),那么就算得上响应式了。为此,我们势必为 obj 的每个键创建代理,使对 obj 的直接操作变成透过代理操作。代理的方式有许多,Object.observeProxy,getter/setter。但 Object.observe 已经被废弃,Proxy 巨硬家从 Edge 才开始支持,IE 全灭,所以可行的只有 getter/setter (IE9 开始支持)。然而 getter/setter 依然有很大的局限性,即只能转化已有属性,因此需要为用户提供特别的函数来设置新属性,这个函数我们最后再提。

obj 的值都转成 getter/setter 了,真实值存在哪呢?Vue 的做法是藏在闭包里。

下面我们定义3个函数/类,尝试递归地设置 obj 的 getter/setter。

index.ts

import { def, hasOwn, isObject, isPlainObject } from "./util";

/**
 * 尝试对 value 创建 Observer 实例,
 * value 如果不是对象或数组,什么都不做。
 * @param value 需要尝试监视的目标
 */
export function observe(value: any) {
  if (!isObject(value)) {
    return;
  }

  let ob: Observer | void;
  if (typeof value.__ob__ !== "undefined") {
    ob = value.__ob__;
  } else {
    ob = new Observer(value);
  }
  return ob;
}

export class Observer {
  constructor(value: any) {
    def(value, "__ob__", this);
    this.walk(value);
  }
  public walk(value: any) {
    for (const key of Object.keys(value)) {
      defineReactive(value, key);
    }
  }
}

function defineReactive(obj: any, key: string, val?: any) {
  // 闭包中的 val 藏着 obj[key] 的真实值
  if (arguments.length === 2) {
    val = obj[key];
  }

  let childOb = observe(val); // val 如果不是对象的话,是返回 undefined 的。
  Object.defineProperty(obj, key, {
    enumerable: true,
    configurable: true,
    get() {
      ////////////////
      console.log("you get " + val);
      ////////////////
      return val;
    },
    set(newVal) {
      if (newVal === val) {
        return;
      }
      ////////////////
      console.log("you set " + newVal);
      ////////////////
      val = newVal;
      childOb = observe(newVal);
    }
  });
}

我们可以试一下

observe(obj);
console.log(obj.a.aa.aaa = 234);

输出应为

you get [object Object]
you get [object Object]
you set 234
234

但是,有个问题,我们不应假设 obj 的每个键就是简单的值,万一本来就是 getter/setter 呢?

let obj2 = {};
Object.defineProperty(obj2, "a", {
    configurable: true,
    enumerable: true,
    get() {
        return obj2._a;
    },
    set(val) {
        obj2._a = val;
    },
});
Object.defineProperty(obj2, "_a", {
    enumerable: false,
    value: 123,
    writable: true,
});

因此,需要修改 defineReactive ,继续用闭包保存 getter/setter 。

function defineReactive(obj: any, key: string, val?: any) {
  const property = Object.getOwnPropertyDescriptor(obj, key);
  if (property && property.configurable === false) {
    return;
  }

  const getter = property!.get; // property! 的叹号是 TypeScript 语法,忽略即可
  const setter = property!.set;

  // 为什么写成 (!getter || setter) ?后面会讨论。
  if ((!getter || setter) && arguments.length === 2) {
    val = obj[key];
  }

  let childOb = observe(val);
  Object.defineProperty(obj, key, {
    enumerable: true,
    configurable: true,
    get() {
      const value = getter ? getter.call(obj) : val;
      ////////////////
      console.log("you get " + value);
      ////////////////
      return value;
    },
    set(newVal) {
      const value = getter ? getter.call(obj) : val;
      if (newVal === value) {
        return;
      }
      ////////////////
      console.log("you set " + newVal);
      ////////////////
      if (setter) {
        setter.call(obj, newVal);
      } else {
        val = newVal;
      }
      childOb = observe(newVal);
    },
  });
}

这样就可以成功地把 obj2 转变成响应式的。

笔者在理解 if ((!getter || setter) && arguments.length === 2) 时遇到过障碍,这其实是讲:

  1. 如果 arguments 长为3,参数 val 存在,就认为是显式地设置了这个键的值,原来的值就不考虑了
  2. 如果 getter setter 都存在,就认为这对 getter/setter 是在代理某个真实值,所以需要 val = obj[key],然后 let childOb = observe(val) 对这个真实值继续进行递归设置
  3. 否则 如果 getter 存在,setter 不存在,认为 getter 大概只是返回某个生成的值,不执行 val = obj[key],也就导致下面 let childOb = observe(undefined)
  4. getter 不存在,setter 存在,这类奇葩事情不在考虑范围内(例如 document.cookie)

这是 v2.5.17-beta.0 的一个 bugfix ,有关的讨论原文来自↓
issue/7280
issue/7302
pull/7981
issue/8494

版本 0.2: 加上对数组的支持

并不是说之前的版本不支持数组,而是一般开发者使用数组与使用对象的方法有区别。数组在 JS 中常被当作栈,队列,集合等数据结构的实现方式,会储存批量的数据以待遍历。编译器对对象与数组的优化也有所不同。所以对数组的处理需要特化出来以提高性能。

首先,不能再对数组每个键设置 getter/setter 了,而是修改覆盖数组的 push, pop, ... 等方法。用户要修改数组只能使用这些方法,否则不会是响应式的(除了 Vue.set, Vue.delete)。

因此,准备一个数组方法的替代品。哪些方法应当替代掉?那些不会干涉原数组的方法不需要修改;删除数组元素的方法需要替代;增加或替换数组元素的方法需要替换,还要尝试把新的值变成响应式的。

array.ts

import { def } from './util';

const arrayProto = Array.prototype as any;
// 建立以 Array.prototype 为原型的 arrayMethods 并导出
export const arrayMethods = Object.create(arrayProto);

// 会干涉原数组的方法
const methodsToPatch = [
  'push',
  'pop',
  'shift',
  'unshift',
  'splice',
  'sort',
  'reverse',
];

methodsToPatch.forEach((method: string) => {
  // 原方法的缓存
  const original = arrayProto[method];

  // 在 arrayMethods 上定义替代方法
  def(arrayMethods, method, function (this: any, ...args: any[]) {
    const result = original.apply(this, args);
    const ob = this.__ob__;

    // 新增的元素
    let inserted: any[] | void;

    switch (method) {
      // 会增加或替换元素的方法
      case 'push':
      case 'unshift':
        inserted = args;
        break;
      case 'splice':
        inserted = args.slice(2);
        break;
    }
    if (inserted){
      ob.observeArray(inserted); // Observer 上新增的方法
    }
    ///////////////////////////////
    console.log("array is modified.");
    ///////////////////////////////
    return result;
  });
});

然后修改 Observer,区别对待数组。

export class Observer {
  constructor(value: any) {
    def(value, "__ob__", this);
    if (Array.isArray(value)) {
      // 替换原型(Object.setPrototype 这个方法执行地比较慢,而且支持情况堪忧)
      Object.setPrototypeOf(value, arrayMethods);
      this.observeArray(value);
    } else {
      this.walk(value);
    }
  }
  public walk(value: any) {
    for (const key of Object.keys(value)) {
      defineReactive(value, key);
    }
  }
  public observeArray(items: any[]) {
    // 设置 l = items.length 防止遍历过程中 items 长度变化
    for (let i = 0, l = items.length; i < l; i++) {
      // 直接观察数组元素,省略在键上设置 getter/setter 的步骤
      observe(items[i]);
    }
  }
}

版本 1.0: 增加 Dep, Watcher

从 API 出发思考写法

vm.$watch( expressionOrFunction, callback [, options] ) 是 Vue 最基础的观察自身 data 的方式。我们参考这个函数,提出适用本文的一个函数:

watch( target, expression, callback )

观察 target 这个对象的表达式 exp 的值,一旦发生变化时执行 callback (同步地)。callback 的第一个参数为新的值,第二个参数为旧的值,this 为 target。

例如 watch(obj, "a.aa.bbb", val => console.log(val)) ,当 obj.a.aa.bbb 改变时,控制台会打印新的值。注意 obj 应该已经经过 observe(obj) 转化过了。

之前版本我们在 getter/setter 处留下了

/////////////////
console.log(XXX)
/////////////////

只要把这些替换成相应代码,就能实现 watch 方法了。

现在来定义一下哪些情况应执行 callback 。

假设 obj.a.aa.bbb = 456 ,我们对这个键进行了 watch :

  1. obj.a.aa.bbb = 456 值没变,不需要
  2. obj.a.aa.bbb = 999 应执行 callback
  3. obj.a.aa = { bbb: 456 } 值没变,不执行 callback
  4. obj.a.aa = { bbb: 999 } 应执行 callback
  5. obj = {a:{aa:{bbb:999}}} 对象都被替换成新的了,想执行 callback 也不可能

假设我们还对 obj.a.aa 进行了 watch :

  1. obj.a.aa.bbb = 999 虽然 obj.a.aa 发生了变异(mutation),但 obj.a.aa 还是它自己,不执行 callback
  2. obj.a.aa = { bbb: 456 } 应执行 callback

简而言之,如果 target 沿着 expression 解析到的值与之前的不全等,就认为需要执行 callback 。对于基础类型来说,就是值的不全等。对于普通对象,就是引用不相同。但数组比较特殊,对数组元素进行了操作,就应执行 callback 。

怎么组织代码呢?Evan You (Vue 作者) 的方法比较巧妙。

Observer, Dep, Watcher

创建两个新的类,Dep, Watcher 。Dep 是 Dependency 的简称,每个 Observer 的实例,成员中都有一个 Dep 的实例。

这个 Dep 的实质是个数组,放置着监听这个 Observer 的 Watcher ,当这个 Observer 对应的值变化时,就通知 Dep 中的所有 Watcher 执行 callback 。

export class Observer {
  constructor(value: any) {
    this.dep = new Dep(); // 新增
    def(value, "__ob__", this);
    if (Array.isArray(value)) {
// .........................

Watcher 是调用 watch 函数产生的,它保存着 callback 并且维护了一个数组,数组存放了所有 存有这个 Watcher 的 Dep 。这样当这个 Watcher 需要被删除时,可以遍历数组,从各个 Dep 中删去自身,也就是 unwatch 的过程。

Watcher 何时被放入 Dep 中的先不谈。先说说 Dep 都在什么地方。

以上说得并不全对,应该说,原始的 Dep 是创建在 defineReactive 的闭包中,Observer 的 dep 成员只是这个原始的 Dep 的备份,始终一起被维护,保持一致。另外,Observer 只会建立在对象或数组的 __ob__ 上,如果键的值不是对象或数组,只会有闭包中的 Dep 保存这个键的 Watcher 。

function defineReactive(obj: any, key: string, val?: any) {
  const dep = new Dep(); // 新增
  const property = Object.getOwnPropertyDescriptor(obj, key);
// ...........................

举例来说,

let obj = {
  // obj.__ob__.dep: 保存 obj 的 dep

  a: { // 闭包中有 obj.a 的 dep
    // obj.a.__ob__.dep: 保存 obj.a 的 dep

    aa: { // 闭包中有 obj.a.aa 的 dep
      // obj.a.aa.__ob__.dep: 保存 obj.a.aa 的 dep

      aaa: 123, // 闭包中有 obj.a.aa.aaa 的 dep
      bbb: 456, // 闭包中有 obj.a.aa.bbb 的 dep
    },
    bb: "obj.a.bb", // 闭包中有 obj.a.bb 的 dep
  },
  b: "obj.b", // 闭包中有 obj.b 的 dep
};
observe(obj);

数组特殊对待,不对数组的成员进行 defineReactive ,

let obj = {
  arr: [ // 闭包中 obj.arr 的 dep
    // obj.arr.__ob__.dep

    2, // 没有 dep ,没有闭包
    3,
    5,
    7,
    11,
    { // 没有闭包
      // obj.arr[6].__ob__.dep 存在
    },
    [ // 没有闭包
      // obj.arr[7].__ob__.dep 存在
    ],
  ],
};
observe(obj);

复习一下,dep 实质是个数组,放着监听这个键的 Watcher 。

当这个键的值被修改时,就应该通知相应 dep 的所有 Watcher ,我们在 Dep 上设置 notify 方法,用来实现这个功能。

为此,修改 setter 的部分。

function defineReactive(obj: any, key: string, val?: any) {
  const dep = new Dep();
  // .................................
  Object.defineProperty(obj, key, {
    enumerable: true,
    configurable: true,
    get() {
      // .............................
    },
    set(newVal) {
      // .............................
      }
      childOb = observe(newVal);
      dep.notify();
    },
  });
}

数组的部分,

array.ts

// ................
  def(arrayMethods, method, function (this: any, ...args: any[]) {
    const result = original.apply(this, args);
    const ob = this.__ob__;
    // .....................
    if (inserted){
      ob.observeArray(inserted);
    }
    ob.dep.notify();
    return result;
  });

如此一来,每当修改值时,相应的 Watcher 都会被通知了。

现在的问题是,何时怎么把 Watcher 放入 dep 中。下面我们先来尝试实现 Dep 。

dep.ts

import { remove } from "./util";
import { Watcher } from "./watcher";

let uid = 0;

export default class Dep {
  public id: number;
  public subs: Watcher[];

  constructor() {
    this.id = uid++;
    this.subs = [];
  }

  public addSub(sub: Watcher) {
    this.subs.push(sub);
  }

  public removeSub(sub: Watcher) {
    remove(this.subs, sub);
  }

  public notify() {
    // 先复制一份,应对通知 Watcher 过程中,this.subs 可能变化的情况
    const subs = this.subs.slice();
    for (let i = 0, l = subs.length; i < l; i++) {
      // Watcher 上定义了 update 方法,用来被通知
      subs[i].update();
    }
  }
}

(重点)用 Touch 的方法,收集依赖

假设用 watch(obj, "a.aa.bbb", val => console.log(val)) ,创建了一个 Watcher ,这个 Watcher 应被放进哪些 Dep 中呢?

因为 obj.a, obj.a.aa 改变时,obj.a.aa.bbb 的值可能改变,所以答案是 obj.a, obj.a.aa, obj.a.aa.bbb 的闭包中的 Dep, 前两者是对象,所以在 __ob__.dep 中再放一份。

因为在对表达式 obj.a.aa.bbb 求值时,会依次执行 obj.a, (obj.a).aa, ((obj.a).aa).bbb 的 getter ,这也正好对应了应被放入 Watcher 的键,所以很自然的一个想法是,

规定一个全局变量,平常是 null ,当在决定某个 Watcher 该放入哪些 Dep 的时候(即 依赖收集 阶段),让这个全局变量指向这个 Watcher 。然后 touch 被监视的那个键,换言之,对那个键求值。途中会调用一连串的 getter ,往那些 getter 所对应的 Dep 里放入这个 Watcher 就对了。之后再将全局变量改回 null 。

这个做法的妙处,还在于它可以同时适用 deep watching计算属性(computed property)。deep watching 后面会再说,对于计算属性,这使得用户直接写函数就行,无需显式说明这个计算属性所依赖的其他属性,十分优雅,因为在运算这个函数时,用到其他属性就会触发 getter ,可能的依赖都会被收集起来。

我们来尝试实现,

export default class Dep {
  // Dep.target 即前文所谓的全局变量
  public static target: Watcher | null = null;

  public id: number;
  public subs: Watcher[];

  public depend() {
    if (Dep.target) {
      this.addSub(Dep.target);
    }
  }
// ...................................................
function defineReactive(obj: any, key: string, val?: any) {
  const dep = new Dep();
  // ...................................................
  let childOb = observe(val);
  Object.defineProperty(obj, key, {
    enumerable: true,
    configurable: true,
    get() {
      const value = getter ? getter.call(obj) : val;

      // 如果处在依赖收集阶段
      if (Dep.target) {
        dep.depend();
        if (childOb) {
          childOb.dep.depend();
        }
      }

      return value;
    },
  // ....................................................
}

现在也该把一直谈论的 Watcher 给实现了。根据前面说的,它应该有个 update 方法。

watcher.ts

import Dep from "./dep";

let uid = 0;

export class Watcher {
  public id: number;
  public value: any;
  public target: any;
  public getter: (target: any) => any;
  public callback: (newVal: any, oldVal: any) => void;

  constructor(
    target: any,
    expression: string,
    callback: (newVal: any, oldVal: any) => void,
  ) {
    this.id = uid++;
    this.target = target;
    this.getter = parsePath(expression);
    this.callback = callback;
    this.value = this.get();
  }

  public get() {
    // 进入依赖收集阶段
    Dep.target = this;

    let value: any;
    const obj = this.target;
    try {
      // 调用了一连串 getter ,对应的键的 dep 中放入了这个 watcher
      value = this.getter(obj);
    } finally {
      // 退出依赖收集阶段
      Dep.target = null;
    }
    return value;
  }

  public update() {
    this.run();
  }
  public run() {
    this.getAndInvoke(this.callback);
  }
  public getAndInvoke(cb: (newVal: any, oldVal: any) => void) {
    const value = this.get();
    if (value !== this.value || isObject(value) /* 监视目标为对象或数组的话,仍应执行回调,因为值可能变异了 */) {
      const oldVal = this.value;
      this.value = value;
      cb.call(this.target, value, oldVal);
    }
  }
}

const bailRE = /[^\w.$]/;
function parsePath(path: string): any {
  if (bailRE.test(path)) {
    return;
  }
  const segments = path.split(".");
  return (obj: any) => {
    for (const segment of segments) {
      if (!obj) { return; }
      obj = obj[segment];
    }
    return obj;
  };
}

版本 1.1: 特化数组的依赖收集

function defineReactive(obj: any, key: string, val?: any) {
// .....................................
      if (Dep.target) {
        dep.depend();
        if (childOb) {
          childOb.dep.depend();
          if (Array.isArray(value)) {
            dependArray(value);
          }
        }
      }
// ......................................
}

function dependArray(value: any[]) {
  for (let e, i = 0, l = value.length; i < l; i++) {
    e = value[i];

    // 若为多维数组,继续递归监视
    e && e.__ob__ && e.__ob__.dep.depend();
    if (Array.isArray(e)) {
      dependArray(e);
    }
  }
}

讨论的原文来自 issue/3883 ,举例而言,

let obj = {
  matrix: [
    [2, 3, 5, 7, 11],
    [13, 17, 19, 23, 29],
  ],
};
observe(obj);
watch(obj, "matrix", val => console.log(val));
obj.matrix[0].push(1);
// 导致 matrix[0].__ob__.dep.notify() ,由于递归监视,这个 dep 里也有上面的 Watcher

版本 1.2: 完善 Watcher 的生命周期

只有 watch 没有 unwatch 自然是不合理的。前面提到,Watcher 也维护了一个数组 deps,存放所有 放了这个 Watcher 的 Dep ,当这个 Watcher 析构时,可以从这些 Dep 中删去自身。

我们给 Watcher 增加 active, deps, depIds, newDeps, newDepIds 属性,addDep, cleanupDeps, teardown 方法,其中 teardown 方法起的是析构的作用,active 标志 Watcher 是否可用,其他的都是围绕着维护 deps 。

export class Watcher {
  // ..............................
  public active = true;
  public deps: Dep[] = [];
  public depIds = new Set<number>();
  public newDeps: Dep[] = [];
  public newDepIds = new Set<number>();

  public run() {
    if (this.active) {
      this.getAndInvoke(this.callback);
    }
  }

  // newDeps 是新一轮收集的依赖,deps 是之前一轮收集的依赖
  public addDep(dep: Dep) {
    const id = dep.id;
    if (!this.newDepIds.has(id)) {
      this.newDepIds.add(id);
      this.newDeps.push(dep);
      if (!this.depIds.has(id)) {
        dep.addSub(this);
      }
    }
  }

  public get() {
    Dep.target = this;

    let value: any;
    const obj = this.target;
    try {
      value = this.getter(obj);
    } finally {
      Dep.target = null;
      this.cleanupDeps();
    }
    return value;
  }

  // 清理依赖
  // 之前收集的依赖 如果不出现在新一轮收集的依赖中,就清除掉
  // 再交换 deps/newDeps, depIds/newDepIds
  public cleanupDeps() {
    let i = this.deps.length;
    while (i--) {
      const dep = this.deps[i];
      if (!this.newDepIds.has(dep.id)) {
        dep.removeSub(this);
      }
    }
    const tmpIds = this.depIds;
    this.depIds = this.newDepIds;
    this.newDepIds = tmpIds;
    this.newDepIds.clear();

    const tmp = this.deps;
    this.deps = this.newDeps;
    this.newDeps = tmp;
    this.newDeps.length = 0;
  }

  public teardown() {
    if (this.active) {
      let i = this.deps.length;
      while (i--) {
        this.deps[i].removeSub(this);
      }
      this.active = false;
    }
  }
}

修改之前的 dep.ts

export default class Dep {
  public depend() {
    if (Dep.target) {
      // this.addSub(Dep.target);
      Dep.target.addDep(this);
    }
  }
}

版本 2.0: deep watching

Deep watching 的原理很简单,就是在用 touch 收集依赖的基础上,递归遍历并 touch 所有子元素,如此一来,所有子元素都被收集到依赖中。其中只有防止对象引用成环需要稍微注意一下,这个用一个集合记录遍历到的元素来解决。

我们给 Watcher 构造函数增加一个 deep 选项。

直接贴代码,

export class Watcher {
  public deep: boolean;
  constructor(
    target: any,
    expression: string,
    callback: (newVal: any, oldVal: any) => void,
    {
      deep = false,
    },
  ) {
    this.deep = deep;
    // ................................
  }
  public get() {
    Dep.target = this;

    let value: any;
    const obj = this.target;
    try {
      value = this.getter(obj);
    } finally {
      if (this.deep) {
        // touch 所有子元素,收集到依赖中
        traverse(value);
      }
      Dep.target = null;
      this.cleanupDeps();
    }
    return value;
  }

  public getAndInvoke(cb: (newVal: any, oldVal: any) => void) {
    const value = this.get();
    if (value !== this.value ||
      isObject(value) ||
      this.deep /* deep watcher 始终执行 */
    ) {
      const oldVal = this.value;
      this.value = value;
      cb.call(this.target, value, oldVal);
    }
  }
}

traverse.ts

import { isObject } from "./util";

const seenObjects = new Set();

export function traverse(val: any) {
  _traverse(val, seenObjects);
  seenObjects.clear();
}

function _traverse(val: any, seen: Set<any>) {
  let i;
  let keys;
  const isA = Array.isArray(val);
  if ((!isA && !isObject(val)) || Object.isFrozen(val)) {
    return;
  }
  if (val.__ob__) {
    const depId = val.__ob__.dep.id;
    if (seen.has(depId)) {
      return;
    }
    seen.add(depId);
  }
  if (isA) {
    i = val.length;
    while (i--) { _traverse(val[i] /* touch */, seen); }
  } else {
    keys = Object.keys(val);
    i = keys.length;
    while (i--) { _traverse(val[keys[i]] /* touch */, seen); }
  }
}

版本 2.1: 异步 Watcher, 异步队列

使用异步 Watcher 可以缓冲在同一次事件循环中发生的所有数据改变。如果在本次执行栈中同一个 Watcher 被多次触发,只会被推入到队列中一次。这样在缓冲时去除重复数据,能够避免不必要的计算,提高性能。

Vue 源码中的异步队列模型比下文中的复杂,因为 Vue 要保证

  1. 从父组件到子组件的更新顺序
  2. 用户定义的 watcher 在 负责渲染的 watcher 之前运行
  3. 若在父组件的 watcher 运行时摧毁了子组件,子组件的 watcher 应被跳过
  4. 被计算属性依赖的另一计算属性先运行

如果这些是你的兴趣,请直接转战源码 src/core/observer/scheduler.js 。

现在修改 watcher.ts ,

export class Watcher {
  public deep: boolean;
  public sync: boolean;
  constructor(
    target: any,
    expression: string,
    callback: (newVal: any, oldVal: any) => void,
    {
      deep = false,
      sync = false, // 增加同步选项
    },
  ) {
    this.deep = deep;
    this.sync = sync;
    // ............................
  }
  public update() {
    if (this.sync) {
      this.run();
    } else {
      queueWatcher(this); // 推入队列
    }
  }
}

创建 scheduler.ts

/// <reference path="next-tick.d.ts" />
import { nextTick } from "./next-tick";
import { Watcher } from "./watcher";

const queue: Watcher[] = [];
let has: { [key: number]: true | null } = {};
let waiting = false;
let flushing = false;
let index = 0;

/**
 * 重置 scheduler 的状态.
 */
function resetSchedulerState() {
  index = queue.length = 0;
  has = {};
  waiting = flushing = false;
}

/**
 * 刷新队列,并运行 watcher
 */
function flushSchedulerQueue() {
  flushing = true;
  let watcher;
  let id;

  queue.sort((a, b) => a.id - b.id);

  
  for (index = 0; index < queue.length /* 不缓存队列长度,因为新的 watcher 可能在执行队列时加进来 */; index++) {
    watcher = queue[index];
    id = watcher.id;
    has[id] = null;
    watcher.run();
  }

  resetSchedulerState();
}

/**
 * 将一个 watcher 推入队列
 * 相同 ID 的 watcher 会被跳过
 * 除非队列中之前的相同ID的 watcher 已被处理掉
 */
export function queueWatcher(watcher: Watcher) {
  const id = watcher.id;
  if (has[id] == null) {
    has[id] = true;
    if (!flushing) {
      queue.push(watcher);
    } else {
      let i = queue.length - 1;

      // 放到队列中相应 ID 的位置
      while (i > index && queue[i].id > watcher.id) {
        i--;
      }
      queue.splice(i + 1, 0, watcher);
    }
    if (!waiting) {
      waiting = true;
      
      // 放入微任务队列
      nextTick(flushSchedulerQueue);
    }
  }
}

如果不清楚微任务队列是什么,可以阅读下 理解浏览器和node.js中的Event loop事件循环

下面贴一下 Vue 的 nextTick 实现。
next-tick.d.ts

// 自己给 next-tick 写了下接口
export declare function nextTick(cb: () => void, ctx?: any): Promise<any> | void;

next-tick.js (注意这是 JS)

import { isNative } from "./util";

const inBrowser = typeof window !== "undefined";
const inWeex = typeof WXEnvironment == "undefined" && !!WXEnvironment.platform;
const weexPlatform = inWeex && WXEnvironment.platform.toLowerCase();
const UA = inBrowser && window.navigator.userAgent.toLowerCase();
const isIOS = (UA && /iphone|ipad|ipod|ios/.test(UA)) || (weexPlatform === "ios");

function noop() {}
function handleError() {}

const callbacks = [];
let pending = false;

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

// Here we have async deferring wrappers using both microtasks and (macro) tasks.
// In < 2.4 we used microtasks everywhere, but there are some scenarios where
// microtasks have too high a priority and fire in between supposedly
// sequential events (e.g. #4521, #6690) or even between bubbling of the same
// event (#6566). However, using (macro) tasks everywhere also has subtle problems
// when state is changed right before repaint (e.g. #6813, out-in transitions).
// Here we use microtask by default, but expose a way to force (macro) task when
// needed (e.g. in event handlers attached by v-on).
let microTimerFunc;
let macroTimerFunc;
let useMacroTask = false;

// Determine (macro) task defer implementation.
// Technically setImmediate should be the ideal choice, but it's only available
// in IE. The only polyfill that consistently queues the callback after all DOM
// events triggered in the same loop is by using MessageChannel.
/* istanbul ignore if */
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);
  };
}

// Determine microtask defer implementation.
/* istanbul ignore next, $flow-disable-line */
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;
}

/**
 * Wrap a function so that if any code inside triggers state change,
 * the changes are queued using a (macro) task instead of a microtask.
 */
export function withMacroTask(fn) {
  return fn._withTask || (fn._withTask = function() {
    useMacroTask = true;
    const res = fn.apply(null, arguments);
    useMacroTask = false;
    return res;
  });
}

export function nextTick(cb, ctx) {
  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();
    }
  }
  // $flow-disable-line
  if (!cb && typeof Promise !== "undefined") {
    return new Promise((resolve) => {
      _resolve = resolve;
    });
  }
}

总结

本文代码已经放在 https://github.com/xyzingh/le... ,运行 npm i && npm run test 可以测试。

相关推荐