vue3响应式模块的实现 efffect reactive ref...

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第29天,点击查看活动详情

简易版的reactive的实现

  • 实现对对象的代理
  • 一个对象只会代理一次,多次代理返回的是同一个对象
  • 对已经是代理对象的对象进行代理,返回值还是代理对象本身
  • 利用的就是代理对象在取值时会触发get钩子,我们只需要在代理对象触发get钩子的时候,直接返回代理对象即可。
  • 此时还没有实现对象属性的二次代理,如果属性值是对象,不会创建代理对象
 import { isObject } from "@vue/shared";
 ​
 /*
  * @Author: 毛毛
  * @Date: 2022-06-25 13:50:54
  * @Last Modified by: 毛毛
  * @Last Modified time: 2022-06-25 14:15:28
  */
 // 缓存已经代理过后的响应式对象
 const reactiveMap = new WeakMap();
 ​
 const enum ReactiveFlags {
   IS_REACTIVE = "__v_isReactive",
 }
 /**
  * 代理对象为响应式
  * @param obj
  */
 export function reactive(target: unknown) {
   if (!isObject(target)) return;
   const existingProxy = reactiveMap.get(target);
   // 目标对象被代理过 返回同一个代理
   if (existingProxy) return existingProxy;
   // 第一个普通对象 创建代理
   // 如果传入的对象 是已经被代理过的对象 我们可以看看这个对象是否有get方法,有表示已经是代理对象
   if (target[ReactiveFlags.IS_REACTIVE]) {
     // TODO 取到true 就返回自身 源码这一步很妙
     return target;
   }
   // 创建代理对象
   const proxy = new Proxy(target, {
     get(target, key, receiver) {
       // 用来判断是否是响应式对象
       // 对象没有被代理之前,没有该key,如果代理对象被用来二次代理,会在上面取值,然后get走到这里,返回true了
       if (key === ReactiveFlags.IS_REACTIVE) {
         return true;
       }
       return Reflect.get(target, key, receiver);
     },
     set(target, key, value, receiver) {
       const flag = Reflect.set(target, key, value, receiver);
       return flag;
     },
   });
   // 已经代理的对象进行缓存 如果再次代理同一个对象 返回同一个代理
   reactiveMap.set(target, proxy);
   return proxy;
 }

当然这样写会有点乱,我们可以把对目标对象的代理操作提取出来:

baseHandler.ts:

 /*
  * @Author: 毛毛
  * @Date: 2022-06-25 14:22:33
  * @Last Modified by: 毛毛
  * @Last Modified time: 2022-06-25 14:25:24
  */
 ​
 export const enum ReactiveFlags {
   IS_REACTIVE = "__v_isReactive",
 }
 export const mutableHandlers = {
   get(target, key, receiver) {
     // 用来判断是否是响应式对象
     // 对象没有被代理之前,没有该key,如果代理对象被用来二次代理,会在上面取值,然后get走到这里,返回true了
     if (key === ReactiveFlags.IS_REACTIVE) {
       return true;
     }
     return Reflect.get(target, key, receiver);
   },
   set(target, key, value, receiver) {
     const flag = Reflect.set(target, key, value, receiver);
     return flag;
   },
 } as ProxyHandler<object>;
 ​

这样看起来更清晰。

实现简易版的effect

该API函数是来收集副作用的。参数是一个函数,当我们函数内使用的变量的值发生改变,会让这个副作用函数重新执行。

这里,我们需要定义一个ReactiveEffect类,也就是响应式的副作用函数,用来记录用户传入的副作用函数,并对其进行一些列扩展,毕竟我们不能随意修改用户传入的函数嘛。

 /*
  * @Author: 毛毛
  * @Date: 2022-06-25 14:00:05
  * @Last Modified by: 毛毛
  * @Last Modified time: 2022-06-25 14:44:53
  */
 /**
  * 传入的副作用函数类型
  */
 type effectFn = () => any;
 export function effect(fn: effectFn) {
   // 创建响应式的effect
   const _effect = new ReactiveEffect(fn);
   // 默认先执行一次副作用函数
   const res = _effect.run();
   return res;
 }
 /**
  * 当前正在执行副作用函数暴露出去
  */
 export let activeEffect: ReactiveEffect = null;
 /**
  * 把副作用函数包装为响应式的effect函数
  */
 export class ReactiveEffect {
   /**
    * 这个effect默认是激活状态
    *
    * @memberof ReactiveEffect
    */
   active = true;
   constructor(public fn: effectFn) {}
   /**
    *
    * run 方法 就是执行传入的副作用函数
    * @memberof ReactiveEffect
    */
   run() {
     let res;
     // 激活状态 才需要收集这个副作用函数fn内用到的响应式数据 也就是我们说的依赖收集
     // 非激活状态 只执行函数 不收集依赖
     if (!this.active) {
       return this.fn();
     }
     try {
       // 激活状态 依赖收集了 核心就是将当前的effect和稍后渲染的属性关联在一起
       activeEffect = this;
       // 执行传入的fn的时候,如果出现了响应式数据的获取操作,就可以获取到这个全局的activeEffect
       res = this.fn();
     } finally {
       activeEffect = null;
     }
     return res;
   }
   /**
    * 取消副作用函数的激活 不再收集依赖
    */
   stop() {
     this.active = false;
   }
 }

此时,虽然我们还没开始进行依赖收集,但是如果使用effect函数,是正常执行用户传入的函数逻辑的。 image.png 但是,如果我们在effect中再次嵌套effect,此时可以发现,当我们内层的effect结束之后,会回到外层的effect,此时的activeEffect变量的值并不会指回外层的effect对象。

 effect(()=>{
     console.log(obj.name)
     effect(()=>{
         console.log(obj)
     })
     // .... 来到这里 activeEffect 变成 null 了
 })

所以我们可以修改记录当前正在执行的可响应式副作用对象的变量为一个栈结构。

 let activeEffect:ReactiveEffect[] = []

每次执行effect的时候,都关联栈中的最后一个元素,执行完当前的副作用函数就弹出,这样内层effect执行完,栈中的最后一个元素还是指向当前正在执行的effect

在vue3.0版本的时候,的确就是采用栈结构来实现的。

但是在最新的3.2版本,又做了一些更改。因为栈结构也是比较消耗性能的。

最新的策略是采用了类似树结构的形式,每个ReactiveEffect对象,都记录自己父ReactiveEffect对象,让activeEffect变量依然指向自身,当自己执行完毕以后,将activeEffect的值指向自己的parent属性,也就是父节点,就实现了在嵌套执行effect的时候,不会弄丢activeEffect指向的问题。

可以这样做,主要还是依赖了js是单线程。

 export class ReactiveEffect {
   /**
    * 记录父ReactiveEffect
    *
    * @type {ReactiveEffect}
    * @memberof ReactiveEffect
    */
   parent: ReactiveEffect = null;
  // ...
   run() {
  // ...
     try {
       // 激活状态 依赖收集了 核心就是将当前的effect和稍后渲染的属性关联在一起
       this.parent = activeEffect
       activeEffect = this;
       // 执行传入的fn的时候,如果出现了响应式数据的获取操作,就可以获取到这个全局的activeEffect
       res = this.fn();
     } finally {
       // 执行完当前的effect 归还上次 activeEffect 变量指向的值
       activeEffect = this.parent;
         this.parent = null
     }
     return res;
   }
 }

依赖收集

实现了reactive和effect函数,接下来就是进行依赖收集。

在副作用函数中,我们如果进行了对响应式数据的取值操作,就会触发get,来到get钩子里,就可以进行对当前对象的当前属性收集正在执行的副作用函数。

也就是说:

 当前对象 -> 取值get -> key -> effects

一个对象有多个属性,每个属性又可能在多个effect中使用,所以一个key对应多个effect,而且应该保证key对应的effect是不重复的(重复的副作用函数有必要吗?很明显没必要)

因此:我们得出可以使用map结构来记录对象和key的关系,用set来记录每个key和effects关系。

image.png

我们只需要这样做,就可以完成三者之间的映射关系。

但是,有时候我们在触发副作用函数的执行的时候,可以会出现需要清理副作用函数的情况:

比如:

image.png

这种情况再触发副作用函数执行以后,第二次外层副作用函数不会再次取age属性值,也就是说在最新执行完本次的副作用函数以后,我们的age对应的effects集合里面不应该包含外层的effect函数,只应该包含内存的effect

因此:我们可以修改一下ReactEffect,让每个effect响应式副作用的对象也收集一下自己用到的属性:

 deps: Set<ReactiveEffect>[] = [];

也就是说,属性知道那些effect用到了自己,而effect对象也知道自己用到了那些属性。也就是多对多的映射。方便我们进行清理操作。

image.png

为了便于理解,这里我们最好是把deps改为dep.

image.png

至此,我们就完成了依赖收集。下一步就是在响应式数据发生改变的情况下,重新执行副作用函数。

触发更新

我们只需要在修改值的时候,也就是触发了代理对象的set操作,在这里进行触发更新,也就是在更新值以后,触发副作用函数的重新执行。

 /**
  * 触发更新
  * @param target
  * @param type 操作类型
  * @param key
  * @param value
  * @param oldValue
  */
 export const trigger = (
   target: object,
   type: Operator,
   key: keyof any,
   value: unknown,
   oldValue: unknown
 ) => {
   const depsMap = targetMap.get(target);
   if (!depsMap) {
     // 在模板中 没有用过这个对象
     return;
   }
   // 拿到属性对应的set effects
   const effects = depsMap.get(key);
   effects && effects.forEach((effect) => effect.run());
 };

此时我们可以看见,在一秒之后,浏览器再次执行了这个副作用函数:

 const { effect, reactive } = VueReactivity
 const obj = reactive({
     name: "张三",
     address: [1],
     age: 22
 })
 let flag = true
 effect(() => {
     console.log(obj.name)
     effect(() => {
         console.log(obj.age)
     })
     if (flag) {
         console.log(obj.age)
         flag = false
     }
 })
 setTimeout(()=>{
     obj.name = '12'
 },1000)
 ​

而且,如果我们给这个对象修改address或者新增属性等,是不会触发副作用函数的执行的。

但是:如果你在effect函数中修改了当前正在触发设值操作的属性,比如外面修改了age,我在effect函数中又修改了age,很明显,会触发爆栈操作,无限循环 了

image.png

这也是很正常的情况,毕竟我们不可以要求用户不在effect中进行设值操作。

我们知道,通过前面的操作,很明显在执行当前副作用函数的时候,我们会把activeEffect的值设置为当前正在执行的响应式副作用对象。因此,当我们发现正在执行的副作用函数activeEffect的值,和我们下一个将要执行的effect的值是一样的时候,跳过此函数的执行即可。

image.png

到现在,我们就实现了一个简易版的依赖收集和触发更新的操作了。当然我们还有很多情况没有考虑,这里只是简单的实现。

到现在实现了一个简单的effect函数。后续的watch,watchEffect等函数都是通过该函数实现的。

响应式分支切换原理

image.png

一个很简单的分支代码,我们发现在修改flag的值为false以后,此时修改name属性的值,依然会触发effect的更新操作。这是为什么?

讲道理来说,我们这里的flag为false,表示我们在effect执行的时候,是不会收集到对name属性的依赖,所以在后面更新name属性值的时候,不应该再次触发effect的执行,但是结果和我们想的有出入:

我们如何保证在这种分支切换的时候,取消对本次不需要依赖属性的变更?

这就需要我们在执行effect之前,清空上次已经收集的属性依赖。

那么我们肯定可以想到如下的操作:在执行effect之前,清空dep set里对当前effect的引用,然后清空当前effect里面收集的所有deps。

 /**
  * 清除effect收集的dep set里 每个属性对当前effect的收集
  * @param effect
  */
 const cleanupEffect = (effect: ReactiveEffect) => {
   const { deps } = effect;
   for (let i = 0; i < deps.length; i++) {
     // 解除key -> effect的关联 执行effect的时候重新收集
     deps[i].delete(effect);
   }
   // 清空当前effect依赖的dep
   effect.deps.length = 0;
 };
 ​

image.png

看似一切正常,两边的绑定都解除了。但是会发现,控制台疯狂输出打印age的值!!!

image.png

有点吓人,按理说没毛病,为什么会出现这种情况?

原因出在执行cleanupEffect函数下面的哪一行代码:也就是执行 this.fn()

我们刚把引用清除,这马上就加了进来。

有点像如下操作:

 const set = new Set([1,2,3])
 set.forEach((val)=>{
     set.delete(1)
     set.add(1)
 })

知道问题,我们就可以解决了。我们在循环执行副作用函数的时候,也就是触发更新哪里,不要直接操作原依赖set集合,把effects集合拷贝一份,然后循环执行再这个拷贝的数组上进行,不对当前还需要收集依赖的set集合上进行执行操作。

image.png

effect函数返回值

对于effect函数,其实是有返回值的。

我们前面给ReactiveEffect对象都有一个stop方法,用来停止依赖收集。所以我们可以让effect方法返回当前effect.run方法,调用这个返回值可以再次开始收集依赖。

image.png

 export function effect(fn: effectFn) {
   // 创建响应式的effect
   const _effect = new ReactiveEffect(fn);
   // 默认先执行一次副作用函数
   _effect.run();
   // effect函数的返回值就是一个runner 可以让失活的effect再次收集依赖
   const runner = _effect.run.bind(_effect); // 绑定this
   runner.effect = _effect; // 将effect挂载到runner上
   return runner;
 }

这样更改代码以后,我们可以在页面手动让effect失活,但是可以通过effect函数的返回值再次执行run函数。不过只能手动触发执行,该effect不会再次收集依赖了。

image.png

image.png

效果上我们依然可以手动触发执行effect,但是这就类似于手动执行函数了。 可以看见后面的更新属性也不会触发effect的执行了。stop函数主要是为了vue3.2新增的effectScope方法服务的。

vue调度器

实现了上述操作,可以发现更新都是同步的,也就是我们同步连续两次修改同一个属性,会触发两次effect的执行。

image.png

image.png

这个从同步代码来说是没有任何问题的。你修改了两次name属性,必须会触发两次更新。

但是如果我们在一次同步代码的执行过程中,修改了几次name属性,就需要触发几次effect的更新,是不是有点消耗性能?其实我们想要的效果就是最后一次修改值生效,也就是只有同步代码执行完,才去触发依赖effect的执行。那么只需要执行一次effect函数即可。

通过上面的操作,我们对effect函数的执行,要不然通过effect的返回值自己控制执行函数,要不然就是通过依赖收集以后自动触发effect的执行。

有时候我们可能需要自己控制effect的执行,也就是希望把传入的函数的执行权归还给我们。这就是调度器的出现由来。

实际上在vue内部,也是通过调度器来实现多次更新批处理(合并,只处理最后一次),以及异步更新等。

在这里只是实现调度器,暂时并没有实现异步更新和批处理等。

更新一下我们的effect参数

 interface EffectOptions {
   scheduler?: EffectScheduler;
 }
 // 调度器
 type EffectScheduler = (effect: ReactiveEffect) => any;
 ​
 /**
  * 创建响应式的effect
  * @param fn 副作用函数
  * @param options 配置对象
  * @returns
  */
 export function effect(fn: effectFn, options?: EffectOptions) {
   // 创建响应式的effect
   const _effect = new ReactiveEffect(fn, options?.scheduler);
   // 默认先执行一次副作用函数
   _effect.run();
   // effect函数的返回值就是一个runner 可以让失活的effect再次执行 只是需要手动触发执行了(不会自动开始收集依赖)
   const runner = _effect.run.bind(_effect); // 绑定this
   runner.effect = _effect; // 将effect挂载到runner上
   return runner;
 }

然后就是更新一下ReactiveEffect的构造函数

 constructor(public fn: effectFn, public scheduler?: EffectScheduler) {}

最后就是在触发更新的时候,把我们的直接执行run方法,修改为在用户传递调度器的情况下,执行用户的逻辑

   if (effects) {
     // 把依赖effects拷贝一份 我们的执行操作在这个数组上 不直接操作原set集合了
     const fns = [...effects];
     fns.forEach((effect) => {
       if (effect !== activeEffect) {
         if (effect.scheduler) {
           // 用户传入了调度函数,使用用户的
           effect.scheduler(effect);
         } else {
           effect.run();
         }
       }
     });
   }

实现属性递归代理

在前面的实现代理的地方,也就是get方法哪里,我们并没有处理被代理对象属性仍然是一个对象的情况,对于属性是对象的时候,我们仍然应该再取值的时候,对该属性进行代理。vue3采取的就是再获取值的时候,如果发现是对象,对该属性进行懒代理。

   get(target, key, receiver) {
     // 用来判断是否是响应式对象
     // 对象没有被代理之前,没有该key,如果代理对象被用来二次代理,会在上面取值,然后get走到这里,返回true了
     if (key === ReactiveFlags.IS_REACTIVE) {
       return true;
     }
     track(target, "get", key);
     // 取值操作
     const value = Reflect.get(target, key, receiver);
     if (isObject(value)) {
       // 属性值是对象 懒代理
       return reactive(value);
     }
     return value;
   }

image.png

image.png

此时可以看见,打印出来的address对象也是响应式的。

而且,如果你没有手动获取过属性是对象的值,那么是不会对该属性进行代理的。

很明显:这样的方式会显著的提示性能,对于vue2是一上来就递归代理所有属性。

计算属性的实现

计算属性,最大的特点就是缓存。计算属性中所依赖的值不发生改变,我们不会重新计算最新的计算属性的值,而是取缓存中的值。

而且,computed也是基于effect实现的。

计算属性也是懒执行的,也就是说,定义的计算属性,如果没有使用这个计算属性的值,我们是不会真的计算值的。

所以,我们在实现计算属性的时候,内部应该是有一个标识的,表名当前计算属性的值是否是脏的,如果是脏的,表示依赖的属性发生改变,需要重新计算,那么在下次取值的时候,就会重新计算最新的值。

所以我们可以认为计算属性就是一个懒执行,有缓存的effect。而且,计算属性因为我们当做属性,所以在取计算属性的值的时候,也会收集所在的effect添加到set上。

 import { ReactiveEffect } from "./effect";
 import { isFunction } from "@vue/shared";
 ​
 /*
  * @Author: 毛毛
  * @Date: 2022-06-26 13:48:11
  * @Last Modified by: 毛毛
  * @Last Modified time: 2022-06-26 14:19:41
  */
 type ComputedGetter = () => any;
 type ComputedSetter = (newVal) => void;
 type ComputedOptions =
   | ComputedGetter
   | {
       get: ComputedGetter;
       set: ComputedSetter;
     };
 export const computed = (getterOrOptions: ComputedOptions) => {
   const onlyGetter = isFunction(getterOrOptions);
   let getter;
   let setter;
   if (onlyGetter) {
     getter = getterOrOptions;
     setter = () => {
       console.warn(`不能修改一个只读的计算属性的值!`);
     };
   } else {
     getter = getterOrOptions.get;
     setter = getterOrOptions.set;
   }
   return new ComputedRefImpl(getter, setter);
 };
 ​
 /**
  * 计算属性的实现
  */
 class ComputedRefImpl {
   private effect: ReactiveEffect;
   // 默认是脏数据 取值的时候会重新计算
   private _dirty = true;
   // 只读
   private __v_isReadonly = true;
   // ref
   private __v_isRef = true;
   // 缓存的计算属性值
   private _value;
   constructor(public getter: ComputedGetter, public setter: ComputedSetter) {
     /*
       计算属性内部就是通过可响应式的effect实现的,通过调度器来自己执行getter
       执行getter的时候 很显然会收集依赖 在依赖变更的时候 会执行我们的调度函数
     */
     this.effect = new ReactiveEffect(getter, (effect) => {
       // 依赖的属性变化 会执行该调度函数
       if (!this._dirty) {
         this._dirty = true; 
       }
     });
   }
   /**
    * 类的属性访问器 -> 其实就是 definePrototype
    *
    * @memberof ComputedRefImpl
    */
   get value() {
     // 先判断是否是脏值
     if (this._dirty) {
       this._value = this.effect.run();
       // 计算最新值了  不是脏值了
       this._dirty = false;
     }
     return this._value;
   }
   set value(newVal) {
     this.setter(newVal);
   }
 }
 ​

此时:我们可以看见在effect函数中,取计算属性的值的时候,是可以正常拿到计算属性的值的

但是在依赖的时候发生更新以后,我们并没有触发effect的重新执行。这是为什么?

我们在调度器中已经修改了计算属性的值是脏值的标记,为什么不会重新执行effect?

     const { effect, reactive, computed } = VueReactivity
     const nums = reactive({
       num1: 10,
       num2: 20
     })
     const res = computed(() => nums.num1 + nums.num2)
     effect(() => {
       console.log(res.value)
     })
     setTimeout(() => {
       nums.num1 = 20
     }, 1000)

image-20220626142232533

原因在于:我们的计算属性此时只是修改了脏值标记,并没有触发计算属性ReactiveEffect的执行。

我们前面说过,计算属性应该也可以收集外层effect,这样在计算属性更新的时候,会自动触发effect的再次执行。

那么我们在取计算属性值的时候,应该进行依赖收集,在计算属性依赖的属性发生更新以后,也应该在修改脏值以后,触发更新。这样在更新的时候就会触发外层effect的执行。

首先优化一下我们的track方法,也就是依赖收集的方法:

 export const track = (target: object, type: Operator, key: keyof any) => {
   // 不是在effect使用,不需要收集依赖
   if (!activeEffect) return;
   let depsMap = targetMap.get(target);
   if (!depsMap) {
     targetMap.set(target, (depsMap = new Map()));
   }
   let dep = depsMap.get(key);
   if (!dep) {
     depsMap.set(key, (dep = new Set()));
   }
   trackEffects(dep);
 };
 /**
  * 收集当前正在执行的effect 放入dep中
  * @param dep
  * @returns
  */
 export const trackEffects = (dep: Set<ReactiveEffect>) => {
   if (!activeEffect) return;
   // 判断是否有当前activeEffect
   // 已经收集过 不需要再次收集 这种情况一般是一个副作用函数中多次使用了该属性
   const shouldTrack = !dep.has(activeEffect);
   if (shouldTrack) {
     // key -> effect [eff1,eff2,...]
     dep.add(activeEffect);
     // 后续方便进行清理操作
     activeEffect.deps.push(dep);
   }
 };

提取一个trackEffect方法,用以方便我们在计算属性这里收集effect

   get value() {
     // 收集依赖
     trackEffects(this.dep || (this.dep = new Set()));
     // 先判断是否是脏值
     if (this._dirty) {
       this._value = this.effect.run();
       // 计算最新值了  不是脏值了
       this._dirty = false;
     }
     return this._value;
   }

优化一下我们的trigger方法:

 export const trigger = (
   target: object,
   type: Operator,
   key: keyof any,
   value?: unknown,
   oldValue?: unknown
 ) => {
   const depsMap = targetMap.get(target);
   if (!depsMap) {
     // 在模板中 没有用过这个对象
     return;
   }
   // 拿到属性对应的set effects
   const effects = depsMap.get(key);
   // 防止死循环 刚删除的引用马上又添加进来
   if (effects) {
     triggerEffects(effects);
   }
 };
 /**
  * 触发执行effects
  * @param effects
  */
 export const triggerEffects = (effects: Set<ReactiveEffect>) => {
   // 把依赖effects拷贝一份 我们的执行操作在这个数组上 不直接操作原set集合了
   const fns = [...effects];
   fns.forEach((effect) => {
     if (effect !== activeEffect) {
       if (effect.scheduler) {
         // 用户传入了调度函数,使用用户的
         effect.scheduler(effect);
       } else {
         effect.run();
       }
     }
   });
 };

这样我们就可以在计算属性的调度函数哪里,进行触发计算属性的更新操作了。

 this.effect = new ReactiveEffect(getter, (effect) => {
     // 依赖的属性变化 会执行该调度函数
     if (!this._dirty) {
         this._dirty = true;
         // 触发更新
         triggerEffects(this.dep);
     }
 });

可以看见控制台打印,在num1发生改变以后,计算属性会进行重新计算,然后导致effect的重新执行。

watch实现

watch可以实现对一个对象监控,当对象的属性发生更新的时候,会执行我们的回调函数。当然参数类型不仅仅是可以传入一个对象,也可以传入一个getter函数,甚至参数还可以是一个数组,也就是同时监控多个对象的变化。但是对于对象来说,实际是不区分新值和老值的,因为新旧对象都是同一个对象。

当我们监控一个对象的时候,其实就相当于让这个对象的每个属性都收集一下watch函数的第二个参数,也就是吧回调函数作为effect,让每个属性都收集。

首先对于观测的对象,肯定需要是响应式的:

 export const isReactive = (val: unknown):boolean => {
   return !!(val && val[ReactiveFlags.IS_REACTIVE]);
 };

然后就可以实现watch了

 import { isReactive } from "./reactive";
 import { ReactiveEffect } from "./effect";
 import { isFunction, isObject } from "@vue/shared";
 /**
  * @Author: 毛毛
  * @Date: 2022-06-26 15:15:59
  * @Last Modified by: 毛毛
  * @Last Modified time: 2022-06-26 15:48:26
  * @description watch API 实际是不在reactivity 包里面的
  */
 type WatchTarget = object | (() => any);
 type WatchCallback = (newVal, oldVal) => any;
 interface WatchOptions {
   immediate?: boolean;
   deep?: boolean;
 }
 /**
  *
  * @param source 对象 或者 getter函数
  * @param callback 回调函数
  */
 export const watch = (
   source: WatchTarget,
   callback: WatchCallback,
   { immediate = false, deep = false }: WatchOptions = {}
 ) => {
   let getter;
   let oldVal;
   if (isFunction(source)) {
     getter = source;
   } else if (isReactive(source)) {
     // 是一个响应式对象 需要对该对象的所有属性进行循环访问一次 递归循环
     getter = () => traversal(source);
   }
   const job = () => {
     // 新值
     const newVal = effect.run();
     callback(oldVal, newVal);
     oldVal = newVal;
   };
   const effect = new ReactiveEffect(getter, job);
   // 记录老值
   oldVal = effect.run();
   if (immediate) {
     // 立即执行一次回调
     callback(oldVal, undefined);
   }
 };
 ​
 /**
  * 考虑循环引用 遍历对象的每一个属性
  * @param val
  * @param set
  */
 const traversal = (val: object, set = new WeakSet()) => {
   if (!isObject(val)) return val;
   if (set.has(val)) return val;
   set.add(val);
   for (const key in val) {
     traversal(val[key]);
   }
   return val;
 };
 ​

此时可以传入getter,或者直接传入一个响应式对象,都可以进行观测。如果需要立即执行一次回调函数,也可以传入参数。

 const { effect, reactive, computed, watch } = VueReactivity
 const nums = reactive({
     num1: 10,
     num2: 20
 })
 watch(nums, (newVal, oldVal) => {
     console.log(newVal, oldVal)
 }, { immediate: true })
 setTimeout(() => {
     nums.num1 = 20
 }, 2000)

image.png

可以看出此时已经实现了对对象的观测,getter函数也是可以的。但是这里getter函数的返回值不要是一个对象,那样会观测失败的。

实际上在vue里面,如果传入的是一个getter,也是不能让返回值直接是一个响应式对象的。

 watch(() => nums, (newVal, oldVal) => {
     console.log(newVal, oldVal)
 }, { immediate: true })
 setTimeout(() => {
     nums.num1 = 20
 }, 2000)

此时是无法执行回调函数的,。。

甚至你还可以取消对某次的操作。比如发起请求等。也就是清除副作用,比如用户在输入框输入文本以后,发起请求,如果发起了两次请求,第二次请求比第一次先回来,此时我们就应该清除第一次请求的副作用。

 export const watch = (
   source: WatchTarget,
   callback: WatchCallback,
   { immediate = false, deep = false }: WatchOptions = {}
 ) => {
   let getter;
   let oldVal;
   if (isFunction(source)) {
     getter = source;
   } else if (isReactive(source)) {
     // 是一个响应式对象 需要对该对象的所有属性进行循环访问一次 递归循环
     getter = () => traversal(source);
   }
   // 保存用户传入的清理函数
   let cleanupCb: null | (() => void);
   const onCleanup = (cb: () => void) => {
     cleanupCb = cb;
   };
   const job = () => {
     // 在下次watch触发effect的执行前 先执行用户传入的回调
     if (cleanupCb) {
       cleanupCb();
       cleanupCb = null;
     }
     // 新值
     const newVal = effect.run();
     callback(oldVal, newVal, onCleanup);
     oldVal = newVal;
   };
   const effect = new ReactiveEffect(getter, job);
   // 记录老值
   oldVal = effect.run();
   if (immediate) {
     // 立即执行一次回调
     callback(oldVal, undefined, onCleanup);
   }
 };

比如如下场景,我们只会渲染第二次请求获取到的数据到页面上,第一次获取的数据我们并不会使用了,可以理解为屏蔽了这个副作用的执行。

 function getData(time) {
     return new Promise((resolve) => {
         setTimeout(resolve, time, `我是${time}ms后请求到的数据`)
     })
 }
 const obj = reactive({
     data: ''
 })
 let time = 2000
 watch(() => obj.data, async (newVal, oldVal, onCleanup) => {
     let clean = true;
     onCleanup(() => {
         clean = false;
     })
     time -= 1000
     const data = await getData(time)
     if (clean) {
         app.innerHTML = data
     }
 })
 obj.data+='1'
 obj.data+='1'

image.png

ref

通过前面的学习,我们知道,reactive能代理的必须是对象。也就是说我们无法代理一个基本类型的数据,比如:字符串,数字,或者一个布尔类型的标记数据。

有时候我们想控制一个loading动画是否显示,在正在请求数据的时候加载loading动画,但是数据回来以后渲染组件,此时我们只需要一个布尔类型的数据作为标记就行了,不需要弄成一个对象,然后在对象里面通过某个属性值来进行判断。

所以为了解决reactive api只能代理对象的情况,vue3推出了refAPI,可以把一个基本类型的数据变成响应式的。

当然了,本质上肯定不可能真的把一个基本类型的数据变成响应式的,基本类型数据是值类型的,值是肯定不会发生改变的。

所以其实vue就在基本类型外部包了一层,将其变成一个对象形式。取值方式就类似于计算属性,通过.value拿到实际的值,。

 export const ref = <T>(value: T) => {
   return new RefImpl(value);
 };
 ​
 class RefImpl<T> {
   private __v_isRef = true;
   private _value: T;
   private rawValue: T; // 原始数据
   private dep:Set<ReactiveEffect> = new Set()
   constructor(_value: T) {
     this._value = toReactive(_value);
     this.rawValue = _value;
   }
   get value() {
     // 依赖收集
     trackEffects(this.dep)
     return this._value;
   }
   set value(newVal) {
     if (newVal !== this.rawValue) {
       this._value = toReactive(newVal);
       this.rawValue = newVal;
       // 触发更新
       triggerEffects(this.dep);
     }
   }
 }
 /**
  * 如果是对象 转为响应式对象
  * @param value
  */
 const toReactive = (value: unknown) => {
   return isObject(value) ? reactive(value) : value;
 };
 const { effect, reactive, computed, watch,ref } = VueReactivity
 const val = ref(10)
 let flag = true
 effect(()=>{
     if(flag){
         console.log(val.value)
     }
 })
 setTimeout(()=>{
     flag = false
     val.value = 20
     console.log(val)
 },1000)

ref的实现和计算属性有点类似,都是通过.vulue访问值。

toRef和toRefs

前面我们的 响应式对象都是通过reactive包裹一层,变成代理对象。

 const { effect, reactive, computed, watch,ref } = VueReactivity
 const obj = reactive({
     name:"zs",
     age:12,
 })
 effect(()=>{
     app.innerHTML = `我是${obj.name}, 我今年 ${obj.age}岁了!`
 })
 setTimeout(()=>{
     obj.age = 2
 })

这么使用当然是没有任何问题的,有时候,这个响应式的对象属性特别多,我们想解构了以后再使用:

 const obj = reactive({
     name:"zs",
     age:12,
 })
 const {name,age} = obj
 effect(()=>{
     app.innerHTML = `我是${name}, 我今年 ${age}岁了!`
 })
 setTimeout(()=>{
     obj.age = 2
 })

此时,这个解构出来的数据就丧失响应式数据的性质了。我们要如何让其仍然具有响应式的特性?

答案就是toReftoRefs、。我们在解构的时候,遍历其对象上的属性,将其转为ref,然后在修改ref的值的时候,代理为还是修改原响应式reactive对象上属性的修改,这样就仍然可以保证响应式的特性不丧失。

 export const toRef = <T>(object: T, key) => {
   return new ObjectRefImpl(object, key);
 };
 ​
 export const toRefs = <T>(object: object | T[]) => {
   const res = isArray(object) ? new Array(object.length) : {};
   for (const key in object) {
     res[key] = toRef(object, key);
   }
   return res;
 };
 ​
 class ObjectRefImpl {
   constructor(private object, private key) {}
   get value() {
     return this.object[this.key];
   }
   set value(newVal) {
     this.object[this.key] = newVal;
   }
 }

此时,可以使用.value的形式访问响应式数据,对ref形式数据的修改,实际上还是代理到了原reactive对象上的操作。

 const { effect, reactive, computed, watch, ref, toRefs } = VueReactivity
 const obj = reactive({
     name: "zs",
     age: 12,
 })
 const { name, age } = toRefs(obj)
 effect(() => {
     app.innerHTML = `我是${name.value}, 我今年 ${age.value}岁了!`
 })
 setTimeout(() => {
     name.value = 'ls' // 页面更新
 })

对于ref的数据,我们在template模板中取值的时候,是不需要使用.value来取真实值的,vue会在我们取name属性的时候,代理到name.value上。

实际上在vue的模板里,就是使用了proxyRefs方法来实现的。

proxyRefs

使用该API,可以把ref的数据转为reactive的形式,和上面的toRefs刚好反过来。

 export const proxyRefs = <T>(obj: RefImpl<T>) => {
   return new Proxy(obj, {
     get(target, key, receiver) {
       // 看key是否是一个ref
       const res = Reflect.get(target, key, receiver);
       return res.__v_isRef ? res.value : res;
     },
     set(target, key, value, receiver) {
       const oldVal = Reflect.get(target, key);
       if (oldVal.__v_isRef) {
         oldVal.value = value;
         return true;
       } else {
         return Reflect.set(target, key, value, receiver);
       }
     },
   });
 };
 const name = ref("zs")
 const age = ref(12)
 const _ctx = proxyRefs({ name, age })
 effect(() => {
     app.innerHTML = `我是${_ctx.name}, 我今年 ${_ctx.age}岁了!`
 })
 setTimeout(()=>{
     _ctx.name = 'ls'
 },1000)

可以看见,视图正常更新。

当然,这个API我们很少使用,这个是vue内部在模板解析的时候所使用的。大家使用的时候也有感受,就是ref的数据在模板里面是不需要.value的,其实就是这个API帮我们做好的。

猜你喜欢

转载自juejin.im/post/7113892191725944839