-
Notifications
You must be signed in to change notification settings - Fork 459
/
shutdown.js
74 lines (61 loc) · 2.97 KB
/
shutdown.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
/*───────────────────────────────────────────────────────────────────────────*\
│ Copyright (C) 2014 eBay Software Foundation │
│ │
│hh ,'""`. │
│ / _ _ \ Licensed under the Apache License, Version 2.0 (the "License"); │
│ |(@)(@)| you may not use this file except in compliance with the License. │
│ ) __ ( You may obtain a copy of the License at │
│ /,'))((`.\ │
│(( (( )) )) http://www.apache.org/licenses/LICENSE-2.0 │
│ `\ `)(' /' │
│ │
│ Unless required by applicable law or agreed to in writing, software │
│ distributed under the License is distributed on an "AS IS" BASIS, │
│ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. │
│ See the License for the specific language governing permissions and │
│ limitations under the License. │
\*───────────────────────────────────────────────────────────────────────────*/
'use strict';
var States = {
CONNECTED: 0,
DISCONNECTING: 2
};
module.exports = function (config) {
var template, timeout, state, app, server;
function close() {
state = States.DISCONNECTING;
app.emit('shutdown', server, timeout);
}
config = config || {};
template = config.template;
timeout = config.timeout || 10 * 1000;
state = States.CONNECTED;
return function shutdown(req, res, next) {
var headers = config.shutdownHeaders || {};
function json() {
res.send('Server is shutting down.');
}
function html() {
template ? res.render(template) : json();
}
if (state === States.DISCONNECTING) {
headers.Connection = headers.Connection || 'close';
res.header(headers);
res.status(503);
res.format({
json: json,
html: html
});
return;
}
if (!app) {
// Lazy-bind - only attempt clean shutdown
// if we've taken at least one request.
app = req.app;
server = req.socket.server;
process.once('SIGTERM', close);
process.once('SIGINT', close);
}
next();
};
};