perf: use queue for pages processing (#238)

package.json

@@ -91,6 +91,7 @@
     "html5parser": "^2.0.2",
     "jsdom": "^19.0.0",
     "kolorist": "^1.5.1",
+    "p-queue": "6.6.2",
     "prettier": "^2.6.0",
     "yargs": "^17.3.1"
   },

src/node/build.ts

@@ -1,6 +1,7 @@
 /* eslint-disable no-console */
 import { dirname, isAbsolute, join, parse } from 'path'
 import { createRequire } from 'module'
+import PQueue from 'p-queue'
 import { blue, cyan, dim, gray, green, red, yellow } from 'kolorist'
 import fs from 'fs-extra'
 import type { InlineConfig, ResolvedConfig } from 'vite'
@@ -52,6 +53,7 @@ export async function build(cliOptions: Partial<ViteSSGOptions> = {}, viteConfig
     dirStyle = 'flat',
     includeAllRoutes = false,
     format = 'esm',
+    concurrency = 20,
     rootContainerId = 'app',
   }: ViteSSGOptions = Object.assign({}, config.ssgOptions || {}, cliOptions)
 
@@ -135,8 +137,10 @@ export async function build(cliOptions: Partial<ViteSSGOptions> = {}, viteConfig
     ? await import('vue/server-renderer')
     : _require('vue/server-renderer')
 
-  await Promise.all(
-    routesPaths.map(async(route) => {
+  const queue = new PQueue({ concurrency })
+
+  for (const route of routesPaths) {
+    queue.add(async() => {
       try {
         const appCtx = await createApp(false, route) as ViteSSGContext<true>
         const { app, router, head, initialState, triggerOnSSRAppRendered, transformState = serializeState } = appCtx
@@ -192,8 +196,10 @@ export async function build(cliOptions: Partial<ViteSSGOptions> = {}, viteConfig
       catch (err: any) {
         throw new Error(`${gray('[vite-ssg]')} ${red(`Error on page: ${cyan(route)}`)}\n${err.stack}`)
       }
-    }),
-  )
+    })
+  }
+
+  await queue.start().onIdle()
 
   await fs.remove(ssgOut)
 

src/types.ts

@@ -103,6 +103,13 @@ export interface ViteSSGOptions {
    * @default `app`
    */
   rootContainerId?: string
+
+  /**
+   * Size of generation processing queue.
+   *
+   * @default 20
+   */
+  concurrency?: number
 }
 
 type PartialKeys<T, Keys extends keyof T> = Omit<T, Keys> & Partial<Pick<T, Keys>>