React官方团队完善原生Hook闭包陷阱

目录

正文useEventuseEvent的实现与开源Hooks的差异总结

正文

我们知道，Hooks使用时存在所谓的闭包陷阱，考虑如下代码：

1. function Chat() {
2.   const [text, setText] = useState('');
3.   const onClick = useCallback(() => {
4.     sendMessage(text);
5.   }, []);
6.   return <SendButton onClick={onClick} />;
7. }

复制代码

我们期望点击后sendMessage能传递text的最新值。
然而实际上，由于回调函数被useCallback缓存，形成闭包，所以点击的效果始终是sendMessage('')。
这就是闭包陷阱。
以上代码的一种解决方式是为useCallback增加依赖项：

1. const onClick = useCallback(() => {
2.   sendMessage(text);
3. }, [text]);

复制代码

但是这么做了后，每当依赖项（text）变化，useCallback会返回一个全新的onClick引用，这就失去了useCallback缓存函数引用的作用。
闭包陷阱的出现，加大了Hooks的上手门槛，也让开发者更容易写出有bug的代码。
现在，React官方团队要出手解决这个问题。

useEvent

解决方式是引入一个新的原生Hook —— useEvent。
他用于定义一个函数，这个函数有2个特性：

在组件多次render时保持引用一致函数内始终能获取到最新的props与state

上面的例子使用useEvent改造后：

1. function Chat() {
2.   const [text, setText] = useState('');
3.   const onClick = useEvent(() => {
4.     sendMessage(text);
5.   });
6.   return <SendButton onClick={onClick} />;
7. }

复制代码

在Chat组件多次render时，onClick始终指向同一个引用。
并且onClick触发时始终能获取到text的最新值。
之所以叫useEvent，是因为React团队认为这个Hook的主要应用场景是：封装事件处理函数。

useEvent的实现

useEvent的实现并不困难，代码类似如下：

1. function useEvent(handler) {
2.   const handlerRef = useRef(null);
3.   // 视图渲染完成后更新`handlerRef.current`指向
4.   useLayoutEffect(() => {
5.     handlerRef.current = handler;
6.   });
7.   // 用useCallback包裹，使得render时返回的函数引用一致
8.   return useCallback((...args) => {
9.     const fn = handlerRef.current;
10.     return fn(...args);
11.   }, []);
12. }

复制代码

整体包括两部分：

返回一个没有依赖项的useCallback，使得每次render时函数的引用一致

1. useCallback((...args) => {
2.   const fn = handlerRef.current;
3.   return fn(...args);
4. }, []);

复制代码

在合适的时机更新handlerRef.current，使得实际执行的函数始终是最新的引用

与开源Hooks的差异

很多开源Hooks库已经实现类似功能（比如ahooks中的useMemoizedFn）
useEvent与这些开源实现的差异主要体现在：
useEvent定位于处理事件回调函数这一单一场景，而useMemoizedFn定位于缓存各种函数。
那么问题来了，既然功能类似，那useEvent为什么要限制自己的使用场景呢？
答案是：为了更稳定。
useEvent能否获取到最新的state与props取决于handlerRef.current更新的时机。
在上面模拟实现中，useEvent更新handlerRef.current的逻辑放在useLayoutEffect回调中进行。
这就保证了handlerRef.current始终在视图完成渲染后再更新：

1. useLayoutEffect(() => {
2.   handlerRef.current = handler;
3. });

复制代码

而事件回调触发的时机显然在视图完成渲染之后，所以能够稳定获取到最新的state与props。
注：源码内的实际更新时机会更早些，但不影响这里的结论
再来看看ahooks中的useMemoizedFn，fnRef.current的更新时机是useMemoizedFn执行时（即组件render时）：

1. function useMemoizedFn<T extends noop>(fn: T) {
2.   const fnRef = useRef<T>(fn);
3.   // 更新fnRef.current
4.   fnRef.current = useMemo(() => fn, [fn]);
5.   // ...省略代码
6. }

复制代码

当React18启用并发更新后，组件render的次数、时机并不确定。
所以useMemoizedFn中fnRef.current的更新时机也是不确定的。
这就增加了在并发更新下使用时潜在的风险。
可以说，useEvent通过限制handlerRef.current更新时机，进而限制应用场景，最终达到稳定的目的。

总结

useEvent当前还处于RFC（Request For Comments）阶段。
很多热心的开发者对这个Hook的命名提出了建议，比如：useStableCallback：

又比如：useLatestClosure：

从这些命名看，他们显然扩大了useEvent的应用场景。
经过本文的分析我们知道，扩大应用场景意味着增加开发者使用时出错的风险，更多关于React Hook闭包陷阱完善的资料请关注中国红客联盟其它相关文章！