Fabio/webssh2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add shutdown middleware to prevent requests

Browse source
This commit is contained in:
Edgar Onghena 2020-03-18 19:35:52 +01:00
parent 61f19ae3ce
commit 61672c3f1e
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
app/server/app.js
View file

@ -121,6 +121,7 @@ var expressOptions = require('./expressOptions')
var favicon = require('serve-favicon'); var favicon = require('serve-favicon');
// express // express
app.use(safeShutdownGuard)
app.use(session) app.use(session)
app.use(myutil.basicAuth) app.use(myutil.basicAuth)
if (config.accesslog) app.use(logger('common')) if (config.accesslog) app.use(logger('common'))
@ -205,6 +206,11 @@ var shutdownMode = false
var shutdownInterval = 0 var shutdownInterval = 0
var connectionCount = 0 var connectionCount = 0
function safeShutdownGuard (req, res, next) {
if (shutdownMode) res.status(503).end('Service unavailable: Server shutting down')
else return next()
}
io.on('connection', function (socket) { io.on('connection', function (socket) {
connectionCount++ connectionCount++