React.memo()、useCallback()、useMemo() 区别及基本使用

先来看个简单的例子

// Parent.jsx
import react, { useState } from 'react';
import Child from '../Child';

function Parent() {
  const [parentCount, setParentCount] = useState(0);
  console.log('父組件重新渲染--------------');
  return (
    <div style={{ background: 'lightseagreen' }}>
      <Child />
      <button type="button" onClick={() => { setParentCount(parentCount + 1); }}>父组件 +1</button>
    </div>
  );
}

export default Parent;

// Child.jsx
import React from 'react';

function Child() {
  console.log('------------子組件重新渲染');
  return (
    <div style={{ background: 'pink', margin: '50px 0' }}>
      <button type="button">子組件</button>
    </div>
  );
}

export default Child;

当我们点击父组件按钮时，父组件的状态parentCount会被更新，导致父组件重新渲染，子组件也会重新渲染；而此时我们的子组件和父组件之间并没有依赖关系，因此这种重复渲染是可以优化掉的，可以使用React.memo 包裹子组件

// Child.jsx
import React from 'react';
// ...other code
export default React.memo(Child);

React.memo(Comp[, fn])

用于减少子组件的重新渲染

React.memo是一个高阶组件(参数为组件，返回值为新组件的函数即为高阶组件)

对外部而言，React.memo会检查props的变更，仅当传入的props发生变化时组件才会重新渲染，这时我们再点击父组件按钮，子组件就不会重新渲染了

React.memo对复杂对象只会做浅层对比，可以通过传入第二个参数来控制对比过程

第二个参数为一个接收重新渲染前后props的函数

function MyComponent(props) {
  /* 使用 props 渲染 */
}
function areEqual(prevProps, nextProps) {
  /*
  如果把 nextProps 传入 render 方法的返回结果与
  将 prevProps 传入 render 方法的返回结果一致则返回 true，
  否则返回 false
  */
}
export default React.memo(MyComponent, areEqual);

useMemo(fn[, DependentArray])

用于减少每次组件重新渲染时重复进行复杂的计算，参数为一个函数和可选的依赖项数组，返回传入函数的调用结果

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

useMemo作用类似于vue 的 computed (计算属性)，不同之处在于需要手动传入依赖项，当依赖项变更时会重新调用传入的函数，返回计算值

传入依赖项为空数组时则直接返回上次的计算结果

不传入依赖项时，每次组件刷新都会重新计算，应该在代码能正常运行的情况下将其作为一种优化策略

修改下我们的例子，注意这里用React.memo 包裹了子组件，保证测试时子组件重新渲染只受传入的props变化的影响

// Parent.jsx
import React, { useState, useMemo } from 'react';
import Child from '../Child';

function Parent() {
  const [parentCount, setParentCount] = useState(0);
  const [otherCount, setOtherCount] = useState(0);
  console.log('父組件重新渲染--------------');

  // 一个复杂的计算
  const computedFn = (a, b) => {
    console.log('----重新执行了计算----');
    return a + b;
  };

  const computedValue = useMemo(() => {
    return computedFn(parentCount, 1);
  }, [parentCount]);
  
  return (
    <div style={{ background: 'lightseagreen' }}>
      <Child parentCount={parentCount} computedValue={computedValue} />
      <button type="button" onClick={() => { setParentCount(parentCount + 1); }}>父组件 +1</button>
      <button type="button" onClick={() => { setOtherCount(otherCount + 1); }}>父组件 otherCount+1</button>
    </div>
  );
}

点击第一个按钮，依赖项变更，输出重新执行了计算，点击第二个按钮，因为更改的不是计算值的依赖项，因此不会重新计算，子组件也不会重新渲染

useCallback(fn[, DependentArray])

用于需要传递给子组件的函数，减少子组件的重复渲染，参数为一个函数和可选的依赖项数组，返回出入函数的记忆版本

// Parent.jsx
import React, { useState } from 'react';
import Child from '../Child';

function Parent() {
  const [parentCount, setParentCount] = useState(0);
  const [otherCount, setOtherCount] = useState(0);
  console.log('父組件重新渲染--------------');

  const computedFn = () => {
    return parentCount + 1;
  };
  
  return (
    <div style={{ background: 'lightseagreen' }}>
      <Child parentCount={parentCount} computedFn={computedFn} />
      <button type="button" onClick={() => { setParentCount(parentCount + 1); }}>父组件 +1</button>
      <button type="button" onClick={() => { setOtherCount(otherCount + 1); }}>父组件 otherCount+1</button>
    </div>
  );
}

export default Parent;

// Child.jsx
import React from 'react';

function Child(props) {
  const { computedValue, computedFn } = props;
  console.log('------------子組件重新渲染');
  return (
    <div style={{ background: 'pink', margin: '50px 0' }}>
      <div>
        父组件传入的计算结果：
        {computedValue}
      </div>
      <button type="button" onClick={computedFn}>子組件</button>
    </div>
  );
}

export default React.memo(Child);

当点击第二个按钮时，子组件也会重新渲染

给computedFn 加上useCallBack

// Parent.jsx
import React, { useState, useCallback } from 'react';

// ...other code

  const computedFn = useCallback(() => {
    console.log(parentCount);
    return parentCount + 1;
  }, [parentCount]) ;

// ...other code

export default Parent;

这时再点击父组件第二个按钮子组件，子组件不会重新渲染，因为useCallback 的依赖项没变更，返回的是上一次渲染的函数，因此传入子组件的props没变，组件不会重新渲染

需要注意的是，被useCallback保存的函数内部作用域也不会变更，因此，当依赖项数组为空的时候，传入useCallback的函数的内部通过闭包取的组件内的变量值终不变

import React, { useState, useCallback } from 'react';
import Child from '../Child';

let a = 0;
function Parent() {
  const [parentCount, setParentCount] = useState(0);
  const [otherCount, setOtherCount] = useState(0);
  console.log('父組件重新渲染--------------');

  const computedFn = useCallback(() => {
    // 依赖项为空，这里的打印值始终不变；
    // 因为组件state变化时会重新渲染整个组件,而这里parentCount取的始终是第一次渲染版本的值
    console.log(parentCount); 
    // 这里的打印值会实时更新，因为变量直接定义在组件外部，不受组件重新渲染影响
    console.log(a);
    return parentCount + 1;
  }, []) ;

  return (
    <div style={{ background: 'lightseagreen' }}>
      <Child parentCount={parentCount} computedFn={computedFn} />
      <button type="button" onClick={() => { setParentCount(parentCount + 1); a += 1; }}>父组件 +1</button>
      <button type="button" onClick={() => { setOtherCount(otherCount + 1); }}>父组件 otherCount+1</button>
    </div>
  );
}

export default Parent;

因为useCallback目的是减少子组件重渲染，因此需要搭配子组件的shouldComponentUpdate或 React.memo 一起使用才有优化意义

以上是依赖项变更不频繁的情况，当依赖项变更频繁时，useCallback的记忆效果就不好，可以使用ref 作为依赖项解决

function Form() {
  const [text, updateText] = useState('');
  const textRef = useRef();

  useEffect(() => {
    textRef.current = text; // 把它写入 ref
  });

  const handleSubmit = useCallback(() => {
    // ref 对象在组件的整个生命周期内保持不变
    // 从 ref 读取它，current的变更不会引起组件的重新渲染，而函数内部又能拿到正确的值
    const currentText = textRef.current; 
    alert(currentText);
  }, [textRef]);

  return (
    <>
      <input value={text} onChange={e => updateText(e.target.value)} />
      <ExpensiveTree onSubmit={handleSubmit} />
    </>
  );
}

useRef

看看官方介绍

const refContainer = useRef(initialValue);

useRef 返回一个可变的 ref 对象，其 .current 属性被初始化为传入的参数（initialValue）。返回的 ref 对象在组件的整个生命周期内保持不变

可以理解为：用useRef创建的对象有个current 属性，这个属性就像个盒子，啥都能存，包括dom节点；返回的 ref 对象在组件的整个生命周期内保持不变，即存在current的值不受组件重新渲染影响，始终保持着一开始的引用；同时该属性的变更也不会触发组件的重新渲染；这个属性的初始值为useRef的参数

看看官方例子

function TextInputWithFocusButton() {
  const inputEl = useRef(null);
  const onButtonClick = () => {
    // `current` 指向已挂载到 DOM 上的文本输入元素
    inputEl.current.focus();
  };
  return (
    <>
      <input ref={inputEl} type="text" />
      <button onClick={onButtonClick}>Focus the input</button>
    </>
  );
}

当把useRef 创建的对象传给DOM 元素的ref属性时，react会把当前DOM元素的引用存入current属性，这样就可以通过ref对象直接操作DOM元素了。

链接: https://fly63.com/article/detial/10788