React16源码: React中的updateMemoComponent的源码实现

updateMemoComponent

1 ）概述

• 在react16.6之后，提供了一个新的API
• 通过 React.memo 来创建一个具有类似于 pure component 特性的 function component
• 现在主要关注其更新过程

2 ）源码

定位到 packages/react-reconciler/src/ReactFiberBeginWork.js#L237
进入 updateMemoComponent

function updateMemoComponent(
  current: Fiber | null,
  workInProgress: Fiber,
  Component: any,
  nextProps: any,
  updateExpirationTime,
  renderExpirationTime: ExpirationTime,
): null | Fiber {
  // 进来之后, 也是要先判断current是否等于null
  // 因为对于current是否等于null, 在update的一个过程当中非常重要
  // 它是用来辨别我们这个组件是第一次渲染还是非第一次渲染的情况
  // 这里是 第一次渲染的情况
  if (current === null) {
    // 获取 type
    // 是在创建 memo 的时候，它 return 的是一个 `{ $$typeof: REACT_MEMO_TYPE, type }`
    // 然后它的type是传进来的 function component
    // 在这里获取的 type 是一个function component
    let type = Component.type;
    // 注意，这里，compare 是 调用 memo 时，传递的 第二个参数
    // 用于对比新老props是否有变化的一个函数, 类似于 SCU
    // 在我们每一次组件更新的过程当中，都会调用那个方法，判断这个组件是否需要更新
    if (isSimpleFunctionComponent(type) && Component.compare === null) {
      // If this is a plain function component without default props,
      // and with only the default shallow comparison, we upgrade it
      // to a SimpleMemoComponent to allow fast path updates.
      workInProgress.tag = SimpleMemoComponent;
      workInProgress.type = type;
      // 符合上述 if 条件，会按照 下面这个函数 去更新组件
      return updateSimpleMemoComponent(
        current,
        workInProgress,
        type,
        nextProps,
        updateExpirationTime,
        renderExpirationTime,
      );
    }
    // 不符合上述条件，第一次渲染的情况，调用 createFiberFromTypeAndProps 获取 child
    // 不需要调用 reconcileChildren 来减少运行时开销，为什么是这样呢？
    // 因为它的children是非常明确的，就是 function component
    // 所以它传入这个type就是function component
    // nextprops就是放到 memo component上面的 props
    // 因为这个props实际是要传给具体的 function component
    // 以及传入 mode 和 renderExpirationTime
    let child = createFiberFromTypeAndProps(
      Component.type,
      null,
      nextProps,
      null,
      workInProgress.mode,
      renderExpirationTime,
    );
    // 并挂载
    child.ref = workInProgress.ref;
    child.return = workInProgress;
    workInProgress.child = child;
    return child;
  }
  // 对于不是第一次渲染
  let currentChild = ((current.child: any): Fiber); // This is always exactly one child
  // 如果组件需要更新
  if (updateExpirationTime < renderExpirationTime) {
    // This will be the props with resolved defaultProps,
    // unlike current.memoizedProps which will be the unresolved ones.
    const prevProps = currentChild.memoizedProps;
    // Default to shallow comparison
    let compare = Component.compare;
    compare = compare !== null ? compare : shallowEqual;
    // 对比新老 props 和 ref 是否有变化，需要更新，但是没有变化，则跳过更新
    if (compare(prevProps, nextProps) && current.ref === workInProgress.ref) {
      // 符合条件，跳过组件更新
      return bailoutOnAlreadyFinishedWork(
        current,
        workInProgress,
        renderExpirationTime,
      );
    }
  }
  // React DevTools reads this flag.
  workInProgress.effectTag |= PerformedWork;
  // 正常更新
  // 
  let newChild = createWorkInProgress(
    currentChild,
    nextProps,
    renderExpirationTime,
  );
  newChild.ref = workInProgress.ref;
  newChild.return = workInProgress;
  workInProgress.child = newChild;
  return newChild;
}

• 进入 isSimpleFunctionComponent

  // 它首先判断这个type是否是一个function，以及是否没有 construct 方法，以及它没有 defaultProps 这个属性
  // 也就是说它没有这些类似于 class component 的一些属性
  // 它就是一个纯 function component
  export function isSimpleFunctionComponent(type: any) {
    return (
      typeof type === 'function' &&
      !shouldConstruct(type) &&
      type.defaultProps === undefined
    );
  }
  

• 进入 updateSimpleMemoComponent

  // 在这里传进来的 Component 已经不是我们的 memo component了
  // 而是这个Component里面的type
  function updateSimpleMemoComponent(
    current: Fiber | null,
    workInProgress: Fiber,
    Component: any,
    nextProps: any,
    updateExpirationTime,
    renderExpirationTime: ExpirationTime,
  ): null | Fiber {
    // 首先是一个常规的判断 current是否等于null 以及它的 updateExpirationTime 相关的一些判断
    if (current !== null && updateExpirationTime < renderExpirationTime) {
      // 拿到 prevProps，就是之前的props
      const prevProps = current.memoizedProps;
      // nextProps 就是现在 fiber 对象上的 pendingProps
      // 判断两个 props 是否相等，以及两个 ref 是否相等
      if (
        shallowEqual(prevProps, nextProps) &&
        current.ref === workInProgress.ref
      ) {
        // 跳过组件的更新
        return bailoutOnAlreadyFinishedWork(
          current,
          workInProgress,
          renderExpirationTime,
        );
      }
    }
    // 需要更新
    // 我们增加了memo这个特性，只是说我们可以通过一些简单的判断，跳过组件的更新
    // 对于纯 function component，每次都需要去执行这个 function component 里面的逻辑的
    // 可能会造成一些性能上面的开销的问题
    return updateFunctionComponent(
      current,
      workInProgress,
      Component,
      nextProps,
      renderExpirationTime,
    );
  }
  

• 我们通过React.memo这个方法创建的组件，它的所有props是直接作用于传入的 function component的

• 为什么在这里它不使用 reconcileChildFibers 方法呢？

• 因为 reconcileChildFibers，它会把props作为我自己当前这个组件

  • 也就是我们的 memo component，它的children去进行一个调和的一个过程
  • 最终得到的是我们在使用 memo component 包裹的那一部分的 props.child里面的东西
  • 但是事实上, 在memo组件里面，它的child是来自于 component.type
  • 也就是我们传入的那个 function component，还是它的child

• 简单总结

  • 我们的 memo component 它拿到的props是作为 Component.type
  • 也就是我们传入的 function component，它的props来进行一个创建他的 childfiber 的一个内容
  • 所以 memo component 的更新方式跟其他的组件会有一定的不同，因为它本身没有任何的一个意义
  • 本身提供的功能只是用来对比新老props是否有变化
  • 所以他接收的其他的props，都是应该作为它的真正的组件也就是function component 的props来进行使用
  • 所以这里面，它通过两个不同的方式，在两个不同的阶段去创建他child的一个过程