fix: re-apply workaround for injectToStream() at stream end race cond…

…ition (#40) This reverts commit 19aec51.
brillout · Jul 1, 2024 · f43bd9b · f43bd9b
1 parent e079cf9
commit f43bd9b
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/server/renderToStream/createReadableWrapper.ts b/src/server/renderToStream/createReadableWrapper.ts
@@ -52,9 +52,12 @@ function createReadableWrapper(readableFromReact: ReadableStream, { stopTimeout
 
     stopTimeout?.()
 
-    onBeforeEnd()
-    controllerOfUserStream.close()
-    onEnded()
+    // Collect `injectToStream()` calls stuck in an async call
+    setTimeout(() => {
+      onBeforeEnd()
+      controllerOfUserStream.close()
+      onEnded()
+    }, 0)
   }
 }