ci: shard benchmarks (#15314)

ematipico · web-flow · commit 0a9777eabc5a · 2026-01-28T15:06:57.000Z
diff --git a/.github/workflows/benchmark.yml b/.github/workflows/benchmark.yml
diff --git a/.github/workflows/continuous_benchmark.yml b/.github/workflows/continuous_benchmark.yml
@@ -26,6 +26,9 @@ env:
 jobs:
   codspeed:
     runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        shard: [ "1/4","2/4", "3/4", "4/4" ]
     permissions:
       contents: read
       pull-requests: write
@@ -58,5 +61,5 @@ jobs:
         timeout-minutes: 30
         with:
           working-directory: ./benchmark
-          run: pnpm bench
+          run: pnpm bench --shard=${{ matrix.shard }}
           token: ${{ secrets.CODSPEED_TOKEN }}
diff --git a/benchmark/bench/build-hybrid.bench.js b/benchmark/bench/build-hybrid.bench.js
@@ -0,0 +1,18 @@
+import { fileURLToPath } from 'node:url';
+import { build } from 'astro';
+import { bench, describe } from 'vitest';
+
+const hybridRoot = fileURLToPath(new URL('../static-projects/build-hybrid', import.meta.url));
+
+describe('Bench build time', () => {
+	bench(
+		'Build: hybrid site (static + server)',
+		async () => {
+			await build({
+				root: hybridRoot,
+				logLevel: 'error',
+			});
+		},
+		{ timeout: 300000, iterations: 3 },
+	);
+});
diff --git a/benchmark/bench/build-server.bench.js b/benchmark/bench/build-server.bench.js
@@ -0,0 +1,18 @@
+import { fileURLToPath } from 'node:url';
+import { build } from 'astro';
+import { bench, describe } from 'vitest';
+
+const serverRoot = fileURLToPath(new URL('../static-projects/build-server', import.meta.url));
+
+describe('Bench build time', () => {
+	bench(
+		'Build: full server site',
+		async () => {
+			await build({
+				root: serverRoot,
+				logLevel: 'error',
+			});
+		},
+		{ timeout: 300000, iterations: 3 },
+	);
+});
diff --git a/benchmark/bench/build-static.bench.js b/benchmark/bench/build-static.bench.js
@@ -0,0 +1,18 @@
+import { fileURLToPath } from 'node:url';
+import { build } from 'astro';
+import { bench, describe } from 'vitest';
+
+const staticRoot = fileURLToPath(new URL('../static-projects/build-static', import.meta.url));
+
+describe('Bench build time', () => {
+	bench(
+		'Build: full static site',
+		async () => {
+			await build({
+				root: staticRoot,
+				logLevel: 'error',
+			});
+		},
+		{ timeout: 300000, iterations: 3 },
+	);
+});
diff --git a/benchmark/bench/render.bench.js b/benchmark/bench/render.bench.js
@@ -1,11 +1,7 @@
 import { existsSync } from 'node:fs';
 import { fileURLToPath } from 'node:url';
-import { build } from 'astro';
 import { beforeAll, bench, describe } from 'vitest';
 
-const staticRoot = fileURLToPath(new URL('../static-projects/build-static', import.meta.url));
-const hybridRoot = fileURLToPath(new URL('../static-projects/build-hybrid', import.meta.url));
-const serverRoot = fileURLToPath(new URL('../static-projects/build-server', import.meta.url));
 const renderRoot = new URL('../projects/render-bench/', import.meta.url);
 
 let streamingApp;
@@ -51,38 +47,3 @@ describe('Bench rendering', () => {
 		await nonStreamingApp.render(request);
 	});
 });
-
-describe('Bench build time', () => {
-	bench(
-		'Build: full static site',
-		async () => {
-			await build({
-				root: staticRoot,
-				logLevel: 'error',
-			});
-		},
-		{ timeout: 300000, iterations: 3 },
-	);
-
-	bench(
-		'Build: hybrid site (static + server)',
-		async () => {
-			await build({
-				root: hybridRoot,
-				logLevel: 'error',
-			});
-		},
-		{ timeout: 300000, iterations: 3 },
-	);
-
-	bench(
-		'Build: full server site',
-		async () => {
-			await build({
-				root: serverRoot,
-				logLevel: 'error',
-			});
-		},
-		{ timeout: 300000, iterations: 3 },
-	);
-});
diff --git a/benchmark/vitest.config.js b/benchmark/vitest.config.js
@@ -3,5 +3,5 @@ import { defineConfig } from 'vitest/config';
 
 export default defineConfig({
 	plugins: process.env.CODSPEED ? [codspeedPlugin()] : [],
-	include: ['./bench/codspeed.bench.js'],
+	include: ['./bench/*.bench.js'],
 });
