Clear stack state when lazy node resolved [perf]

overlookmotel · Mar 17, 2019 · 5640b04 · 5640b04
1 parent 218a763
commit 5640b04
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lib/renderer.js b/lib/renderer.js
@@ -120,6 +120,7 @@ class PartialRenderer extends ReactDOMServerRenderer {
 
 			if (type === TYPE_PROMISE && !node.resolved) {
 				node.resolved = true;
+				node.stackState = null;
 				abort(node.promise);
 			}
 
@@ -404,6 +405,9 @@ class PartialRenderer extends ReactDOMServerRenderer {
 		this.suspenseNode = node.parentSuspense;
 		this.restoreStack(node.stackState, element);
 
+		// Clear stack state on node
+		node.stackState = null;
+
 		// Render element
 		this.cycle();
 
@@ -441,6 +445,7 @@ class PartialRenderer extends ReactDOMServerRenderer {
 
 		// Promise is no longer awaited
 		node.resolved = true;
+		node.stackState = null;
 		this.numAwaiting--;
 
 		// Abort all rendering
@@ -499,6 +504,7 @@ class PartialRenderer extends ReactDOMServerRenderer {
 			}
 		} else if (type === TYPE_PROMISE && !node.resolved) {
 			node.resolved = true;
+			node.stackState = null;
 			this.numAwaiting--;
 			abort(node.promise);
 			inLazy = true;