从根上理解 React Hooks 的闭包陷阱

现在开发 React 组件基本都是用 hooks 了，hooks 很方便，但一不注意也会遇到闭包陷阱的坑。

相信很多用过 hooks 的人都遇到过这个坑，今天我们来探究下 hooks 闭包陷阱的原因和怎么解决吧。

首先这样一段代码，大家觉得有问题没：

import { useEffect, useState } from 'react';

function Dong() {

    const [count,setCount] = useState(0);

    useEffect(() => {
        setInterval(() => {
            setCount(count + 1);
        }, 500);
    }, []);

    useEffect(() => {
        setInterval(() => {
            console.log(count);
        }, 500);
    }, []);

    return <div>guang</div>;
}

export default Dong;

用 useState 创建了个 count 状态，在一个 useEffect 里定时修改它，另一个 useEffect 里定时打印最新的 count 值。

我们跑一下：

打印的并不是我们预期的 0、1、2、3，而是 0、0、0、0，这是为什么呢？

这就是所谓的闭包陷阱。

首先，我们回顾下 hooks 的原理：hooks 就是在 fiber 节点上存放了 memorizedState 链表，每个 hook 都从对应的链表元素上存取自己的值。

比如上面 useState、useEffect、useEffect 的 3 个 hook 就对应了链表中的 3 个 memorizedState：

然后 hook 是存取各自的那个 memorizedState 来完成自己的逻辑。

hook 链表有创建和更新两个阶段，也就是 mount 和 update，第一次走 mount 创建链表，后面都走 update。

比如 useEffect 的实现：

特别要注意 deps 参数的处理，如果 deps 为 undefined 就被当作 null 来处理了。

那之后又怎么处理的呢？

会取出新传入的 deps 和之前存在 memorizedState 的 deps 做对比，如果没有变，就直接用之前传入的那个函数，否则才会用新的函数。

deps 对比的逻辑很容易看懂，如果是之前的 deps 是 null，那就返回 false 也就是不相等，否则遍历数组依次对比：

所以：

如果 useEffect 第二个参数传入 undefined 或者 null，那每次都会执行。

如果传入了一个空数组，只会执行一次。

否则会对比数组中的每个元素有没有改变，来决定是否执行。

这些我们应该比较熟了，但是现在从源码理清了。

同样，useMemo、useCallback 等也是同样的 deps 处理：

理清了 useEffect 等 hook 是在哪里存取数据的，怎么判断是否执行传入的函数的之后，再回来看下那个闭包陷阱问题。

我们是这样写的：

useEffect(() => {
    const timer = setInterval(() => {
        setCount(count + 1);
    }, 500);
}, []);

useEffect(() => {
    const timer = setInterval(() => {
        console.log(count);
    }, 500);
}, []);

deps 传入了空数组，所以只会执行一次。

对应的源码实现是这样的：

如果是需要执行的 effect 会打上 HasEffect 的标记，然后后面会执行：

因为 deps 数组是空数组，所以没有 HasEffect 的标记，就不会再执行。

我们知道了为什么只执行一次，那只执行一次有什么问题呢？定时器确实只需要设置一次呀？

定时器确实只需要设置一次没错，但是在定时器里用到了会变化的 state，这就有问题了：

deps 设置了空数组，那多次 render，只有第一次会执行传入的函数：

但是 state 是变化的呀，执行的那个函数却一直引用着最开始的 state。

怎么解决这个问题呢？

每次 state 变了重新创建定时器，用新的 state 变量不就行了：

也就是这样的：

import { useEffect, useState } from 'react';

function Dong() {

    const [count,setCount] = useState(0);

    useEffect(() => {
        setInterval(() => {
            setCount(count + 1);
        }, 500);
    }, [count]);

    useEffect(() => {
        setInterval(() => {
            console.log(count);
        }, 500);
    }, [count]);

    return <div>guang</div>;
}

export default Dong;

这样每次 count 变了就会执行引用了最新 count 的函数了：

现在确实不是全 0 了，但是这乱七八遭的打印是怎么回事？

那是因为现在确实是执行传入的 fn 来设置新定时器了，但是之前的那个没有清楚呀，需要加入一段清除逻辑：

import { useEffect, useState } from 'react';

function Dong() {

    const [count,setCount] = useState(0);

    useEffect(() => {
        const timer = setInterval(() => {
            setCount(count + 1);
        }, 500);
        return () => clearInterval(timer);
    }, [count]);

    useEffect(() => {
        const timer = setInterval(() => {
            console.log(count);
        }, 500);
        return () => clearInterval(timer);
    }, [count]);

    return <div>guang</div>;
}

export default Dong;