Code-MCP

import cron from "node-cron"; import { logger } from "./logger"; // Assuming logger exists // ============================================ // Job Definitions // ============================================ const jobs = { // Run every minute heartbeat: { schedule: "* * * * *", task: async () => { logger.info("Heartbeat: System is alive"); }, }, // Run every night at midnight dailyCleanup: { schedule: "0 0 * * *", task: async () => { logger.info("Starting daily cleanup..."); // Db cleanup logic here await new Promise((resolve) => setTimeout(resolve, 1000)); logger.info("Daily cleanup completed"); }, }, // Run every Monday at 9am weeklyReport: { schedule: "0 9 * * 1", task: async () => { logger.info("Generating weekly report..."); // Report logic here }, }, }; // ============================================ // Scheduler Service // ============================================ type JobName = keyof typeof jobs; export class Scheduler { private tasks: Map<JobName, cron.ScheduledTask> = new Map(); init() { logger.info("Initializing scheduler..."); Object.entries(jobs).forEach(([name, config]) => { const task = cron.schedule( config.schedule, async () => { logger.info(`Starting job: ${name}`); try { await config.task(); logger.info(`Job completed: ${name}`); } catch (error) { logger.error(`Job failed: ${name}`, error); } }, { scheduled: false, // Don't start immediately timezone: "UTC", }, ); this.tasks.set(name as JobName, task); }); this.startAll(); } startAll() { this.tasks.forEach((task) => task.start()); logger.info("All scheduled tasks started"); } stopAll() { this.tasks.forEach((task) => task.stop()); logger.info("All scheduled tasks stopped"); } stop(name: JobName) { const task = this.tasks.get(name); if (task) { task.stop(); logger.info(`Stopped job: ${name}`); } } start(name: JobName) { const task = this.tasks.get(name); if (task) { task.start(); logger.info(`Started job: ${name}`); } } } export const scheduler = new Scheduler(); // Start automatically if not imported as module if (require.main === module) { scheduler.init(); }