如何通过 useMemo 和 useCallback 提升你的 React 应用性能

背景

在 React 中，useMemo 和 useCallback 这两个 hook 是我们优化应用性能的有力工具。它们会返回 memoized 版本的值或函数，只在依赖项发生变化时才进行重新计算或定义。

Hook 介绍

useMemo

useMemo 的作用是返回一个 memoized 值，它接受两个参数：一个函数和一个依赖数组。只有当依赖项中的一个值变化，才会重新计算并返回新的 memoized 值。

const memoizedValue = useMemo(
  () => computeExpensiveValue(a, b),
  [a, b],
);

上述代码例子中，useMemo 返回了一个 memoized 版本的 computeExpensiveValue(a, b) 函数结果，a 或 b 改变时，才会重新计算。

举个例子

假设 computeExpensiveValue 是一个需要大量计算资源的函数，比如它需要计算斐波那契数列的第 n 项。在这种情况下，每次组件渲染时重新计算显然是一种资源浪费。通过使用 useMemo，我们可以在 a 或者 b 变化时，才重新进行这种耗时的计算。

useCallback

useCallback 与 useMemo 用法类似，不过它返回的是一个 memoized 函数。这在我们需要将函数作为 prop 传递给子组件时非常有用，避免由于父组件重渲染导致不必要的函数重新创建。

const memoizedCallback = useCallback(
  () => {
    doSomething(a, b);
  },
  [a, b],
);

在此代码例子中，useCallback 返回的是一个 memoized 版本的函数 () => doSomething(a, b)，只有当 a 或 b 发生变化时，才会重新创建新的函数定义。

举个例子

假设我们在一个父组件中定义了一个事件处理器，这个处理器依赖于父组件的一些 prop。如果我们每次在渲染父组件时都创建新的事件处理器，那么即使 prop 没有变化，子组件还是会进行不必要的重渲染。

为了解决这个问题，我们可以使用 useCallback 来 memoize 事件处理器，只有当依赖的 prop 变化时，才重新创建事件处理器。

建议

虽然 useMemo 和 useCallback 都可用于提升性能，但我们需要谨慎选择使用场景。过度使用可能导致更多的计算开销。例如在计算新的函数或值的过程本身不需要很多计算资源，但由于频繁的依赖项检查和新值的计算，可能会导致性能反而下降。

总结

useMemo 和 useCallback 是 React 提供的强有力的性能优化工具。掌握正确的使用场景和方式，可以有效地提升你的应用性能，而避免不必要的计算开销。