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

发布时间:2024年01月21日

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的一个过程
文章来源:https://blog.csdn.net/Tyro_java/article/details/135724190
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。