React Hooks 性能优化的正确姿势

前言

React Hooks 出来很长一段时间了，相信有不少的朋友已经深度使用了。无论是 React 本身，还是其生态中，都在摸索着进步。鉴于我使用的 React 的经验，给大家分享一下。对于 React hooks，性能优化可以从以下几个方面着手考虑。

场景 1

在使用了 React Hooks 后，很多人都会抱怨渲染的次数变多了。没错，官方就是这么推荐的：

我们推荐把 state 切分成多个 state 变量，每个变量包含的不同值会在同时发生变化。

function Box() { const [position, setPosition] = useState({ left: 0, top: 0 }); const [size, setSize] = useState({ width: 100, height: 100 }); // ... } 

这种写法在同时 setPosition 和 setSize 的时候，相较于 class 写法会额外多出一次 render。这也就是渲染次数变多的根本原因。当然这种写法仍然值得推荐，可读性和可维护性更高，能更好的逻辑分离。

针对这种场景若出现十几或几十个 useState 的时候，可读性就会变差，这个时候就需要相关性的组件化了。以逻辑为导向，抽离在不同的文件中，借助 React.memo 来屏蔽其他 state 导致的 rerender。

const Position = React.memo(({ position }: PositionProps) => { // position 相关逻辑 return ( <div>{position.left}</div> ); }); 

因此在 React hooks 组件中尽量不要写流水线代码，保持在 200 行左右最佳，通过组件化降低耦合和复杂度，还能优化一定的性能。

场景 2

class 对比 hooks，上代码：

class Counter extends React.Component { state = { count: 0, }; increment = () => { this.setState((prev) => ({ count: prev.count + 1, })); }; render() { const { count } = this.state; return <ChildComponent count={count} OnClick={this.increment} />; } } 

function Counter() { const [count, setCount] = React.useState(0); function increment() { setCount((n) => n + 1); } return <ChildComponent count={count} OnClick={increment} />; } 

凭直观感受，你是否会觉得 hooks 等同于 class 的写法？错，hooks 的写法已经埋了一个坑。在 count 状态更新的时候， Counter 组件会重新执行，这个时候会重新创建一个新的函数 increment。这样传递给 ChildComponent 的 onClick 每次都是一个新的函数，从而导致 ChildComponent 组件的 React.memo 失效。

解决办法：

function usePersistFn<T extends (...args: any[]) => any>(fn: T) { const ref = React.useRef<Function>(() => { throw new Error('Cannot call function while rendering.'); }); ref.current = fn; return React.useCallback(ref.current as T, [ref]); } 

// 建议使用 `usePersistFn` const increment = usePersistFn(() => { setCount((n) => n + 1); }); // 或者使用 useCallback const increment = React.useCallback(() => { setCount((n) => n + 1); }, []); 

上面声明了 usePersistFn 自定义 hook，可以保证函数地址在本组件中永远不会变化。完美解决 useCallback 依赖值变化而重新生成新函数的问题，逻辑量大的组件强烈建议使用。

不仅仅是函数，比如每次 render 所创建的新对象，传递给子组件都会有此类问题。尽量不在组件的参数上传递因 render 而创建的对象，比如 style={{ width: 0 }} 此类的代码用 React.useMemo 来优化。

const CustomCompOnent= React.memo(({ width }: CustomComponentProps) => { const style = React.useMemo(() => ({ width } as React.CSSProperties), [width]); return <ChildComponent style={style} />; }); 

style 若不需改变，可以提取到组件外面声明。尽管这样做写法感觉太繁琐，但是不依赖 React.memo 重新实现的情况下，是优化性能的有效手段。

场景 3

对于复杂的场景，使用 useWhyDidYouUpdate hook 来调试当前的可变变量引起的 rerender。这个函数也可直接使用 ahooks 中的实现。

function useWhyDidYouUpdate(name, props) { const previousProps = useRef(); useEffect(() => { if (previousProps.current) { const allKeys = Object.keys({ ...previousProps.current, ...props }); const changesObj = {}; allKeys.forEach(key => { if (previousProps.current[key] !== props[key]) { changesObj[key] = { from: previousProps.current[key], to: props[key] }; } }); if (Object.keys(changesObj).length) { console.log('[why-did-you-update]', name, changesObj); } } previousProps.current = props; }); } const Counter = React.memo(props => { useWhyDidYouUpdate('Counter', props); return <div style={props.style}>{props.count}</div>; }); 

当 useWhyDidYouUpdate 中所监听的 props 发生了变化，则会打印对应的值对比，是调试中的神器，极力推荐。

场景 4

借助 Chrome Performance 代码进行调试，录制一段操作，在 Timings 选项卡中分析耗时最长逻辑在什么地方，会展现出组件的层级栈，然后精准优化。

场景 5

在 React 中是极力推荐函数式编程，可以让数据不可变性作为我们优化的手段。我在 React class 时代大量使用了 immutable.js 结合 redux 来搭建业务，与 React 中 PureComponnet 完美配合，性能保持非常好。但是在 React hooks 中再结合 typescript 它就显得有点格格不入了，类型支持得不是很完美。这里可以尝试一下 immer.js，引入成本小，写法也简洁了不少。

const nextState = produce(currentState, (draft) => { draft.p.x.push(2); }) // true currentState === nextState; 

场景 6

复杂场景使用 Map 对象代替数组操作，map.get(), map.has()，与数组查找相比尤其高效。

// Map const map = new Map([['a', { id: 'a' }], ['b', { id: 'b' }], ['c', { id: 'c' }]]); // 查找值 map.has('a'); // 获取值 map.get('a'); // 遍历 map.forEach(n => n); // 它可以很容易转换为数组 Array.from(map.values()); // 数组 const list = [{ id: 'a' }, { id: 'b' }, { id: 'c' }]; // 查找值 list.some(n => n.id === 'a'); // 获取值 list.find(n => n.id === 'a'); // 遍历 list.forEach(n => n); 

结语

React 性能调优，除了阻止 rerender，还有与写代码的方式有关系。最后，我要推一下近期写的 React 状态管理库 https://github.com/MinJieLiu/heo，也可以作为性能优化的一个手段，希望大家从 redux 的繁琐中解放出来，省下的时间用来享受生活