一文掌握React 组件树遍历技巧

引言本文对应的 react 版本是 18 2 0 下面的 dom 结构react 内部是如何遍历的 const App = () => {return (

引言

本文对应的 react 版本是 18.2.0

下面的 dom 结构react 内部是如何遍历的

const App = () => {
  return (
    <div>
      <button>+1</button>
      <A count={0} />
    </div>
  );
};
const A = (props) => {
  useEffect(() => {
    console.log(props.count);
  }, [props.count]);
  return <div>{props.count}</div>;
};

react 内部遍历核心逻辑：

在 render 时调用 commitPassiveUnmountOnFiber 函数
commitPassiveUnmountOnFiber 处理不同的 WorkTag，并调用 recursivelyTraversePassiveUnmountEffects
recursivelyTraversePassiveUnmountEffects 根据当前 Fiber 的子节点有没有 passive effect（useEffect，useLayoutEffect）来决定是否遍历当前 Fiber 的子节点
- 如果子节点有 passive effect，则优先遍历子节点 (深度优先)，直到找到最终的叶子节点，退出当前循环
- 然后进入兄弟节点，开始遍历兄弟节点的子节点
  - 具体从哪个兄弟节点开始遍历，react 选择的是离退出循环的那个叶子节点的父节点，检查有没有子节点，以此循环遍历
- 直到最后找到所有有 passive effect 的节点

代码简化：

commitPassiveUnmountOnFiber(root.current);
function commitPassiveUnmountOnFiber(finishedWork) {
  // 省略了处理不同的 WorkTag
  recursivelyTraversePassiveUnmountEffects(finishedWork);
}
function recursivelyTraversePassiveUnmountEffects(parentFiber) {
  // 省略了其他处理
  if (parentFiber.subtreeFlags & PassiveMask) {
    let child = parentFiber.child;
    while (child !== null) {
      commitPassiveUnmountOnFiber(child);
      child = child.sibling;
    }
  }
}

所以对于这段 dom 的遍历逻辑是：

首先从根组件开始 FiberRootNode，取到 current
- 也就是说 FiberRootNode.current 是 div#root 这是一个 fiber，它的 tag 是 3
由于 App 的子组件有 passive effect，所以会进入 App 组件，它的 tag 是 0
App 组件中节点是 <div>，<di > 的 tag 是 5
- <div> 下面有两个子元素 <button>、<A>
先遍历 <button> 它的 tag 是 5
<button> 内部只有一个文本节点，没有 passive effect
- 所以 react 不遍历了（跳出当前遍历的循环，也就是 button 这条不在遍历了）
跳出循环后，查看 button 的兄弟节点，它的兄弟节点是 <A>，<A> 的 tag 是 0
由于 <A> 节点的子节点没有 passive effect，所以跳出循环，结束整个遍历