React官方團隊出手,補齊原生Hook短板
大家好,我卡頌。
我們知道,Hooks使用時存在所謂的「閉包陷阱」,考慮如下代碼:
function Chat() {
const [text, setText] = useState('');
const onClick = useCallback(() => {
sendMessage(text);
}, []);
return <SendButton onClick={onClick} />;
}
我們期望點擊后sendMessage能傳遞text的最新值。
然而實際上,由于回調函數被useCallback緩存,形成閉包,所以點擊的效果始終是sendMessage('')。
這就是「閉包陷阱」。
以上代碼的一種解決方式是「為useCallback增加依賴項」:
const onClick = useCallback(() => {
sendMessage(text);
}, [text]);
但是這么做了后,每當依賴項(text)變化,useCallback會返回一個全新的onClick引用,這就失去了useCallback「緩存函數引用」的作用。
「閉包陷阱」的出現,加大了Hooks的上手門檻,也讓開發者更容易寫出有bug的代碼。
現在,React官方團隊要出手解決這個問題。
useEvent
解決方式是引入一個新的原生Hook —— useEvent。
他用于定義一個函數,這個函數有2個特性:
- 在組件多次render時保持引用一致。
- 函數內始終能獲取到最新的props與state。
上面的例子使用useEvent改造后:
function Chat() {
const [text, setText] = useState('');
const onClick = useEvent(() => {
sendMessage(text);
});
return <SendButton onClick={onClick} />;
}
在Chat組件多次render時,onClick始終指向同一個引用。
并且onClick觸發時始終能獲取到text的最新值。
之所以叫useEvent,是因為React團隊認為這個Hook的主要應用場景是:「封裝事件處理函數」。
useEvent的實現
useEvent的實現并不困難,代碼類似如下:
function useEvent(handler) {
const handlerRef = useRef(null);
// 視圖渲染完成后更新`handlerRef.current`指向
useLayoutEffect(() => {
handlerRef.current = handler;
});
// 用useCallback包裹,使得render時返回的函數引用一致
return useCallback((...args) => {
const fn = handlerRef.current;
return fn(...args);
}, []);
}
整體包括兩部分:
返回一個沒有依賴項的useCallback,使得每次render時函數的引用一致。
useCallback((...args) => {
const fn = handlerRef.current;
return fn(...args);
}, []);
在合適的時機更新handlerRef.current,使得實際執行的函數始終是最新的引用。
與開源Hooks的差異
很多開源Hooks庫已經實現類似功能(比如ahooks中的useMemoizedFn)。
useEvent與這些開源實現的差異主要體現在:
useEvent定位于「處理事件回調函數」這一單一場景,而useMemoizedFn定位于「緩存各種函數」。
那么問題來了,既然功能類似,那useEvent為什么要限制自己的使用場景呢?
答案是:為了更穩定。
useEvent能否獲取到最新的state與props取決于handlerRef.current更新的時機。
在上面模擬實現中,useEvent更新handlerRef.current的邏輯放在useLayoutEffect回調中進行。
這就保證了handlerRef.current始終在「視圖完成渲染」后再更新:
useLayoutEffect(() => {
handlerRef.current = handler;
});
而「事件回調」觸發的時機顯然在「視圖完成渲染」之后,所以能夠穩定獲取到最新的state與props。
注:源碼內的實際更新時機會更早些,但不影響這里的結論。
再來看看ahooks中的useMemoizedFn,fnRef.current的更新時機是「useMemoizedFn執行時」(即「組件render時」):
function useMemoizedFn<T extends noop>(fn: T) {
const fnRef = useRef<T>(fn);
// 更新fnRef.current
fnRef.current = useMemo(() => fn, [fn]);
// ...省略代碼
}
當React18啟用「并發更新」后,組件render的次數、時機并不確定。
所以useMemoizedFn中fnRef.current的更新時機也是不確定的。
這就增加了在「并發更新」下使用時潛在的風險。
可以說,useEvent通過限制handlerRef.current更新時機,進而限制應用場景,最終達到穩定的目的。
總結
useEvent當前還處于RFC(Request For Comments)[1]階段。
很多熱心的開發者對這個Hook的命名提出了建議,比如:useStableCallback:
又比如:useLatestClosure:
從這些命名看,他們顯然擴大了useEvent的應用場景。
經過本文的分析我們知道,「擴大應用場景」意味著「增加開發者使用時出錯的風險」。
參考資料
[1]RFC(Request For Comments):
https://github.com/reactjs/rfcs/blob/useevent/text/0000-useevent.md。