上一节我们了解到render阶段的工作可以分为“递”阶段和“归”阶段。其中“递”阶段会执行beginWork,“归”阶段会执行completeWork。这一节我们看看“递”阶段的beginWork方法究竟做了什么。

# 方法概览

可以从源码这里 (opens new window)看到beginWork的定义。

注意

本文主线仍然沿用beginWork的核心流程讲解。随着SuspenseOffscreenCacheActivityViewTransition等能力加入,当前 React 源码中的beginWork分支比早期版本更多,但主干逻辑仍然是“处理当前Fiber节点,生成或复用子Fiber节点”。

从上一节我们已经知道,beginWork的工作是传入当前Fiber节点,创建子Fiber节点,我们从传参来看看具体是如何做的。

# 从传参看方法执行

function beginWork(
  current: Fiber | null,
  workInProgress: Fiber,
  renderLanes: Lanes
): Fiber | null {
  // ...省略函数体
}

其中传参:

  • current:当前组件对应的Fiber节点在上一次更新时的Fiber节点,即workInProgress.alternate
  • workInProgress:当前组件对应的Fiber节点
  • renderLanes:优先级相关,在讲解Scheduler时再讲解

双缓存机制一节我们知道,除rootFiber以外, 组件mount时,由于是首次渲染,是不存在当前组件对应的Fiber节点在上一次更新时的Fiber节点,即mountcurrent === null

组件update时,由于之前已经mount过,所以current !== null

所以我们可以通过current === null ?来区分组件是处于mount还是update

基于此原因,beginWork的工作可以分为两部分:

  • update时:如果current存在,在满足一定条件时可以复用current节点,这样就能克隆current.child作为workInProgress.child,而不需要新建workInProgress.child

  • mount时:除fiberRootNode以外,current === null。会根据fiber.tag不同,创建不同类型的子Fiber节点

function beginWork(
  current: Fiber | null,
  workInProgress: Fiber,
  renderLanes: Lanes
): Fiber | null {
  // update时:如果current存在可能存在优化路径,可以复用current(即上一次更新的Fiber节点)
  if (current !== null) {
    // ...省略

    // 复用current
    return bailoutOnAlreadyFinishedWork(current, workInProgress, renderLanes);
  } else {
    didReceiveUpdate = false;
  }

  // mount时:根据tag不同,创建不同的子Fiber节点
  switch (workInProgress.tag) {
    case IndeterminateComponent:
    // ...省略
    case LazyComponent:
    // ...省略
    case FunctionComponent:
    // ...省略
    case ClassComponent:
    // ...省略
    case HostRoot:
    // ...省略
    case HostComponent:
    // ...省略
    case HostText:
    // ...省略
    // ...省略其他类型
  }
}

# update 时

我们可以看到,当propscontext、开发环境热更新下的type等没有变化,且当前Fiber节点没有本次renderLanes需要处理的updatecontext变化时,didReceiveUpdate === false。在不处于错误或Suspense捕获后的二次渲染路径时,就可能进入复用前一次更新的子Fiber的优化路径。

可以将其概括为:

  1. oldProps === newProps && workInProgress.type === current.type,即propsfiber.type不变
  2. 当前Fiber节点没有与本次renderLanes匹配的updatecontext变化,会在讲解Scheduler时介绍
  3. 没有处于DidCapture等需要重新进入渲染的特殊路径
if (current !== null) {
  const oldProps = current.memoizedProps;
  const newProps = workInProgress.pendingProps;

  if (
    oldProps !== newProps ||
    hasLegacyContextChanged() ||
    (__DEV__ ? workInProgress.type !== current.type : false)
  ) {
    didReceiveUpdate = true;
  } else if (
    !checkScheduledUpdateOrContext(current, renderLanes) &&
    (workInProgress.flags & DidCapture) === NoFlags
  ) {
    didReceiveUpdate = false;
    switch (
      workInProgress.tag
      // 省略处理
    ) {
    }
    return bailoutOnAlreadyFinishedWork(current, workInProgress, renderLanes);
  } else {
    didReceiveUpdate = false;
  }
} else {
  didReceiveUpdate = false;
}

# mount 时

当不满足优化路径时,我们就进入第二部分,新建子Fiber

我们可以看到,根据fiber.tag不同,进入不同类型Fiber的创建逻辑。

可以从这里 (opens new window)看到tag对应的组件类型

// mount时:根据tag不同,创建不同的Fiber节点
switch (workInProgress.tag) {
  case IndeterminateComponent:
  // ...省略
  case LazyComponent:
  // ...省略
  case FunctionComponent:
  // ...省略
  case ClassComponent:
  // ...省略
  case HostRoot:
  // ...省略
  case HostComponent:
  // ...省略
  case HostText:
  // ...省略
  // ...省略其他类型
}

对于我们常见的组件类型,如(FunctionComponent/ClassComponent/HostComponent),最终会进入reconcileChildren (opens new window)方法。

# reconcileChildren

从该函数名就能看出这是Reconciler模块的核心部分。那么他究竟做了什么呢?

  • 对于mount的组件,他会创建新的子Fiber节点

  • 对于update的组件,他会将当前组件与该组件在上次更新时对应的Fiber节点比较(也就是俗称的Diff算法),将比较的结果生成新Fiber节点

export function reconcileChildren(
  current: Fiber | null,
  workInProgress: Fiber,
  nextChildren: any,
  renderLanes: Lanes
) {
  if (current === null) {
    // 对于mount的组件
    workInProgress.child = mountChildFibers(
      workInProgress,
      null,
      nextChildren,
      renderLanes
    );
  } else {
    // 对于update的组件
    workInProgress.child = reconcileChildFibers(
      workInProgress,
      current.child,
      nextChildren,
      renderLanes
    );
  }
}

从代码可以看出,和beginWork一样,他也是通过current === null ?区分mountupdate

不论走哪个逻辑,最终他会生成新的子Fiber节点并赋值给workInProgress.child,作为本次beginWork返回值,并作为下次performUnitOfWork执行时workInProgress的传参。

注意

值得一提的是,mountChildFibersreconcileChildFibers这两个方法的逻辑基本一致。核心区别是:reconcileChildFibers会追踪用于commit阶段的副作用,必要时为生成的Fiber节点设置flags或记录deletionsmountChildFibers通常不追踪插入、删除、移动等副作用。

# flags

我们知道,render阶段的工作是在内存中进行,当工作结束后会通知Renderer需要执行的DOM操作。要执行DOM操作的具体类型就保存在fiber.flags中。

你可以从这里 (opens new window)看到flags对应的DOM操作

比如:

// DOM需要插入到页面中
export const Placement = /* */ 0b0000000000000000000000000000010;
// DOM需要更新
export const Update = /*    */ 0b0000000000000000000000000000100;
// DOM需要删除
export const ChildDeletion = /* */ 0b0000000000000000000000000010000;

通过二进制表示flags,可以方便的使用位操作为fiber.flags赋值多个flag

那么,如果要通知RendererFiber节点对应的DOM节点插入页面中,需要满足两个条件:

  1. fiber.stateNode存在,即Fiber节点中保存了对应的DOM节点

  2. (fiber.flags & Placement) !== 0,即Fiber节点存在Placement flag

我们知道,mount时,fiber.stateNode === null,且在reconcileChildren中调用的mountChildFibers通常不会为Fiber节点追踪插入相关的flags。那么首屏渲染如何完成呢?

针对第一个问题,fiber.stateNode会在completeWork中创建,我们会在下一节介绍。

第二个问题的答案十分巧妙:假设mountChildFibers也会为整棵新建子树追踪插入相关的flags,那么可以预见mount时整棵Fiber树所有节点都会有Placement flag。那么commit阶段在执行DOM操作时每个节点都会执行一次插入操作,这样大量的DOM操作是极低效的。

为了解决这个问题,初次渲染时,HostRoot由于存在alternate,会通过reconcileChildFibers创建顶层子Fiber,并为需要插入的顶层子树标记Placement。顶层子树内部的后续mount路径通常通过mountChildFibers创建,不会为每个后代都标记Placement。因此commit阶段可以围绕顶层插入点完成整棵DOM树的插入,避免逐个节点重复插入。

根 Fiber 节点 Demo

借用上一节的 Demo,第一个进入beginWork方法的Fiber节点就是rootFiber,他的alternate指向current rootFiber(即他存在current)。

为什么rootFiber节点存在current(即rootFiber.alternate),我们在双缓存机制一节 mount 时的第二步已经讲过

由于存在currentrootFiberreconcileChildren时会走reconcileChildFibers逻辑,所以它创建出的顶层子Fiber会被标记Placement

而之后通过beginWork创建的Fiber节点是不存在current的(即 fiber.alternate === null),会走mountChildFibers逻辑,不会为整棵新建子树的每个后代都追踪插入副作用。

关注公众号 魔术师卡颂,后台回复531获得在线 Demo 地址

# 参考资料

beginWork流程图

beginWork流程图