实现vue3响应式系统核心-增强对象拦截

简介

在之前的文章中我们实现一个响应式系统的 MVP 模型,也实现了 computedwatch 等。 今天再来看看对于对象的拦截,我们思考以下几个问题:

  • 如何拦截 in操作符呢?
  • 如何拦截 for in 循环呢?
  • 如何拦截对象的删除操作呢?

接下来我们会一步步实现这些功能,进一步增强 MVP 模型。

《实现vue3响应式系统核心》 系列文章

代码地址: github.com/SuYxh/share...

代码并没有按照源码的方式去进行组织,目的是学习、实现 vue3 响应式系统的核心,用最少的代码去实现最核心的能力,减少我们的学习负担,并且所有的流程都会有配套的图片,图文 + 代码,让我们学习更加轻松、快乐。

每一个功能都会提交一个 commit ,大家可以切换查看,也顺变练习练习 git 的使用。

对象读取操作

先看看一个普通对象所有可能的读取操作有哪些?

  • 访问属性:obj.foo
  • 判断对象或原型上是否存在给定的 key:key in obj
  • 使用 for...in 循环遍历对象:for (const key in obj){}

Proxy 内部方法

Proxy对象部署的所有内部方法以及用来自定义内部方法和行为的拦截函数名字

拦截 in 操作符

现给出结论:我们可以通过 has 拦截函数实现对 in 操作符。

为什么是 has 呢?

在 ECMA-262 规范的 13.10.1 节中,明确定义了 in 操作符的运行时逻辑,如图所示:

关键点在第 6 步,可以发现,in 操作符的运算结果是通过调用一个叫作 HasProperty的抽象方法得到的。关于 HasProperty抽象方法,可以在 ECMA-262 规范的 7.3.11 节中找到,它的操作如图所示:

可以看到 HasProperty 抽象方法的返回值是通过调用对象的内部方法 [[HasProperty]]得到的。而[[HasProperty]]内部方法可以在Proxy内部方法中找到,它对应的拦截函数名叫 has,因此我们可以通过 has拦截函数实现对 in 操作符的代理。

看不懂也无所谓,只需要知道 has 可以拦截 in 操作。

单元测试

js 复制代码
it("拦截 in 操作符", () => {
  const mockFn = vi.fn();

  // 创建响应式对象
  const obj = reactive({ foo: 100 });

  effect(function effectFn1() {
    mockFn()
    console.log('foo' in obj);
  })

  expect(mockFn).toHaveBeenCalledTimes(1);

  delete obj.foo

  expect(mockFn).toHaveBeenCalledTimes(2);
});

代码实现

js 复制代码
// 拦截 in 操作符
has(target, key) {
  track(target, key);
  return Reflect.has(target, key);
},

按照以往,这里应该是运行 case 的时间,但是我们还并没有实现拦截删除,所以这里无法跑通,等到文末在运行单测。

但是可以通过调试看到,已经被收集到了。

接下来看一下 for in循环如何去拦截。

拦截 for...in 循环

这里直接给出答案:可以使用ownKeys拦截函数来拦截。

单元测试

js 复制代码
it("拦截 for in", () => {
  // 创建响应式对象
  const obj = reactive({ foo: 100 });
  const mockFn = vi.fn();

  effect(function effectFn1() {
    mockFn()

    for (const key in obj) {
      console.log(key);
    }
  })
  expect(mockFn).toHaveBeenCalledTimes(1);

  obj.bar = 2
  expect(mockFn).toHaveBeenCalledTimes(2);

  obj.foo = 100
  expect(mockFn).toHaveBeenCalledTimes(2);
});

代码实现

js 复制代码
const ITERATE_KEY = "iterate-key";


// 拦截 for in 循环
ownKeys(target) {
  track(target, ITERATE_KEY);
  return Reflect.ownKeys(target);
},

原因分析

ITERATE_KEY 作为追踪的 key ,为什么这么做呢?

这是因为 ownKeys 拦截函数与 get/set 拦截函数不同,在set /get中,我们可以得到具体操作的 key,但是在ownKeys中,我们只能拿到目标对象 targetownKeys 用来获取一个对象的所有属于自己的键值,这个操作明显不与任何具体的键进行绑定,因此我们只能够构造唯一的 key 作为标识,即 ITERATE_KEY

既然追踪的是 ITERATE_KEY,那么相应地,在触发响应的时候也应该触发它才行。但是在什么情况下,对数据的操作需要触发与 ITERATE_KEY 相关联的副作用函数重新执行呢?

为对象添加了新属性。因为,当为对象添加新属性时,会对 for...in 循环产生影响,所以需要触发与ITERATE_KEY相关联的副作用函数重新执行。

在我们之前写的 set函数中,当为对象 obj 添加新的 bar 属性时,会触发 set拦截函数执行。此时 set拦截函数接收到的 key就是字符串 bar,因此最终调用 trigger函数时也只是触发了与 bar相关联的副作用函数重新执行。

我们知道 for...in循环是在副作用函数与 ITERATE_KEY之间建立联系,这和 bar一点儿关系都没有,因此当我们尝试执行 obj.bar = 2操作时,并不能正确地触发响应。

通过调试可以看到:

解决

当添加属性时,我们将那些与 ITERATE_KEY 相关联的副作用函数也取出来执行就可以了:

js 复制代码
function trigger(target, key) {
  const depsMap = bucket.get(target);
  if (!depsMap) return;
  // 取得与 key 相关联的副作用函数
  const effects = depsMap.get(key);
  // 取得与 ITERATE_KEY 相关联的副作用函数
  const iterateEffects = depsMap.get(ITERATE_KEY);

  const effectsToRun = new Set();
  // 将与 key 相关联的副作用函数添加到 effectsToRun
  effects && effects.forEach(effectFn => {
    if (effectFn !== activeEffect) {
      effectsToRun.add(effectFn);
    }
  });
  // 将与 ITERATE_KEY 相关联的副作用函数也添加到 effectsToRun
  iterateEffects && iterateEffects.forEach(effectFn => {
    if (effectFn !== activeEffect) {
      effectsToRun.add(effectFn);
    }
  });

  effectsToRun.forEach(effectFn => {
    if (effectFn.options.scheduler) {
      effectFn.options.scheduler(effectFn);
    } else {
      effectFn();
    }
  });
}

运行单测

我们可以看到,单测并没有通过。从单测可以看出来,当我们修改值的时候,也触发了副作用函数的执行。

这又是怎么回事呢?

问题分析-修改 foo

与添加新属性不同,修改属性不会产生新的 key ,所以不会对 for...in 循环产生影响。所以在这种情况下,我们不需要触发副作用函数重新执行,否则会造成不必要的性能开销。

解决

那么我们在set 拦截函数内能够区分操作的类型,到底是添加新属性还是设置已有属性:

js 复制代码
// 拦截设置操作
set(target, key, newVal, receiver) {
  // 如果属性不存在,则说明是在添加新属性,否则是设置已有属性
  const type = Object.prototype.hasOwnProperty.call(target, key)
    ? TriggerType.SET
    : TriggerType.ADD;

  // 设置属性值
  const res = Reflect.set(target, key, newVal, receiver);
  // 派发更新
  trigger(target, key, type);
  return res;
},

我们优先使用Object.prototype.hasOwnProperty检查当前操作的属性是否已经存在于目标对象上,如果存在,则说明当前操作类型为 SET,即修改属性值;否则认为当前操作类型为 ADD,即添加新属性。

trigger 函数内就可以通过类型 type来区分当前的操作类型,并且只有当操作类型 typeADD时,才会触发与ITERATE_KEY相关联的副作用函数重新执行,这样就避免了不必要的性能损耗:

js 复制代码
function trigger (target, key, type) {
  
  // ... 
  
  // 只有当操作类型为 'ADD' 时,才触发与 ITERATE_KEY 相关联的副作用函数重新执行
  if (type === TriggerType.ADD) {
    // 取得与 ITERATE_KEY 相关联的副作用函数
    const iterateEffects = depsMap.get(ITERATE_KEY);

    iterateEffects &&
      iterateEffects.forEach((effectFn) => {
        if (effectFn !== activeEffect) {
          effectsToRun.add(effectFn);
        }
      });
  }
  
  // ...
}

再次运行单测

单测就已经通过!

如何拦截对象的删除操作呢?

规范的 13.5.1.2 节中明确定义了 delete 操作符的行为,如图所示:

由第 5 步中的 d 子步骤可知,delete 操作符的行为依赖 [[Delete]]内部方法。根据 Proxy 内部方法可知,该内部方法可以使用 deleteProperty 拦截:

单元测试

js 复制代码
it("拦截对象删除操作", () => {
  const mockFn = vi.fn();

  // 创建响应式对象
  const obj = reactive({ foo: 100 });

  effect(function effectFn1() {
    mockFn()
    console.log(obj.foo);
  })

  expect(mockFn).toHaveBeenCalledTimes(1);

  delete obj.foo

  expect(mockFn).toHaveBeenCalledTimes(2);
});

代码实现

js 复制代码
// 拦截删除
deleteProperty(target, key) {
  // 检查被操作的属性是否是对象自己的属性
  const hadKey = Object.prototype.hasOwnProperty.call(target, key);
  // 使用 Reflect.deleteProperty 完成属性的删除
  const res = Reflect.deleteProperty(target, key);

  if (res && hadKey) {
    // 只有当被删除的属性是对象自己的属性并且成功删除时,才触发更新
    trigger(target, key, TriggerType.DEL);
  }

  return res;
},

检查被删除的属性是否属于对象自身,然后调用Reflect.deleteProperty函数完成属性的删除工作,只有当这两步的结果都满足条件时,才调用trigger函数触发副作用函数重新执行。

⚠️注意: 由于删除操作会使得对象的键变少,它会影响 for...in 循环的次数,因此当操作类型为 DELETE时,我们也应该触发那些与 ITERATE_KEY 相关联的副作用函数重新执行:

js 复制代码
// 增加一个条件判断
if (type === TriggerType.ADD || type === TriggerType.DEL) {
  // 取得与 ITERATE_KEY 相关联的副作用函数
  const iterateEffects = depsMap.get(ITERATE_KEY);
  // 将与 ITERATE_KEY 相关联的副作用函数也添加到 effectsToRun
  iterateEffects &&
    iterateEffects.forEach((effectFn) => {
      if (effectFn !== activeEffect) {
        effectsToRun.add(effectFn);
      }
    });
}

运行单测

回顾 in 操作符单测

也可以跑通!

运行测试

bash 复制代码
pnpm test

没有问题!

到此我们就解决了开头我们提出的这几个问题:

  • 如何拦截 in操作符呢?
  • 如何拦截 for in 循环呢?
  • 如何拦截对象的删除操作呢?

进一步完善了我们的响应式系统。

相关推荐
ZBY52031几秒前
【Vue】 npm install amap-js-api-loader指南
javascript·vue.js·npm
猫爪笔记1 分钟前
前端:HTML (学习笔记)【2】
前端·笔记·学习·html
brief of gali13 分钟前
记录一个奇怪的前端布局现象
前端
计算机毕设指导617 分钟前
基于 SpringBoot 的作业管理系统【附源码】
java·vue.js·spring boot·后端·mysql·spring·intellij-idea
前端拾光者1 小时前
利用D3.js实现数据可视化的简单示例
开发语言·javascript·信息可视化
Json_181790144801 小时前
电商拍立淘按图搜索API接口系列,文档说明参考
前端·数据库
风尚云网2 小时前
风尚云网前端学习:一个简易前端新手友好的HTML5页面布局与样式设计
前端·css·学习·html·html5·风尚云网
木子02042 小时前
前端VUE项目启动方式
前端·javascript·vue.js
GISer_Jing2 小时前
React核心功能详解(一)
前端·react.js·前端框架
捂月2 小时前
Spring Boot 深度解析:快速构建高效、现代化的 Web 应用程序
前端·spring boot·后端