fix: prevent 100% CPU usage in large-scale test suites (#606)

wellwelwel · Jul 26, 2024 · 382631b · 382631b
1 parent 2c002c0
commit 382631b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/src/polyfills/cpus.ts b/src/polyfills/cpus.ts
@@ -0,0 +1,9 @@
+import {
+ availableParallelism as nodeAvailableParallelism,
+ cpus,
+} from 'node:os';
+
+export const availableParallelism = (): number =>
+ typeof nodeAvailableParallelism === 'function'
+ ? Math.floor(nodeAvailableParallelism() / 2)
+ : (cpus()?.length ?? 1);
diff --git a/src/services/run-tests.ts b/src/services/run-tests.ts
@@ -14,6 +14,7 @@ import { format } from './format.js';
 import { runTestFile } from './run-test-file.js';
 import { isQuiet } from '../parsers/output.js';
 import { results } from '../configs/poku.js';
+import { availableParallelism } from '../polyfills/cpus.js';
 
 const cwd = processCWD();
 
@@ -101,7 +102,7 @@ export const runTestsParallel = async (
  ? [sanitizePath(dir)]
  : await listFiles(testDir, configs);
  const filesByConcurrency: string[][] = [];
- const concurrencyLimit = configs?.concurrency || 0;
+ const concurrencyLimit = configs?.concurrency ?? availableParallelism();
  const concurrencyResults: (boolean | undefined)[][] = [];
  const showLogs = !isQuiet(configs);