Close the server

daquinoaldo · daquinoaldo · commit 0a6fe8a5866f · 2018-09-01T16:55:27.000+02:00
diff --git a/README.md b/README.md
@@ -41,6 +41,9 @@ const app = require('https-localhost')
 app.get('/', (req, res) => res.send('Hello World!'))
 ```
 
+You can close the server with `app.close()`.  
+Closing the server is an async operation, you can get notified with `app.close(callback)` or `app.close().then(...)`.
+
 ---
 
 ### License
diff --git a/index.js b/index.js
@@ -15,26 +15,56 @@ const certOptions = {
 
 // run express on 443
 const app = express()
-https.createServer(certOptions, app).listen(443)
+app.server = https.createServer(certOptions, app).listen(443)
+// save sockets for fast close
+const sockets = []
+let nextSocketId = 0
+app.server.on('connection', socket => {
+  const socketId = nextSocketId++
+  sockets[socketId] = socket
+  socket.on('close', () => delete sockets[socketId])
+})
 
 // gzip compression and minify
 app.use(compression())
 app.set('json spaces', 0)
 
 // redirect http to https
-http.createServer(function (req, res) {
+app.http = http.createServer(function (req, res) {
   res.writeHead(301, { "Location": "https://" + req.headers['host'] + req.url })
   res.end()
 }).listen(80)
+// save sockets for fast close
+app.server.on('connection', socket => {
+  const socketId = nextSocketId++
+  sockets[socketId] = socket
+  socket.on('close', () => delete sockets[socketId])
+})
 
 // ready
-console.info("Server running on port 443.")
+if (!process.env.TEST) console.info("Server running on port 443.")
 
 // serve static files, launch as: 'node index.js <static-path>'
 if (require.main === module) {  // called directly (not through require)
   const staticPath = process.argv[2]
   app.use(express.static(staticPath || process.cwd()))
 }
 
+// close the app
+app.close = (callback) => {
+  const promises = [
+    new Promise(resolve => app.http.close(resolve)),
+    new Promise(resolve => app.server.close(resolve))
+  ]
+  // destroy all opens
+  for (const socketId in sockets)
+    sockets[socketId].destroy()
+
+  return Promise.all(promises).then(() => {
+    if (!process.env.TEST) console.info("Server closed.")
+    if (callback) callback()
+  })
+}
+
 // export as module
 module.exports = app
