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
|
'use strict'
const https = require('https')
module.exports = (server, grace) => {
grace = typeof grace === 'undefined' ? Infinity : grace
const reqsPerSocket = new Map()
let stopped = false
let gracefully = true
if (server instanceof https.Server) {
server.on('secureConnection', onConnection)
} else {
server.on('connection', onConnection)
}
server.on('request', onRequest)
server.stop = stop
server._pendingSockets = reqsPerSocket
return server
function onConnection (socket) {
reqsPerSocket.set(socket, 0)
socket.once('close', () => reqsPerSocket.delete(socket))
}
function onRequest (req, res) {
reqsPerSocket.set(req.socket, reqsPerSocket.get(req.socket) + 1)
res.once('finish', () => {
const pending = reqsPerSocket.get(req.socket) - 1
reqsPerSocket.set(req.socket, pending)
if (stopped && pending === 0) {
req.socket.end()
}
})
}
function stop (callback) {
// allow request handlers to update state before we act on that state
setImmediate(() => {
stopped = true
if (grace < Infinity) {
setTimeout(destroyAll, grace).unref()
}
server.close(e => {
if (callback) {
callback(e, gracefully)
}
})
reqsPerSocket.forEach(endIfIdle)
})
}
function endIfIdle (requests, socket) {
if (requests === 0) socket.end()
}
function destroyAll () {
gracefully = false
reqsPerSocket.forEach((reqs, socket) => socket.end())
setImmediate(() => {
reqsPerSocket.forEach((reqs, socket) => socket.destroy())
})
}
}
|
