n8n-enterprise-unlocked/packages/core/src/ScheduledTaskManager.ts

import { Service } from 'typedi';
import { CronJob } from 'cron';
import type { CronExpression, Workflow } from 'n8n-workflow';

@Service()
export class ScheduledTaskManager {
	readonly cronJobs = new Map<string, CronJob[]>();

	registerCron(workflow: Workflow, cronExpression: CronExpression, onTick: () => void) {
		const cronJob = new CronJob(cronExpression, onTick, undefined, true, workflow.timezone);
		const cronJobsForWorkflow = this.cronJobs.get(workflow.id);
		if (cronJobsForWorkflow) {
			cronJobsForWorkflow.push(cronJob);
		} else {
			this.cronJobs.set(workflow.id, [cronJob]);
		}
	}

	deregisterCrons(workflowId: string) {
		const cronJobs = this.cronJobs.get(workflowId) ?? [];
		for (const cronJob of cronJobs) {
			cronJob.stop();
		}
	}

	deregisterAllCrons() {
		for (const workflowId of Object.keys(this.cronJobs)) {
			this.deregisterCrons(workflowId);
		}
	}
}