一文掌握React?組件樹遍歷技巧

更新時(shí)間：2023年04月17日 09:56:07 作者：uccs

這篇文章主要為大家介紹了React?組件樹遍歷技巧的掌握，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

引言

本文對應(yīng)的 react 版本是 18.2.0

下面的 dom 結(jié)構(gòu)react 內(nèi)部是如何遍歷的

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 內(nèi)部遍歷核心邏輯：

在 render 時(shí)調(diào)用 commitPassiveUnmountOnFiber 函數(shù)
commitPassiveUnmountOnFiber 處理不同的 WorkTag，并調(diào)用 recursivelyTraversePassiveUnmountEffects
recursivelyTraversePassiveUnmountEffects 根據(jù)當(dāng)前 Fiber 的子節(jié)點(diǎn)有沒有 passive effect（useEffect，useLayoutEffect）來決定是否遍歷當(dāng)前 Fiber 的子節(jié)點(diǎn)
- 如果子節(jié)點(diǎn)有 passive effect，則優(yōu)先遍歷子節(jié)點(diǎn) (深度優(yōu)先)，直到找到最終的葉子節(jié)點(diǎn)，退出當(dāng)前循環(huán)
- 然后進(jìn)入兄弟節(jié)點(diǎn)，開始遍歷兄弟節(jié)點(diǎn)的子節(jié)點(diǎn)
  - 具體從哪個(gè)兄弟節(jié)點(diǎn)開始遍歷，react 選擇的是離退出循環(huán)的那個(gè)葉子節(jié)點(diǎn)的父節(jié)點(diǎn)，檢查有沒有子節(jié)點(diǎn)，以此循環(huán)遍歷
- 直到最后找到所有有 passive effect 的節(jié)點(diǎn)

代碼簡化：

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 這是一個(gè) fiber，它的 tag 是 3
由于 App 的子組件有 passive effect，所以會(huì)進(jìn)入 App 組件，它的 tag 是 0
App 組件中節(jié)點(diǎn)是 <div>，<di > 的 tag 是 5
- <div> 下面有兩個(gè)子元素 <button>、<A>
先遍歷 <button> 它的 tag 是 5
<button> 內(nèi)部只有一個(gè)文本節(jié)點(diǎn)，沒有 passive effect
- 所以 react 不遍歷了（跳出當(dāng)前遍歷的循環(huán)，也就是 button 這條不在遍歷了）
跳出循環(huán)后，查看 button 的兄弟節(jié)點(diǎn)，它的兄弟節(jié)點(diǎn)是 <A>，<A> 的 tag 是 0
由于 <A> 節(jié)點(diǎn)的子節(jié)點(diǎn)沒有 passive effect，所以跳出循環(huán)，結(jié)束整個(gè)遍歷