wassname/talk
1
0
Fork 0
mirror of https://github.com/wassname/talk.git synced 2026-08-14 12:50:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added PID support

Browse Source
This commit is contained in:
Wyatt Johnson
2017-01-25 11:11:33 -07:00
parent f023d1de08
commit edf9ced454
6 changed files with 78 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
util.js
+14 -5
View File
@@ -1,3 +1,4 @@
const debug = require('debug')('talk:util');
const util = module.exports = {};
/**
@@ -11,10 +12,16 @@ util.toshutdown = [];
* Calls all the shutdown functions and then ends the process.
* @param {Number} [defaultCode=0] default return code upon sucesfull shutdown.
*/
util.shutdown = (defaultCode = 0) => {
util.shutdown = (defaultCode = 0, signal = null) => {
if (signal) {
debug(`Reached ${signal} signal`);
}
Promise
.all(util.toshutdown.map((func) => func ? func() : null).filter((func) => func))
.all(util.toshutdown.map((func) => func ? func(signal) : null).filter((func) => func))
.then(() => {
debug('Shutdown complete, now exiting');
process.exit(defaultCode);
})
.catch((err) => {
@@ -32,6 +39,8 @@ util.shutdown = (defaultCode = 0) => {
*/
util.onshutdown = (jobs) => {
debug(`${jobs.length} jobs registered`);
// Add the new jobs to shutdown to the object reference.
util.toshutdown = util.toshutdown.concat(jobs);
};
@@ -39,6 +48,6 @@ util.onshutdown = (jobs) => {
// Attach to the SIGTERM + SIGINT handles to ensure a clean shutdown in the
// event that we have an external event. SIGUSR2 is called when the app is asked
// to be 'killed', same procedure here.
process.on('SIGTERM', () => util.shutdown());
process.on('SIGINT', () => util.shutdown());
process.once('SIGUSR2', () => util.shutdown());
process.on('SIGTERM', () => util.shutdown(0, 'SIGTERM'));
process.on('SIGINT', () => util.shutdown(0, 'SIGINT'));
process.once('SIGUSR2', () => util.shutdown(0, 'SIGUSR2'));