livedash-node/lib/schedulers.ts

// Combined scheduler initialization with graceful shutdown

import { auditLogScheduler } from "./auditLogScheduler";
import { prisma } from "./prisma";
import { startProcessingScheduler } from "./processingScheduler";
import { startCsvImportScheduler } from "./scheduler";

/**
 * Initialize all schedulers
 * - CSV import scheduler (runs every 15 minutes)
 * - Session processing scheduler (runs every hour)
 * - Audit log retention scheduler (runs weekly by default)
 */
export function initializeSchedulers() {
  // Start the CSV import scheduler
  startCsvImportScheduler();

  // Start the session processing scheduler
  startProcessingScheduler();

  // Start the audit log retention scheduler
  if (process.env.AUDIT_LOG_RETENTION_ENABLED !== "false") {
    auditLogScheduler.start();
    console.log("Audit log retention scheduler started");
  } else {
    console.log("Audit log retention scheduler disabled");
  }

  console.log("All schedulers initialized successfully");

  // Set up graceful shutdown for schedulers
  setupGracefulShutdown();
}

/**
 * Set up graceful shutdown handling for schedulers and database connections
 */
function setupGracefulShutdown() {
  const shutdown = async (signal: string) => {
    console.log(`\nReceived ${signal}. Starting graceful shutdown...`);

    try {
      // Stop the audit log scheduler
      auditLogScheduler.stop();
      console.log("Audit log scheduler stopped.");

      // Disconnect from database
      await prisma.$disconnect();
      console.log("Database connections closed.");

      // Exit the process
      process.exit(0);
    } catch (error) {
      console.error("Error during shutdown:", error);
      process.exit(1);
    }
  };

  // Handle various termination signals
  process.on("SIGTERM", () => shutdown("SIGTERM"));
  process.on("SIGINT", () => shutdown("SIGINT"));
  process.on("SIGUSR2", () => shutdown("SIGUSR2")); // Nodemon restart

  // Handle uncaught exceptions
  process.on("uncaughtException", (error) => {
    console.error("Uncaught Exception:", error);
    shutdown("uncaughtException");
  });

  process.on("unhandledRejection", (reason, promise) => {
    console.error("Unhandled Rejection at:", promise, "reason:", reason);
    shutdown("unhandledRejection");
  });
}