Vue3源码reactive和readonly对象嵌套转换，及实现shallowReadonly

前言

官方文档中对reactive的描述：

响应式转换是“深层”的：它会影响到所有嵌套的属性。一个响应式对象也将深层地解包任何 ref 属性，同时保持响应性。

官方文档中对readonly的描述:

只读代理是深层的：对任何嵌套属性的访问都将是只读的。它的 ref 解包行为与 reactive() 相同，但解包得到的值是只读的。

这意味着嵌套对象内的对象拥有和原对象一样的功能。

简单的来实践测试一下：

<script setup>
import { isReactive, isReadonly, reactive, readonly } from "vue";const original = reactive({nested: {foo: 1,},array: [{ bar: 2 }],
});const copy = readonly(original);
</script><template>{{ isReactive(original.nested) }}{{ isReactive(original.array) }}{{ isReactive(original.array[0]) }}{{ isReadonly(copy.nested) }}
</template>

页面中显示四个true。

那测试用例可以完善一下，测试之前实现的代码是否支持这样的功能。

reactive

在reactive.spec.ts中添加测试用例：

it("nested reactive", () => {const original = reactive({nested: { foo: 1 },array: [{ bar: 2 }],});expect(isReacive(original.nested)).toBe(true);expect(isReacive(original.array)).toBe(true);expect(isReacive(original.array[0])).toBe(true);
});

执行单测，如期不通过。

实现

实现思路，访问嵌套对象内对象时候，将内部对象也实现响应式，即调用reactive方法。那前提是访问的是对象，则需要添加一个是否是对象的判断。

import { isObject } from "../shared";function createGetter(isReadonly = false) {return function get(target, key) {let res = Reflect.get(target, key);if (key === ReactiveFlags.IS_REACTIVE) {return !isReadonly;} else if (key === ReactiveFlags.IS_READONLY) {return isReadonly;}if (isObject(res)) {return reactive(res);}if (!isReadonly) {track(target, key);}return res;};
}

在公共方法中导出isObject方法，

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

执行单测yarn test reactive

readonly

在readonly.spec.ts中完善对嵌套对象的断言，

it("happy path", () => {const original = { foo: 1, bar: { baz: 2 } };const wapper = readonly(original);expect(wapper).not.toBe(original);expect(wapper.foo).toBe(1);expect(isReadonly(wapper)).toBe(true);expect(isReadonly(original)).toBe(false);// 判断嵌套对象expect(isReadonly(wapper.bar)).toBe(true);
});

执行单测yarn test readonly，预期不通过。

实现

readonly的实现和reactive一致。

if (isObject(res)) {return isReadonly ? readonly(res) : reactive(res);
}

执行单测，

最后再执行全部测试用例，验证是否对其他功能存在影响。

shallowReadonly

如果你在上文中点进了官方文档的链接，在readonly的详细描述中：

要避免深层级的转换行为，请使用 shallowReadonly() 作替代。

和readonly()不同，shallowReadonly没有深层级的转换：只有根层级的属性变为了只读。属性的值都会被原样存储和暴露，这也意味着值为ref的属性不会被自动解包了。

单测

新增shallowReadonly.spec.ts。

shallowReadonly是浅层转换，还是用isReadonly进行检测，最外层的是可读的，深层次的不是。

it("should not make non-reactive properties reactive", () => {const original = reactive({ foo: 1, bar: { baz: 2 } });const wapper = shallowReadonly(original);expect(isReadonly(wapper)).toBe(true);expect(isReadonly(wapper.bar)).toBe(false);
});

实现

在reactive.ts导出shallowReadonly方法，和readonly类似，只是handler参数不同，

export function shallowReadonly(raw) {return createActiveObject(raw, shallowReadonlyHandler);
}

在baseHandler.ts导出shallowReadonlyHandler对象，

export const shallowReadonlyHandler = {}

handler中get操作都是根据createGetter方法生成，那shallowReadnoly的特点是最外层的是可读的，深层次的不是，可以像readonly一样添加一个参数来判断，如果是shallow直接返回结果，也不需要进行深层次的转换和track。

function createGetter(isReadonly = false, shallow = false) {return function get(target, key) {let res = Reflect.get(target, key);if (key === ReactiveFlags.IS_REACTIVE) {return !isReadonly;} else if (key === ReactiveFlags.IS_READONLY) {return isReadonly;}if (shallow) {return res;}if (isObject(res)) {return isReadonly ? readonly(res) : reactive(res);}if (!isReadonly) {track(target, key);}return res;};
}

全局定义：

const shallowReadonlyGet = createGetter(true, true);

shallowReadonlyHandler对象和readonlyHandler相似，只是get不同，可以复用之前实现的extend方法。

export const shallowReadonlyHandler = extend({}, readonlyHandler, {get: shallowReadonlyGet,
});

那可以顺便验证shallowReadonly的更新操作的测试用例，

it("should call console warn when call set", () => {console.warn = jest.fn();const original = shallowReadonly({ foo: 1 });original.foo = 2;expect(console.warn).toHaveBeenCalled();
});