only track react client module loading in dynamicIO prerenders

lubieowoce · lubieowoce · commit 7c082cbc2f9c · 2025-05-16T21:22:42.000+02:00
diff --git a/packages/next/src/server/app-render/app-render.tsx b/packages/next/src/server/app-render/app-render.tsx
@@ -1203,18 +1203,37 @@ async function renderToHTMLOrFlightImpl(
     // to treat chunk/module loading with similar semantics as cache reads to avoid
     // module loading from causing a prerender to abort too early.
 
+    const shouldTrackModuleLoading = () => {
+      if (process.env.NEXT_RUNTIME === 'edge') {
+        return false
+      } else {
+        if (!renderOpts.experimental.dynamicIO) {
+          return false
+        }
+        const workUnitStore = workUnitAsyncStorage.getStore()
+        return !!(
+          workUnitStore &&
+          (workUnitStore.type === 'prerender' || workUnitStore.type === 'cache')
+        )
+      }
+    }
+
     const __next_require__: typeof instrumented.require = (...args) => {
       const exportsOrPromise = instrumented.require(...args)
-      // requiring an async module returns a promise.
-      trackPendingImport(exportsOrPromise)
+      if (shouldTrackModuleLoading()) {
+        // requiring an async module returns a promise.
+        trackPendingImport(exportsOrPromise)
+      }
       return exportsOrPromise
     }
     // @ts-expect-error
     globalThis.__next_require__ = __next_require__
 
     const __next_chunk_load__: typeof instrumented.loadChunk = (...args) => {
       const loadingChunk = instrumented.loadChunk(...args)
-      trackPendingChunkLoad(loadingChunk)
+      if (shouldTrackModuleLoading()) {
+        trackPendingChunkLoad(loadingChunk)
+      }
       return loadingChunk
     }
     // @ts-expect-error
