Function withCodSpeed

benchmark/with-codspeed.mjs:87–180

(bench)

 * @returns {Bench}
 */
export function withCodSpeed(bench) {
	const mode = getCodspeedRunnerMode();
	if (mode === "disabled" || mode === "walltime") return bench;

	// --- simulation mode ---

	const meta = getOrCreateMeta(bench);
	const rawAdd = bench.add.bind(bench);

	bench.add = (name, fn, opts) => {
		const callingFile = getCallingFile();
		const uri = `${callingFile}::${name}`;
		meta.set(name, { uri, fn, opts });
		return rawAdd(name, fn, opts);
	};

	const setup = () => {
		setupCore();
		console.log("[CodSpeed] running in simulation mode");
	};

	const teardown = () => {
		teardownCore();
		console.log(`[CodSpeed] Done running ${bench.tasks.length} benches.`);
		return bench.tasks;
	};

	/**
	 * @param {Fn} fn
	 * @param {boolean} isAsync
	 * @returns {Fn}
	 */
	const wrapFrame = (fn, isAsync) => {
		if (isAsync) {
			// eslint-disable-next-line camelcase
			return async function __codspeed_root_frame__() {
				await fn();
			};
		}
		// eslint-disable-next-line camelcase
		return function __codspeed_root_frame__() {
			fn();
		};
	};

	bench.run = async () => {
		setup();
		for (const task of bench.tasks) {
			const m = /** @type {TaskMeta} */ (meta.get(task.name));

			// Warm-up: run the body a few times to stabilise caches / JIT.
			for (let i = 0; i < bench.iterations - 1; i++) {
				await m.fn();
			}

			// Instrumented run.
			global.gc?.();
			InstrumentHooks.startBenchmark();