useCallback 與 useMemo 的正確使用時機

2024年8月13日4 分鐘閱讀

useCallback的用途和使用時機

useCallback hook 最主要的作用在於幫助其他 React 效能優化的手段維持正常,例如:React.memouseMemouseEffect 等等,而非效能優化。

將函式傳入 useCallback hook ,並且傳入一個依賴陣列,useCallback hook 會回傳一個 memoized 函式,這個 memoized 函式只有在依賴陣列中的值有變化時才會重新計算,否則會回傳上一次的函式。 以下是一個簡單的範例說明:

使用 useCallback hook 前

1function Card ({ count , handleClick }:props){
2 return (
3 <div>
4 <div>{count}</div>
5 <Button onClick={handleClick}>Click me</Button>
6 </div>
7 );
8}
9export default memo(Card);
1function App() {
2 const [count, setCount] = useState(0);
3 const updateCount = () => {
4 setCount((prevCount) => prevCount + 1);
5 };
6 const handleClick = () => {
7 alert(count);
8 };
9
10 return (
11 <div>
12 <Card count={count} handleClick={handleClick} />
13 <Button onClick={updateCount}>Update count</Button>
14 </div>
15 );
16}
17export default App;

在這個例子中,handleClick 這個函式在每一次 render 的過程中都會重新產生一個新的函式,這樣會導致 Card 元件因為感知到 props 中的 handleClick 函式有變化而重新渲染。

在這裡看起來似乎沒什麼問題,但是問題在 Card 元件被包裹在 React.memo 中,React.memo是 hoc (higher-order component) ,會在元件 re-render 的時候檢查 props 是否有變化,如果有變化就會重新渲染,否則就會透過快取返回上一次的結果,以達到效能優化的目的。

但是在上方的例子中,可以看到 handleClick 函式在每次 render 時都會重新產生一個新的函式,每一次 re-render 傳入的 props 的 handleClick 都是不同的函式進而導致 Card 元件重新渲染,如此一來,就失去了 React.memo 原有的效能優化的目的。

使用 useCallback hook 後

1function Card ({ count , handleClick }:props){
2 return (
3 <div>
4 <div>{count}</div>
5 <Button onClick={handleClick}>Click me</Button>
6 </div>
7 );
8}
9export default memo(Card);
1function App() {
2 const [count, setCount] = useState(0);
3
4 const updateCount = () => {
5 setCount((prevCount) => prevCount + 1);
6 };
7 const handleClick = useCallback(() => {
8 alert(count);
9 }, [count]);
10
11 return (
12 <div>
13 <Card count={count} handleClick={handleClick} />
14 <Button onClick={updateCount}>Update count</Button>
15 </div>
16 );
17}
18export default App;

可以上方 handleClick 使用 useCallback hook 來包裹,並且傳入一個依賴陣列 [count],當 count 資料有變化時,handleClick才會重新產生一個新的函式,否則會回傳上一次的函式,Card 元件所包裹的react.memo就可以正常運作,達到效能優化的目的。

如此一來可以讓函式可以參與資料流的變化,進而幫助 React hook dependencies 判斷資料流的邏輯運作正常。

useMemo的用途和使用時機

useMemo 用法跟 useCallback 類似,但是 useMemo 是用來快取陣列、物件或者是通常來用處理複雜的計算,以節省效能。

1function App() {
2 const [count, setCount] = useState(0);
3 const isEven = useMemo(() => {
4 return count % 2 === 0;
5 }, [count]);
6 return (
7 <div>
8 <div>{count}</div>
9 <div>{isEven ? "Even" : "Odd"}</div>
10 <Button onClick={() => setCount((prev) => prev + 1)}>Increment</Button>
11 </div>
12 );
13}

在以上的例子中,我們實作一個 button 來增加 count 的值,並且使用 useMemo hook 來快取 isEven 變數,當 count 的值有變化時,isEven 變數才會重新計算,否則會快取回傳上一次的結果。

標籤:

2024iT鐵人賽ReactcomponentuseCallbackuseMemo