chore: add numPosts as a benchmark flag

withastro · Mar 7, 2023 · ad20691 · ad20691
1 parent 06a0e2f
commit ad20691
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/packages/astro/performance/content-benchmark.mjs b/packages/astro/performance/content-benchmark.mjs
@@ -16,14 +16,12 @@ const extByFixture = {
 	mdoc: '.mdoc',
 };
 
-const NUM_POSTS = 1000;
-
-async function benchmark({ fixtures, templates }) {
+async function benchmark({ fixtures, templates, numPosts }) {
 	for (const fixture of fixtures) {
 		const root = new URL(`./fixtures/${fixture}/`, import.meta.url);
 		await generatePosts({
 			postsDir: fileURLToPath(new URL('./src/content/generated/', root)),
-			numPosts: NUM_POSTS,
+			numPosts,
 			ext: extByFixture[fixture],
 			template: templates[fixture],
 		});
@@ -55,10 +53,12 @@ async function benchmark({ fixtures, templates }) {
 			? [flags.format]
 			: ['md', 'mdx', 'mdoc'];
 
+		const numPosts = flags.numPosts || 1000;
+
 		if (test.includes('simple')) {
 			const fixtures = formats;
 			console.log(
-				`\n${bold('Simple')} ${dim(`${NUM_POSTS} posts (${formatsToString(fixtures)})`)}`
+				`\n${bold('Simple')} ${dim(`${numPosts} posts (${formatsToString(fixtures)})`)}`
 			);
 			process.env.ASTRO_PERFORMANCE_TEST_NAME = 'simple';
 			await benchmark({
@@ -68,14 +68,15 @@ async function benchmark({ fixtures, templates }) {
 					mdx: 'simple.md',
 					mdoc: 'simple.md',
 				},
+				numPosts,
 			});
 		}
 
 		if (test.includes('with-astro-components')) {
 			const fixtures = formats.filter((format) => format !== 'md');
 			console.log(
 				`\n${bold('With Astro components')} ${dim(
-					`${NUM_POSTS} posts (${formatsToString(fixtures)})`
+					`${numPosts} posts (${formatsToString(fixtures)})`
 				)}`
 			);
 			process.env.ASTRO_PERFORMANCE_TEST_NAME = 'with-astro-components';
@@ -85,14 +86,15 @@ async function benchmark({ fixtures, templates }) {
 					mdx: 'with-astro-components.mdx',
 					mdoc: 'with-astro-components.mdoc',
 				},
+				numPosts,
 			});
 		}
 
 		if (test.includes('with-react-components')) {
 			const fixtures = formats.filter((format) => format !== 'md');
 			console.log(
 				`\n${bold('With React components')} ${dim(
-					`${NUM_POSTS} posts (${formatsToString(fixtures)})`
+					`${numPosts} posts (${formatsToString(fixtures)})`
 				)}`
 			);
 			process.env.ASTRO_PERFORMANCE_TEST_NAME = 'with-react-components';
@@ -102,6 +104,7 @@ async function benchmark({ fixtures, templates }) {
 					mdx: 'with-react-components.mdx',
 					mdoc: 'with-react-components.mdoc',
 				},
+				numPosts,
 			});
 		}
 	} finally {