CLI: scope task maintenance to gateway runtime

mbelinky · mbelinky · commit 7c4554204eac · 2026-03-29T12:36:02.000+02:00
diff --git a/src/cli/run-main.exit.test.ts b/src/cli/run-main.exit.test.ts
@@ -8,6 +8,8 @@ const ensurePathMock = vi.hoisted(() => vi.fn());
 const assertRuntimeMock = vi.hoisted(() => vi.fn());
 const closeActiveMemorySearchManagersMock = vi.hoisted(() => vi.fn(async () => {}));
 const hasMemoryRuntimeMock = vi.hoisted(() => vi.fn(() => false));
+const ensureTaskRegistryReadyMock = vi.hoisted(() => vi.fn());
+const startTaskRegistryMaintenanceMock = vi.hoisted(() => vi.fn());
 const outputRootHelpMock = vi.hoisted(() => vi.fn());
 const buildProgramMock = vi.hoisted(() => vi.fn());
 const maybeRunCliInContainerMock = vi.hoisted(() =>
@@ -49,6 +51,14 @@ vi.mock("../plugins/memory-state.js", () => ({
   hasMemoryRuntime: hasMemoryRuntimeMock,
 }));
 
+vi.mock("../tasks/task-registry.js", () => ({
+  ensureTaskRegistryReady: ensureTaskRegistryReadyMock,
+}));
+
+vi.mock("../tasks/task-registry.maintenance.js", () => ({
+  startTaskRegistryMaintenance: startTaskRegistryMaintenanceMock,
+}));
+
 vi.mock("./program/root-help.js", () => ({
   outputRootHelp: outputRootHelpMock,
 }));
@@ -76,6 +86,8 @@ describe("runCli exit behavior", () => {
     expect(maybeRunCliInContainerMock).toHaveBeenCalledWith(["node", "openclaw", "status"]);
     expect(tryRouteCliMock).toHaveBeenCalledWith(["node", "openclaw", "status"]);
     expect(closeActiveMemorySearchManagersMock).not.toHaveBeenCalled();
+    expect(ensureTaskRegistryReadyMock).not.toHaveBeenCalled();
+    expect(startTaskRegistryMaintenanceMock).not.toHaveBeenCalled();
     expect(exitSpy).not.toHaveBeenCalled();
     exitSpy.mockRestore();
   });
diff --git a/src/cli/run-main.ts b/src/cli/run-main.ts
@@ -133,11 +133,6 @@ export async function runCli(argv: string[] = process.argv) {
   // Enforce the minimum supported runtime before doing any work.
   assertSupportedRuntime();
 
-  const { ensureTaskRegistryReady } = await import("../tasks/task-registry.js");
-  ensureTaskRegistryReady();
-  const { startTaskRegistryMaintenance } = await import("../tasks/task-registry.maintenance.js");
-  startTaskRegistryMaintenance();
-
   try {
     if (shouldUseRootHelpFastPath(normalizedArgv)) {
       const { outputRootHelp } = await import("./program/root-help.js");
diff --git a/src/gateway/server.impl.ts b/src/gateway/server.impl.ts
@@ -72,6 +72,7 @@ import {
 } from "../secrets/runtime.js";
 import { onSessionLifecycleEvent } from "../sessions/session-lifecycle-events.js";
 import { onSessionTranscriptUpdate } from "../sessions/transcript-events.js";
+import { startTaskRegistryMaintenance } from "../tasks/task-registry.maintenance.js";
 import { runSetupWizard } from "../wizard/setup.js";
 import { createAuthRateLimiter, type AuthRateLimiter } from "./auth-rate-limit.js";
 import { startChannelHealthMonitor } from "./channel-health-monitor.js";
@@ -876,6 +877,7 @@ export async function startGatewayServer(
         });
 
     if (!minimalTestGateway) {
+      startTaskRegistryMaintenance();
       ({ tickInterval, healthInterval, dedupeCleanup, mediaCleanup } =
         startGatewayMaintenanceTimers({
           broadcast,
