Update only parents that may be inconsistent

Author: okmttdhr

packages/react-reconciler/src/ReactFiberNewContext.new.js

@@ -142,6 +142,7 @@ export function popProvider(
 export function scheduleWorkOnParentPath(
   parent: Fiber | null,
   renderLanes: Lanes,
+  stopAt?: Fiber | null = null,
 ) {
   // Update the child lanes of all the ancestors, including the alternates.
   let node = parent;
@@ -160,6 +161,11 @@ export function scheduleWorkOnParentPath(
     } else {
       // Neither alternate was updated, which means the rest of the
       // ancestor path already has sufficient priority.
+      if (stopAt === null) {
+        break;
+      }
+    }
+    if (stopAt && node === stopAt) {
       break;
     }
     node = node.return;
@@ -285,16 +291,8 @@ function propagateContextChange_eager<T>(
       // because we want to schedule this fiber as having work
       // on its children. We'll use the childLanes on
       // this fiber to indicate that a context has changed.
-      scheduleWorkOnParentPath(parentSuspense, renderLanes);
+      scheduleWorkOnParentPath(parentSuspense, renderLanes, parentSuspense);
       nextFiber = fiber.sibling;
-    } else if (
-      fiber.tag === SuspenseComponent &&
-      workInProgress.tag === ContextProvider
-    ) {
-      // We don't know if it will have any context consumers in it.
-      // Schedule this fiber as having work on its children.
-      scheduleWorkOnParentPath(fiber.child, renderLanes);
-      nextFiber = fiber.child;
     } else {
       // Traverse down.
       nextFiber = fiber.child;
@@ -381,7 +379,11 @@ function propagateContextChanges<T>(
               // on its children. We'll use the childLanes on
               // this fiber to indicate that a context has changed.
               const primaryChildFragment = workInProgress.child;
-              scheduleWorkOnParentPath(primaryChildFragment, renderLanes);
+              scheduleWorkOnParentPath(
+                primaryChildFragment,
+                renderLanes,
+                workInProgress,
+              );
             } else {
               scheduleWorkOnParentPath(consumer.return, renderLanes);
             }
@@ -425,7 +427,7 @@ function propagateContextChanges<T>(
       // because we want to schedule this fiber as having work
       // on its children. We'll use the childLanes on
       // this fiber to indicate that a context has changed.
-      scheduleWorkOnParentPath(parentSuspense, renderLanes);
+      scheduleWorkOnParentPath(parentSuspense, renderLanes, parentSuspense);
       nextFiber = null;
     } else {
       // Traverse down.

packages/react-reconciler/src/ReactFiberNewContext.old.js

@@ -142,6 +142,7 @@ export function popProvider(
 export function scheduleWorkOnParentPath(
   parent: Fiber | null,
   renderLanes: Lanes,
+  stopAt?: Fiber | null = null,
 ) {
   // Update the child lanes of all the ancestors, including the alternates.
   let node = parent;
@@ -160,6 +161,11 @@ export function scheduleWorkOnParentPath(
     } else {
       // Neither alternate was updated, which means the rest of the
       // ancestor path already has sufficient priority.
+      if (stopAt === null) {
+        break;
+      }
+    }
+    if (stopAt && node === stopAt) {
       break;
     }
     node = node.return;
@@ -285,16 +291,8 @@ function propagateContextChange_eager<T>(
       // because we want to schedule this fiber as having work
       // on its children. We'll use the childLanes on
       // this fiber to indicate that a context has changed.
-      scheduleWorkOnParentPath(parentSuspense, renderLanes);
+      scheduleWorkOnParentPath(parentSuspense, renderLanes, parentSuspense);
       nextFiber = fiber.sibling;
-    } else if (
-      fiber.tag === SuspenseComponent &&
-      workInProgress.tag === ContextProvider
-    ) {
-      // We don't know if it will have any context consumers in it.
-      // Schedule this fiber as having work on its children.
-      scheduleWorkOnParentPath(fiber.child, renderLanes);
-      nextFiber = fiber.child;
     } else {
       // Traverse down.
       nextFiber = fiber.child;
@@ -381,7 +379,11 @@ function propagateContextChanges<T>(
               // on its children. We'll use the childLanes on
               // this fiber to indicate that a context has changed.
               const primaryChildFragment = workInProgress.child;
-              scheduleWorkOnParentPath(primaryChildFragment, renderLanes);
+              scheduleWorkOnParentPath(
+                primaryChildFragment,
+                renderLanes,
+                workInProgress,
+              );
             } else {
               scheduleWorkOnParentPath(consumer.return, renderLanes);
             }
@@ -425,7 +427,7 @@ function propagateContextChanges<T>(
       // because we want to schedule this fiber as having work
       // on its children. We'll use the childLanes on
       // this fiber to indicate that a context has changed.
-      scheduleWorkOnParentPath(parentSuspense, renderLanes);
+      scheduleWorkOnParentPath(parentSuspense, renderLanes, parentSuspense);
       nextFiber = null;
     } else {
       // Traverse down.