62 lines
1.6 KiB
JavaScript
62 lines
1.6 KiB
JavaScript
import cron from 'node-cron';
|
|
import { cron_task_list } from '../config/cron_tasks.js';
|
|
import { execute_action_and_record } from './task_executor.js';
|
|
import { get_flow_runner } from './flows/flow_registry.js';
|
|
|
|
const cron_jobs = [];
|
|
const running_task_name_set = new Set();
|
|
|
|
async function run_cron_task(task) {
|
|
if (!task || !task.type) {
|
|
throw new Error('cron_task 缺少 type');
|
|
}
|
|
|
|
if (task.type === 'action') {
|
|
await execute_action_and_record({
|
|
action_name: task.action_name,
|
|
action_payload: task.action_payload || {},
|
|
source: 'cron'
|
|
});
|
|
return;
|
|
}
|
|
|
|
if (task.type === 'flow') {
|
|
const run_flow = get_flow_runner(task.flow_name);
|
|
await run_flow(task.flow_payload || {});
|
|
return;
|
|
}
|
|
|
|
throw new Error(`cron_task type 不支持: ${task.type}`);
|
|
}
|
|
|
|
export async function start_all_cron_tasks() {
|
|
for (const task of cron_task_list) {
|
|
const task_name = task && task.name ? String(task.name) : 'cron_task';
|
|
const job = cron.schedule(task.cron_expression, async () => {
|
|
if (running_task_name_set.has(task_name)) {
|
|
// eslint-disable-next-line no-console
|
|
console.log('[cron] skip (already running)', { name: task_name });
|
|
return;
|
|
}
|
|
|
|
running_task_name_set.add(task_name);
|
|
try {
|
|
await run_cron_task(task);
|
|
} finally {
|
|
running_task_name_set.delete(task_name);
|
|
}
|
|
});
|
|
console.log('job', { task_name, });
|
|
|
|
cron_jobs.push(job);
|
|
}
|
|
}
|
|
|
|
export function stop_all_cron_tasks() {
|
|
for (const job of cron_jobs) {
|
|
job.stop();
|
|
}
|
|
cron_jobs.length = 0;
|
|
running_task_name_set.clear();
|
|
}
|