奇怪的useMemo知识增加了

作为「性能优化」手段，一般用useMemo缓存函数组件中比较消耗性能的计算结果：

function App() {
  const memoizedValue = useMemo(
    () => computeExpensiveValue(a, b),
    [a, b]
  );
  // ...
}

只有在依赖项改变后才会重新计算新的memoizedValue。

你有没有想过，如果用useMemo缓存函数组件的返回值，会怎么样呢？

举个例子

我们有个全局context —— AppContext。

由于同学们偷懒，随着项目的迭代，新增的context都选择放在AppContext里，导致AppContext包含的内容越来越多。

现在我们有个Tree组件，他会渲染一个很耗性能的大组件ExpensiveTree。

function Tree() {
  let appContextValue = useContext(AppContext);
  let theme = appContextValue.theme;

  return <ExpensiveTree className={theme} />;
}

该组件内部依赖AppContext中的theme状态。

由于AppContext中包含很多与theme无关的state，导致每次其他无关的state更新，Tree都会重新render，进而ExpensiveTree组件也重新render。

现在这个优化任务交到了你手上，该怎么办呢？

优化ExpensiveTree

这时候，useMemo就能派上用场：

function Tree() {
  let appContextValue = useContext(AppContext);
  let theme = appContextValue.theme;

  return useMemo(() => {
    return <ExpensiveTree className={theme} />;
  }, [theme])
}

我们将返回的ExpensiveTree作为useMemo返回值，theme作为依赖。

这样，即使AppContext改变导致Tree反复render，ExpensiveTree也只会在theme改变后render。

原理解析

要理解这么做有效的原因，需要了解三点：

1. useMemo返回值是什么

2. 函数组件的返回值是什么

3. React组件在什么时候render

回答第一个问题：useMemo会将第一个参数（函数）的返回值保存在组件对应fiber中，只有在依赖项（第二个参数）变化后才会重新调用第一个参数（函数）计算一个新值。

回答第二个问题：函数组件的返回值是JSX对象。

同一个函数组件调用多次，返回的是多个「不同」的JSX对象（即使props未变，但JSX是新的引用）。

按照以上两个回答，我们可以得出结论：

以上useMemo用法实际上在函数组件对应的fiber中缓存了一个完整的JSX对象

第三个问题，函数组件需要同时满足如下条件才不会render：

1. oldProps === newProps

前后两次更新props全等，注意是「全等」。

2. 组件context没有变化

3. workInProgress.type === current.type

组件更新前后fiber.type未变化，比如div没有变为p。

4. !includesSomeLane(renderLanes, updateLanes)

当前fiber上不存在更新，或者存在更新但优先级低。

更详细的解释，可以参考这篇文章：React组件到底什么时候render？

当我们不使用useMemo包裹返回值，每次Tree render返回的都是全新的JSX对象。

所以对于ExpensiveTree，oldProps !== newProps。

再看2：ExpensiveTree内部context没变，满足

再看3：ExpensiveTree更新前后type都是ExpensiveTree，满足

再看4: ExpensiveTree内没有状态更新，满足

所以，当我们使用useMemo包裹ExpensiveTree后，当theme不变，每次Tree render后返回的都是同一个JSX对象，满足第一条。

基于这个原因，ExpensiveTree不会render。

总结

这篇文章提到的useMemo用法，并未在官网文档中体现，而是在#15156^[1]中由Dan介绍。

相比Vue，React更灵活，开发过程中需要开发者注意更多细节。要完全了解React，可能需要学习一些源码层面的知识。

参考资料