content-benchmark.mjs — astro Source File

Source Code

import { fileURLToPath } from 'node:url';
import { parseArgs } from 'node:util';
import colors from 'picocolors';
import { loadFixture } from '../test/test-utils.js';
import { generatePosts } from './scripts/generate-posts.mjs';

// Skip nonessential remark / rehype plugins for a fair comparison.
// This includes heading ID generation, syntax highlighting, GFM, and Smartypants.
process.env.ASTRO_PERFORMANCE_BENCHMARK = true;

const extByFixture = {
	md: '.md',
	mdx: '.mdx',
	mdoc: '.mdoc',
};

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,
			ext: extByFixture[fixture],
			template: templates[fixture],
		});
		console.info(`[${fixture}] Generated posts`);

		const { build } = await loadFixture({
			root,
		});
		const now = performance.now();
		console.info(`[${fixture}] Building...`);
		await build();
		console.info(
			colors.cyan(`[${fixture}] Built in ${colors.bold(getTimeStat(now, performance.now()))}.`),
		);
	}
}

// Test the build performance for content collections across multiple file types (md, mdx, mdoc)
(async function benchmarkAll() {
	try {
		const { values: flags } = parseArgs({ strict: false });
		const test = Array.isArray(flags.test)
			? flags.test
			: typeof flags.test === 'string'
				? [flags.test]
				: ['simple', 'with-astro-components', 'with-react-components'];

		const formats = Array.isArray(flags.format)
			? flags.format
			: typeof flags.format === 'string'
				? [flags.format]
				: ['md', 'mdx', 'mdoc'];

		const numPosts = flags.numPosts || 1000;

		if (test.includes('simple')) {
			const fixtures = formats;
			console.info(
// ... (63 more lines)