/
index.js
112 lines (90 loc) · 2.24 KB
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
'use strict'
const { promisify } = require('util')
const sleep = promisify(setTimeout)
closeWithGrace.closing = false
function closeWithGrace (opts, fn) {
if (typeof opts === 'function') {
fn = opts
opts = {}
}
const delay = opts.delay ? opts.delay : 10000
const logger = opts.logger ? opts.logger : console
process.once('SIGTERM', onSignal)
process.once('SIGINT', onSignal)
process.once('uncaughtException', onError)
process.once('unhandledRejection', onError)
const sleeped = Symbol('sleeped')
return {
close () {
run({ manual: true })
},
uninstall () {
process.removeListener('SIGTERM', onSignal)
process.removeListener('SIGINT', onSignal)
process.removeListener('uncaughtException', onError)
process.removeListener('unhandledRejection', onError)
}
}
function onSignal (signal) {
run({ signal })
}
function afterFirstSignal (signal) {
logger.error(`second ${signal}, exiting`)
process.exit(1)
}
function onError (err) {
run({ err })
}
function afterFirstError (err) {
logger.error('second error, exiting')
logger.error(err)
process.exit(1)
}
function exec (out) {
const res = fn(out, done)
if (res && typeof res.then === 'function') {
return res
}
let _resolve
let _reject
const p = new Promise(function (resolve, reject) {
_resolve = resolve
_reject = reject
})
return p
function done (err) {
if (!_resolve) {
return
}
if (err) {
_reject(err)
return
}
_resolve()
}
}
async function run (out) {
process.on('SIGTERM', afterFirstSignal)
process.on('SIGINT', afterFirstSignal)
process.on('uncaughtException', afterFirstError)
process.on('unhandledRejection', afterFirstError)
closeWithGrace.closing = true
try {
const res = await Promise.race([
// We create the timer first as fn
// might block the event loop
sleep(delay, sleeped),
exec(out)
])
if (res === sleeped || out.err) {
process.exit(1)
} else {
process.exit(0)
}
} catch (err) {
logger.error(err)
process.exit(1)
}
}
}
module.exports = closeWithGrace