add experimental option for shared worker pool

vercel · Aug 10, 2021 · 614bf80 · 614bf80
1 parent cba88fd
commit 614bf80
Show file tree

Hide file tree

Showing 3 changed files with 182 additions and 161 deletions.
diff --git a/packages/next/build/index.ts b/packages/next/build/index.ts
@@ -91,8 +91,6 @@ import { normalizeLocalePath } from '../shared/lib/i18n/normalize-locale-path'
 import { isWebpack5 } from 'next/dist/compiled/webpack/webpack'
 import { NextConfigComplete } from '../server/config-shared'
 
-const staticWorker = require.resolve('./worker')
-
 export type SsgRoute = {
   initialRevalidateSeconds: number | false
   srcRoute: string | null
@@ -671,6 +669,10 @@ export default async function build(
     ) as BuildManifest
 
     const timeout = config.experimental.staticPageGenerationTimeout || 0
+    const sharedPool = config.experimental.sharedPool || false
+    const staticWorker = sharedPool
+      ? require.resolve('./worker')
+      : require.resolve('./utils')
     let infoPrinted = false
     const staticWorkers = new Worker(staticWorker, {
       timeout: timeout * 1000,
@@ -705,12 +707,14 @@ export default async function build(
       },
       numWorkers: config.experimental.cpus,
       enableWorkerThreads: config.experimental.workerThreads,
-      exposedMethods: [
-        'hasCustomGetInitialProps',
-        'isPageStatic',
-        'getNamedExports',
-        'exportPage',
-      ],
+      exposedMethods: sharedPool
+        ? [
+            'hasCustomGetInitialProps',
+            'isPageStatic',
+            'getNamedExports',
+            'exportPage',
+          ]
+        : ['hasCustomGetInitialProps', 'isPageStatic', 'getNamedExports'],
     }) as Worker &
       Pick<
         typeof import('./worker'),
@@ -1107,10 +1111,14 @@ export default async function build(
           pages: combinedPages,
           outdir: path.join(distDir, 'export'),
           statusMessage: 'Generating static pages',
-          exportPageWorker: staticWorkers.exportPage.bind(staticWorkers),
-          endWorker: async () => {
-            await staticWorkers.end()
-          },
+          exportPageWorker: sharedPool
+            ? staticWorkers.exportPage.bind(staticWorkers)
+            : undefined,
+          endWorker: sharedPool
+            ? async () => {
+                await staticWorkers.end()
+              }
+            : undefined,
         }
         const exportConfig: any = {
           ...config,

diff --git a/packages/next/server/config-shared.ts b/packages/next/server/config-shared.ts
@@ -88,6 +88,7 @@ export type NextConfig = { [key: string]: any } & {
     swcMinify?: boolean
     swcLoader?: boolean
     cpus?: number
+    sharedPool?: boolean
     plugins?: boolean
     profiling?: boolean
     isrFlushToDisk?: boolean
@@ -165,6 +166,7 @@ export const defaultConfig: NextConfig = {
       (Number(process.env.CIRCLE_NODE_TOTAL) ||
         (os.cpus() || { length: 1 }).length) - 1
     ),
+    sharedPool: false,
     plugins: false,
     profiling: false,
     isrFlushToDisk: true,