bopodev-api 0.1.8 → 0.1.9

package/src/worker/scheduler.ts

@@ -1,80 +1,23 @@
-import { sql } from "drizzle-orm";
 import type { BopoDb } from "bopodev-db";
 import type { RealtimeHub } from "../realtime/hub";
 import { runHeartbeatSweep } from "../services/heartbeat-service";
 
-interface HeartbeatSchedulerOptions {
-  companyIds?: string[];
-}
-
-export function createHeartbeatScheduler(db: BopoDb, realtimeHub?: RealtimeHub, options?: HeartbeatSchedulerOptions) {
+export function createHeartbeatScheduler(db: BopoDb, companyId: string, realtimeHub?: RealtimeHub) {
   const intervalMs = Number(process.env.BOPO_HEARTBEAT_SWEEP_MS ?? 60_000);
-  const warnedMissingConfiguredCompanyIds = new Set<string>();
   let running = false;
   const timer = setInterval(() => {
     if (running) {
       return;
     }
     running = true;
-    void sweepScheduledCompanies(db, realtimeHub, options, warnedMissingConfiguredCompanyIds)
+    void runHeartbeatSweep(db, companyId, { realtimeHub })
+      .catch((error) => {
+        // eslint-disable-next-line no-console
+        console.error("[scheduler] heartbeat sweep failed", error);
+      })
       .finally(() => {
         running = false;
       });
   }, intervalMs);
   return () => clearInterval(timer);
 }
-
-async function sweepScheduledCompanies(
-  db: BopoDb,
-  realtimeHub?: RealtimeHub,
-  options?: HeartbeatSchedulerOptions,
-  warnedMissingConfiguredCompanyIds?: Set<string>
-) {
-  const companyIds = options?.companyIds?.length
-    ? await resolveConfiguredCompanyIds(db, options.companyIds, warnedMissingConfiguredCompanyIds ?? new Set<string>())
-    : await listCompanyIds(db);
-  for (const companyId of companyIds) {
-    try {
-      await runHeartbeatSweep(db, companyId, { realtimeHub });
-    } catch (error) {
-      // eslint-disable-next-line no-console
-      console.error(`[scheduler] heartbeat sweep failed for company '${companyId}': ${toErrorMessage(error)}`);
-    }
-  }
-}
-
-async function listCompanyIds(db: BopoDb) {
-  const result = await db.execute(sql`
-    SELECT id
-    FROM companies
-    ORDER BY created_at ASC
-  `);
-  return (result.rows ?? [])
-    .map((row) => row.id)
-    .filter((id): id is string => typeof id === "string" && id.length > 0);
-}
-
-async function resolveConfiguredCompanyIds(
-  db: BopoDb,
-  configuredCompanyIds: string[],
-  warnedMissingConfiguredCompanyIds: Set<string>
-) {
-  const existingCompanyIds = new Set(await listCompanyIds(db));
-  const activeCompanyIds: string[] = [];
-  for (const companyId of configuredCompanyIds) {
-    if (existingCompanyIds.has(companyId)) {
-      activeCompanyIds.push(companyId);
-      continue;
-    }
-    if (!warnedMissingConfiguredCompanyIds.has(companyId)) {
-      warnedMissingConfiguredCompanyIds.add(companyId);
-      // eslint-disable-next-line no-console
-      console.warn(`[scheduler] skipping unknown company id '${companyId}' from BOPO_SCHEDULER_COMPANY_IDS`);
-    }
-  }
-  return activeCompanyIds;
-}
-
-function toErrorMessage(error: unknown) {
-  return error instanceof Error ? error.message : String(error);
-}