Add some features to the admin panel

2025-06-07 14:06:11 +09:00
parent 9b18963041
commit bc2be914df
5 changed files with 512 additions and 49 deletions
--- a/services/batch-scheduler.ts
+++ b/services/batch-scheduler.ts
@ -0,0 +1,132 @@
+import { batchProcess } from "../scripts/fetch_and_generate.js";
+
+interface BatchSchedulerState {
+  enabled: boolean;
+  lastRun?: string;
+  nextRun?: string;
+  isRunning: boolean;
+  intervalId?: NodeJS.Timeout;
+}
+
+class BatchScheduler {
+  private state: BatchSchedulerState = {
+    enabled: true,
+    isRunning: false,
+  };
+
+  private readonly SIX_HOURS_MS = 6 * 60 * 60 * 1000; // 6 hours in milliseconds
+
+  constructor() {
+    // Start with initial delay and then schedule regular runs
+    this.scheduleInitialRun();
+  }
+
+  private scheduleInitialRun() {
+    setTimeout(async () => {
+      if (this.state.enabled) {
+        await this.runBatch();
+      }
+      if (this.state.enabled) {
+        this.scheduleRegularRuns();
+      }
+    }, 10000); // Wait 10 seconds after startup
+  }
+
+  private scheduleRegularRuns() {
+    if (this.state.intervalId) {
+      clearTimeout(this.state.intervalId);
+    }
+
+    if (!this.state.enabled) {
+      this.state.nextRun = undefined;
+      return;
+    }
+
+    const nextRunTime = Date.now() + this.SIX_HOURS_MS;
+    this.state.nextRun = new Date(nextRunTime).toISOString();
+
+    console.log(
+      `🕕 Next batch process scheduled for: ${new Date(nextRunTime).toLocaleString()}`
+    );
+
+    this.state.intervalId = setTimeout(async () => {
+      if (this.state.enabled) {
+        await this.runBatch();
+        this.scheduleRegularRuns(); // Schedule next run
+      }
+    }, this.SIX_HOURS_MS);
+  }
+
+  private async runBatch(): Promise<void> {
+    if (this.state.isRunning) {
+      console.log("⚠️  Batch process already running, skipping");
+      return;
+    }
+
+    this.state.isRunning = true;
+    this.state.lastRun = new Date().toISOString();
+
+    try {
+      console.log("🔄 Running scheduled batch process...");
+      await batchProcess();
+      console.log("✅ Scheduled batch process completed");
+    } catch (error) {
+      console.error("❌ Error during scheduled batch process:", error);
+    } finally {
+      this.state.isRunning = false;
+    }
+  }
+
+  public async triggerManualRun(): Promise<void> {
+    console.log("🚀 Manual batch process triggered");
+    await this.runBatch();
+  }
+
+  public enable(): void {
+    if (this.state.enabled) {
+      console.log("ℹ️  Batch scheduler already enabled");
+      return;
+    }
+
+    this.state.enabled = true;
+    console.log("✅ Batch scheduler enabled");
+    this.scheduleRegularRuns();
+  }
+
+  public disable(): void {
+    if (!this.state.enabled) {
+      console.log("ℹ️  Batch scheduler already disabled");
+      return;
+    }
+
+    this.state.enabled = false;
+    console.log("⏸️  Batch scheduler disabled");
+
+    if (this.state.intervalId) {
+      clearTimeout(this.state.intervalId);
+      this.state.intervalId = undefined;
+    }
+
+    this.state.nextRun = undefined;
+  }
+
+  public getStatus(): BatchSchedulerState {
+    return {
+      ...this.state,
+      intervalId: undefined, // Don't expose the timeout ID
+    };
+  }
+
+  public isEnabled(): boolean {
+    return this.state.enabled;
+  }
+
+  public isRunning(): boolean {
+    return this.state.isRunning;
+  }
+}
+
+// Export singleton instance
+export const batchScheduler = new BatchScheduler();
+
+export type { BatchSchedulerState };