109 lines
2.8 KiB
JavaScript
109 lines
2.8 KiB
JavaScript
let instance = null;
|
|
|
|
const defaultTimeout = 15*1000;//15 sec
|
|
const exitSignals = ['SIGINT', 'SIGTERM', 'SIGBREAK', 'SIGHUP', 'uncaughtException', 'SIGUSR2'];
|
|
|
|
//singleton
|
|
class AsyncExit {
|
|
constructor() {
|
|
if (!instance) {
|
|
this.onSignalCallbacks = new Map();
|
|
this.callbacks = new Map();
|
|
this.afterCallbacks = new Map();
|
|
this.exitTimeout = defaultTimeout;
|
|
this.inited = false;
|
|
instance = this;
|
|
}
|
|
|
|
return instance;
|
|
}
|
|
|
|
init(signals = exitSignals, codeOnSignal = 2) {
|
|
if (this.inited)
|
|
throw new Error('AsyncExit: initialized already');
|
|
|
|
const runSingalCallbacks = async(signal) => {
|
|
for (const signalCallback of this.onSignalCallbacks.keys()) {
|
|
try {
|
|
await signalCallback(signal);
|
|
} catch(e) {
|
|
console.error(e);
|
|
}
|
|
}
|
|
};
|
|
|
|
for (const signal of signals) {
|
|
process.once(signal, async() => {
|
|
await runSingalCallbacks(signal);
|
|
this.exit(codeOnSignal);
|
|
});
|
|
}
|
|
|
|
this.inited = true;
|
|
}
|
|
|
|
onSignal(signalCallback) {
|
|
if (!this.onSignalCallbacks.has(signalCallback)) {
|
|
this.onSignalCallbacks.set(signalCallback, true);
|
|
}
|
|
}
|
|
|
|
add(exitCallback) {
|
|
if (!this.callbacks.has(exitCallback)) {
|
|
this.callbacks.set(exitCallback, true);
|
|
}
|
|
}
|
|
|
|
addAfter(exitCallback) {
|
|
if (!this.afterCallbacks.has(exitCallback)) {
|
|
this.afterCallbacks.set(exitCallback, true);
|
|
}
|
|
}
|
|
|
|
remove(exitCallback) {
|
|
if (this.callbacks.has(exitCallback)) {
|
|
this.callbacks.delete(exitCallback);
|
|
}
|
|
if (this.afterCallbacks.has(exitCallback)) {
|
|
this.afterCallbacks.delete(exitCallback);
|
|
}
|
|
}
|
|
|
|
setExitTimeout(timeout) {
|
|
this.exitTimeout = timeout;
|
|
}
|
|
|
|
exit(code = 0) {
|
|
if (this.exiting)
|
|
return;
|
|
|
|
this.exiting = true;
|
|
|
|
const timer = setTimeout(() => { process.exit(code); }, this.exitTimeout);
|
|
|
|
(async() => {
|
|
for (const exitCallback of this.callbacks.keys()) {
|
|
try {
|
|
await exitCallback();
|
|
} catch(e) {
|
|
console.error(e);
|
|
}
|
|
}
|
|
|
|
for (const exitCallback of this.afterCallbacks.keys()) {
|
|
try {
|
|
await exitCallback();
|
|
} catch(e) {
|
|
console.error(e);
|
|
}
|
|
}
|
|
|
|
clearTimeout(timer);
|
|
//console.log('Exited gracefully');
|
|
process.exit(code);
|
|
})();
|
|
}
|
|
}
|
|
|
|
module.exports = AsyncExit;
|