67 lines
1.6 KiB
JavaScript
67 lines
1.6 KiB
JavaScript
const EventEmitter = require('events');
|
|
const path = require('path');
|
|
const { fork } = require('child_process');
|
|
const { getId } = require('../../../utils/index')
|
|
|
|
const daemonPath = `${__dirname}/daemon.js`;
|
|
|
|
class BackgroundScheduledTask extends EventEmitter {
|
|
constructor(cronExpression, taskPath, options) {
|
|
super();
|
|
if (!options) {
|
|
options = {
|
|
scheduled: true,
|
|
recoverMissedExecutions: false,
|
|
};
|
|
}
|
|
this.cronExpression = cronExpression;
|
|
this.taskPath = taskPath;
|
|
this.options = options;
|
|
this.options.name = this.options.name || getId()
|
|
|
|
if (options.scheduled) {
|
|
this.start();
|
|
}
|
|
}
|
|
|
|
start() {
|
|
this.stop();
|
|
this.forkProcess = fork(daemonPath);
|
|
|
|
this.forkProcess.on('message', (message) => {
|
|
switch (message.type) {
|
|
case 'task-done':
|
|
this.emit('task-done', message.result);
|
|
break;
|
|
}
|
|
});
|
|
|
|
let options = this.options;
|
|
options.scheduled = true;
|
|
|
|
this.forkProcess.send({
|
|
type: 'register',
|
|
path: path.resolve(this.taskPath),
|
|
cron: this.cronExpression,
|
|
options: options
|
|
});
|
|
}
|
|
|
|
stop() {
|
|
if (this.forkProcess) {
|
|
this.forkProcess.kill();
|
|
}
|
|
}
|
|
|
|
pid() {
|
|
if (this.forkProcess) {
|
|
return this.forkProcess.pid;
|
|
}
|
|
}
|
|
|
|
isRunning() {
|
|
return !this.forkProcess.killed;
|
|
}
|
|
}
|
|
|
|
module.exports = BackgroundScheduledTask; |