Set current fiber during before-mutation traversal (#13219)

facebook · Jul 17, 2018 · d6a0626 · d6a0626
1 parent fd410f4
commit d6a0626
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/packages/react-reconciler/src/ReactFiberScheduler.js b/packages/react-reconciler/src/ReactFiberScheduler.js
@@ -428,8 +428,11 @@ function commitAllHostEffects() {
 
 function commitBeforeMutationLifecycles() {
   while (nextEffect !== null) {
-    const effectTag = nextEffect.effectTag;
+    if (__DEV__) {
+      ReactCurrentFiber.setCurrentFiber(nextEffect);
+    }
 
+    const effectTag = nextEffect.effectTag;
     if (effectTag & Snapshot) {
       recordEffect();
       const current = nextEffect.alternate;
@@ -440,6 +443,10 @@ function commitBeforeMutationLifecycles() {
     // This will be done by commitAllLifeCycles()
     nextEffect = nextEffect.nextEffect;
   }
+
+  if (__DEV__) {
+    ReactCurrentFiber.resetCurrentFiber();
+  }
 }
 
 function commitAllLifeCycles(