132 lines
3.1 KiB
TypeScript
132 lines
3.1 KiB
TypeScript
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 }; |