从源码理清 useEffect 第二个参数是怎么处理的

useEffect 是常用的 hook，它支持两个参数，第一个参数是回调函数，第二个参数是依赖。

当第二个参数为 null 或 undefined 的时候，回调函数每次 render 都会执行，而参数为数组的时候，只有依赖项变了才会执行。

这些我们都很熟悉了，但它是怎么实现的呢？我们来从源码来找下答案。

useEffect 的第二个参数

我们先来试一下第二个参数传入 undefined、空数组、有依赖的数组时的效果。

准备这样一段代码：

import { useEffect, useRef, useState } from 'react';

function Dong() {
    const ref = useRef(1);

    const [,setState] = useState();

    useEffect(() => {
        console.log(111);
    });

    useEffect(() => {
        console.log(222);
    }, []);

    useEffect(() => {
        console.log(333);
    }, [ref.current]);

    useEffect(() => {
        setInterval(() => {
            setState([]);
        }, 1000);
    
        setTimeout(() => {
            ref.current = 2;
        }, 3000);
    }, []);

    return <div>dongdiv>;
}

我们用写了三个 useEffect，第二个参数分别为 undefined、[]、有一个依赖的数组，回调函数里分别打印 111、222、333。

然后 useState 声明了一个 state，用 setInterval 定时修改，这样能不断触发 render。

又用 useRef 声明了一个对象，它的特点是每次 render 都是返回的同一个对象，我们用 setTimeout 在 2s 后修改了它的值。

执行的结果大家应该很容易想到：

111 每次都会打印，因为第二个参数为 undefined。

222 只打印一次，因为第二个参数为 []。

333 打印两次，因为第二个参数有一个依赖，这个依赖在 2s 的时候会变一次。

这些我们都很熟悉了，但是它为什么是这样呢？

我们来看下源码：

useEffect 相关源码

react hooks 的原理前面一篇文章写过，我们再过一遍：

jsx 编译产生 render function，执行返回 vdom，但是为了提高性能，React 16 引入 fiber 架构，会先把 vdom 转成 fiber，然后再去更新到 dom。

vdom 转 fiber 的过程叫做 reconcile，更新到 dom 的过程叫做 commit。reconcile 的过程是可打断的，需要 schedule。

hooks 也是基于 fiber 来实现的，它在 fiber 节点上维护了一个链表（memorizedState 属性），用来保存数据，每个 hook 都是从对应的链表元素上存取各自的数据。

比如上面那个组件的 6 个 hook 就对应着 fiber 节点上 memorizedState 链表的 6 个元素：

每个 hook 都是在对应的链表元素上存取数据的。

这个链表有个建立的过程，叫做 mount，后面只需要 update，所以每个 hook 的实现都会分为 mount 和 update 两个阶段。

我们看下 useEffect 相关的源码：

它也是分为了 mountEffect 和 updateEffect 两个函数，最终都是在 hook.memorizedState 存取元素的。这就是 hook 的通用原理。

第二个参数对应的就是 deps，它是怎么判断是否要更新的呢？

我们着重看下这段逻辑：

deps 是新传入的参数，如果是 undefined 会作为 null。

hook.memorizedState.deps 取到的是之前的 deps。

然后新旧 deps 会做下对比，如果返回 true 才会执行 effect。

对比的逻辑在 areHookInputsEqual 这个函数里：

如果 prevDeps 是 null，那就直接返回 false，这就是 useEffect 第二个参数传 undefined 或者 null 的话 effect 函数都会执行的原因。

否则，才会新旧的 deps 数组中每个元素做对比，有一个不一样就返回 false。

这已经解释了上面那个案例，deps 数组传 undefined、[]、[dep] 时 effect 执行的不同情况。

其实还有一种情况也会导致 effect 执行，就是上面这段逻辑：

当热更新的时候，就算依赖没有变，也需要重新执行 effect，这个是通过 ignorePreviousDependencies 变量来控制的。

这个估计很多人都不知道，因为热更新是工具实现的。

我们从源码层面解释清楚了 useEffect 第二个参数的处理机制。

其实 useCallback、useMemo 的 deps 参数处理逻辑也是一样的，源码都差不多：

总结

useEffect 第二个参数传入 undefined、[]、[a,b,c] 时执行的效果不同， undefined 每次都会执行，而依赖数组只有在依赖变了才会执行，空数组只会执行一次。

我们从源码层面解释了原因：

hooks 是在 fiber 节点的 memorizedState 属性上存取数据的，会组织一个和 hook 一一对应的链表。

构建这个链表的阶段叫 mount，后面只需要 update，所以所有的 hook 的实现都分为了 mountXxx 和 updateXxx 两部分。

useEffect 在 update 时会对比新传入的 deps 和之前存在 memorizedState 上的 deps 来确定是否执行 effect 回调，它做了这样的处理：

当 dep 是 null（undefined 也会处理成 null）时，判定为不相等。如果是热更新的时候，判定为不相等。否则会对比数组的每个依赖项来判断是否相等。只要新旧 deps 不相等就执行 effect。

useCallback、useMemo 的 deps 处理也是一样的，我们从源码层面理清楚了 deps 参数的处理机制。