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 (
    

      
       { setParentCount(parentCount + 1); }}>父组件 +1
    
  );
}

export default Parent;

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

function Child() {
  console.log('------------子組件重新渲染');
  return (
    

      子組件
    
  );
}

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 (
    

      
       { setParentCount(parentCount + 1); }}>父组件 +1
       { setOtherCount(otherCount + 1); }}>父组件 otherCount+1
    
  );
}

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

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 (
    

      
       { setParentCount(parentCount + 1); }}>父组件 +1
       { setOtherCount(otherCount + 1); }}>父组件 otherCount+1
    
  );
}

export default Parent;

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

function Child(props) {
  const { computedValue, computedFn } = props;
  console.log('------------子組件重新渲染');
  return (
    

      

        父组件传入的计算结果：
        {computedValue}
      
      子組件
    
  );
}

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 (
      
  
        
         { setParentCount(parentCount + 1); a += 1; }}>父组件 +1
         { setOtherCount(otherCount + 1); }}>父组件 otherCount+1
      
    );
  }
  
  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 (
    <>
       updateText(e.target.value)} />
      
    
  );
}

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 (
    <>
      
      Focus the input
    
  );
}

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