React官方團隊完善原生Hook閉包陷阱

更新時間：2022年07月11日 14:36:15 作者：卡頌

這篇文章主要為大家介紹了React官方團隊出手，補齊原生Hook短板閉包陷阱的示例詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪

正文

我們知道，Hooks使用時存在所謂的閉包陷阱，考慮如下代碼：

function Chat() {
  const [text, setText] = useState('');
  const onClick = useCallback(() => {
    sendMessage(text);
  }, []);
  return <SendButton onClick={onClick} />;
}

我們期望點擊后sendMessage能傳遞text的最新值。

然而實際上，由于回調(diào)函數(shù)被useCallback緩存，形成閉包，所以點擊的效果始終是sendMessage('')。

這就是閉包陷阱。

以上代碼的一種解決方式是為useCallback增加依賴項：

const onClick = useCallback(() =&gt; {
  sendMessage(text);
}, [text]);

但是這么做了后，每當依賴項（text）變化，useCallback會返回一個全新的onClick引用，這就失去了useCallback緩存函數(shù)引用的作用。

閉包陷阱的出現(xiàn)，加大了Hooks的上手門檻，也讓開發(fā)者更容易寫出有bug的代碼。

現(xiàn)在，React官方團隊要出手解決這個問題。

useEvent

解決方式是引入一個新的原生Hook —— useEvent。

他用于定義一個函數(shù)，這個函數(shù)有2個特性：

在組件多次render時保持引用一致
函數(shù)內(nèi)始終能獲取到最新的props與state

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

function Chat() {
  const [text, setText] = useState('');
  const onClick = useEvent(() => {
    sendMessage(text);
  });
  return <SendButton onClick={onClick} />;
}

在Chat組件多次render時，onClick始終指向同一個引用。

并且onClick觸發(fā)時始終能獲取到text的最新值。

之所以叫useEvent，是因為React團隊認為這個Hook的主要應用場景是：封裝事件處理函數(shù)。

useEvent的實現(xiàn)

useEvent的實現(xiàn)并不困難，代碼類似如下：

function useEvent(handler) {
  const handlerRef = useRef(null);
  // 視圖渲染完成后更新`handlerRef.current`指向
  useLayoutEffect(() => {
    handlerRef.current = handler;
  });
  // 用useCallback包裹，使得render時返回的函數(shù)引用一致
  return useCallback((...args) => {
    const fn = handlerRef.current;
    return fn(...args);
  }, []);
}

整體包括兩部分：

返回一個沒有依賴項的useCallback，使得每次render時函數(shù)的引用一致

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

在合適的時機更新handlerRef.current，使得實際執(zhí)行的函數(shù)始終是最新的引用

與開源Hooks的差異

很多開源Hooks庫已經(jīng)實現(xiàn)類似功能（比如ahooks中的useMemoizedFn）

useEvent與這些開源實現(xiàn)的差異主要體現(xiàn)在：

useEvent定位于處理事件回調(diào)函數(shù)這一單一場景，而useMemoizedFn定位于緩存各種函數(shù)。

那么問題來了，既然功能類似，那useEvent為什么要限制自己的使用場景呢？

答案是：為了更穩(wěn)定。

useEvent能否獲取到最新的state與props取決于handlerRef.current更新的時機。

在上面模擬實現(xiàn)中，useEvent更新handlerRef.current的邏輯放在useLayoutEffect回調(diào)中進行。

這就保證了handlerRef.current始終在視圖完成渲染后再更新：

useLayoutEffect(() =&gt; {
  handlerRef.current = handler;
});

而事件回調(diào)觸發(fā)的時機顯然在視圖完成渲染之后，所以能夠穩(wěn)定獲取到最新的state與props。

注：源碼內(nèi)的實際更新時機會更早些，但不影響這里的結論

再來看看ahooks中的useMemoizedFn，fnRef.current的更新時機是useMemoizedFn執(zhí)行時（即組件render時）：

function useMemoizedFn<T extends noop>(fn: T) {
  const fnRef = useRef<T>(fn);
  // 更新fnRef.current
  fnRef.current = useMemo(() => fn, [fn]);
  // ...省略代碼
}

當React18啟用并發(fā)更新后，組件render的次數(shù)、時機并不確定。

所以useMemoizedFn中fnRef.current的更新時機也是不確定的。

這就增加了在并發(fā)更新下使用時潛在的風險。

可以說，useEvent通過限制handlerRef.current更新時機，進而限制應用場景，最終達到穩(wěn)定的目的。