mirror of
https://github.com/JonasunderscoreJones/api-worker.git
synced 2025-10-23 10:29:19 +02:00
Initial commit (by create-cloudflare CLI)
This commit is contained in:
commit
58a42872a0
1745 changed files with 741893 additions and 0 deletions
65
node_modules/stoppable/lib/stoppable.js
generated
vendored
Normal file
65
node_modules/stoppable/lib/stoppable.js
generated
vendored
Normal file
|
@ -0,0 +1,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())
|
||||
})
|
||||
}
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue