@@ -1,8 +1,12 @@
11#!/usr/bin/env node
223+import path from "node:path";
34import { resolveExtensionBatchPlan } from "./lib/extension-test-plan.mjs";
45import { isDirectScriptRun, runVitestBatch } from "./lib/vitest-batch-runner.mjs";
567+const FS_MODULE_CACHE_PATH_ENV_KEY = "OPENCLAW_VITEST_FS_MODULE_CACHE_PATH";
8+const PARALLEL_ENV_KEY = "OPENCLAW_EXTENSION_BATCH_PARALLEL";
9+610function printUsage() {
711console.error("Usage: pnpm test:extensions:batch <extension[,extension...]> [vitest args...]");
812console.error(
@@ -27,6 +31,114 @@ function parseExtensionIds(rawArgs) {
2731return { extensionIds, passthroughArgs: args };
2832}
293334+function parsePositiveInt(value) {
35+const parsed = Number.parseInt(value ?? "", 10);
36+return Number.isFinite(parsed) && parsed > 0 ? parsed : null;
37+}
38+39+export function resolveExtensionBatchParallelism(groupCount, env = process.env) {
40+const override = parsePositiveInt(env[PARALLEL_ENV_KEY]);
41+return Math.min(Math.max(1, override ?? 1), Math.max(1, groupCount));
42+}
43+44+function sanitizeCacheSegment(value) {
45+return (
46+value
47+.replace(/[^a-zA-Z0-9._-]+/gu, "-")
48+.replace(/^-+|-+$/gu, "")
49+.slice(0, 180) || "default"
50+);
51+}
52+53+function createGroupEnv({ baseEnv, group, groupIndex, useDedicatedCache }) {
54+if (!useDedicatedCache || baseEnv[FS_MODULE_CACHE_PATH_ENV_KEY]?.trim()) {
55+return baseEnv;
56+}
57+58+return {
59+ ...baseEnv,
60+[FS_MODULE_CACHE_PATH_ENV_KEY]: path.join(
61+process.cwd(),
62+"node_modules",
63+".experimental-vitest-cache",
64+"extension-batch",
65+sanitizeCacheSegment(`${groupIndex}-${group.config}`),
66+),
67+};
68+}
69+70+function orderPlanGroups(planGroups, parallelism) {
71+if (parallelism <= 1) {
72+return planGroups;
73+}
74+return [...planGroups].toSorted((left, right) => {
75+if (left.estimatedCost !== right.estimatedCost) {
76+return right.estimatedCost - left.estimatedCost;
77+}
78+if (left.testFileCount !== right.testFileCount) {
79+return right.testFileCount - left.testFileCount;
80+}
81+return left.config.localeCompare(right.config);
82+});
83+}
84+85+async function runPlanGroup(group, params) {
86+console.log(
87+`[test-extension-batch] ${group.config}: ${group.extensionIds.join(", ")} (${group.testFileCount} files)`,
88+);
89+return await params.runGroup({
90+args: params.vitestArgs,
91+config: group.config,
92+env: createGroupEnv({
93+baseEnv: params.env,
94+ group,
95+groupIndex: params.groupIndex,
96+useDedicatedCache: params.useDedicatedCache,
97+}),
98+targets: group.roots,
99+});
100+}
101+102+export async function runExtensionBatchPlan(batchPlan, params = {}) {
103+const env = params.env ?? process.env;
104+const vitestArgs = params.vitestArgs ?? [];
105+const runGroup = params.runGroup ?? runVitestBatch;
106+const parallelism = resolveExtensionBatchParallelism(batchPlan.planGroups.length, env);
107+const orderedGroups = orderPlanGroups(batchPlan.planGroups, parallelism);
108+const useDedicatedCache = parallelism > 1;
109+110+if (parallelism > 1) {
111+console.log(`[test-extension-batch] Running up to ${parallelism} config groups in parallel`);
112+}
113+114+let nextGroupIndex = 0;
115+let exitCode = 0;
116+async function worker() {
117+while (exitCode === 0) {
118+const groupIndex = nextGroupIndex;
119+nextGroupIndex += 1;
120+const group = orderedGroups[groupIndex];
121+if (!group) {
122+return;
123+}
124+const groupExitCode = await runPlanGroup(group, {
125+ env,
126+ groupIndex,
127+ runGroup,
128+ useDedicatedCache,
129+ vitestArgs,
130+});
131+if (groupExitCode !== 0) {
132+exitCode = groupExitCode;
133+return;
134+}
135+}
136+}
137+138+await Promise.all(Array.from({ length: parallelism }, () => worker()));
139+return exitCode;
140+}
141+30142async function run() {
31143const rawArgs = process.argv.slice(2);
32144if (rawArgs.includes("--help") || rawArgs.includes("-h")) {
@@ -51,19 +163,12 @@ async function run() {
51163`[test-extension-batch] Running ${batchPlan.testFileCount} test files across ${batchPlan.extensionCount} extensions`,
52164);
5316554-for (const group of batchPlan.planGroups) {
55-console.log(
56-`[test-extension-batch] ${group.config}: ${group.extensionIds.join(", ")} (${group.testFileCount} files)`,
57-);
58-const exitCode = await runVitestBatch({
59-args: vitestArgs,
60-config: group.config,
61-env: process.env,
62-targets: group.roots,
63-});
64-if (exitCode !== 0) {
65-process.exit(exitCode);
66-}
166+const exitCode = await runExtensionBatchPlan(batchPlan, {
167+env: process.env,
168+ vitestArgs,
169+});
170+if (exitCode !== 0) {
171+process.exit(exitCode);
67172}
68173}
69174此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。