fast immediates: initial implementation

Author: lubieowoce

packages/next/errors.json

@@ -917,5 +917,7 @@
   "916": "Unexpected stream chunk while in Before stage",
   "917": "getFlightStream should always receive a ReadableStream when using the edge runtime",
   "918": "nodeStreamFromReadableStream cannot be used in the edge runtime",
-  "919": "createNodeStreamFromChunks cannot be used in the edge runtime"
+  "919": "createNodeStreamFromChunks cannot be used in the edge runtime",
+  "920": "install() was not called",
+  "921": "The \"callback\" argument must be of type function. Received %s"
 }

packages/next/src/server/app-render/app-render-prerender-utils.ts

@@ -1,4 +1,5 @@
 import { InvariantError } from '../../shared/lib/invariant-error'
+import { runPendingImmediatesAfterCurrentTask } from './fast-set-immediate.external'
 
 /**
  * This is a utility function to make scheduling sequential tasks that run back to back easier.
@@ -16,6 +17,7 @@ export function prerenderAndAbortInSequentialTasks<R>(
     return new Promise((resolve, reject) => {
       let pendingResult: Promise<R>
       setTimeout(() => {
+        runPendingImmediatesAfterCurrentTask()
         try {
           pendingResult = prerender()
           pendingResult.catch(() => {})
@@ -48,6 +50,7 @@ export function prerenderAndAbortInSequentialTasksWithStages<R>(
     return new Promise((resolve, reject) => {
       let pendingResult: Promise<R>
       setTimeout(() => {
+        runPendingImmediatesAfterCurrentTask()
         try {
           pendingResult = prerender()
           pendingResult.catch(() => {})
@@ -56,6 +59,7 @@ export function prerenderAndAbortInSequentialTasksWithStages<R>(
         }
       }, 0)
       setTimeout(() => {
+        runPendingImmediatesAfterCurrentTask()
         advanceStage()
       }, 0)
       setTimeout(() => {

packages/next/src/server/app-render/app-render-render-utils.ts

@@ -1,4 +1,5 @@
 import { InvariantError } from '../../shared/lib/invariant-error'
+import { runPendingImmediatesAfterCurrentTask } from './fast-set-immediate.external'
 
 /**
  * This is a utility function to make scheduling sequential tasks that run back to back easier.
@@ -16,6 +17,7 @@ export function scheduleInSequentialTasks<R>(
     return new Promise((resolve, reject) => {
       let pendingResult: R | Promise<R>
       setTimeout(() => {
+        runPendingImmediatesAfterCurrentTask()
         try {
           pendingResult = render()
         } catch (err) {
@@ -48,6 +50,7 @@ export function pipelineInSequentialTasks<A, B, C>(
     return new Promise((resolve, reject) => {
       let oneResult: A
       setTimeout(() => {
+        runPendingImmediatesAfterCurrentTask()
         try {
           oneResult = one()
         } catch (err) {
@@ -60,6 +63,7 @@ export function pipelineInSequentialTasks<A, B, C>(
 
       let twoResult: B
       const twoId = setTimeout(() => {
+        runPendingImmediatesAfterCurrentTask()
         // if `one` threw, then this timeout would've been cleared,
         // so if we got here, we're guaranteed to have a value.
         try {