persist games in fixed intervals and on shutdown

common/Util.js

@@ -7,6 +7,19 @@ export const randomInt = (min, max) => Math.floor(Math.random() * (max - min + 1
 // get one random item from the given array
 export const choice = (array) => array[randomInt(0, array.length - 1)]
 
+export const throttle = (fn, delay) => {
+  let canCall = true
+  return (...args) => {
+    if (canCall) {
+      fn.apply(null, args)
+      canCall = false
+      setTimeout(() => {
+        canCall = true
+      }, delay)
+    }
+  }
+}
+
 // return a shuffled (shallow) copy of the given array
 export const shuffle = (array) => {
   let arr = array.slice()
@@ -37,6 +50,7 @@ export default {
   uniqId,
   randomInt,
   choice,
+  throttle,
   shuffle,
   timestamp,
 }

scripts/server

@@ -2,4 +2,4 @@
 
 cd "$RUN_DIR/server"
 
-nodemon -d 2 index.js
+nodemon index.js

server/Game.js

@@ -30,18 +30,21 @@ function loadAllGames() {
   }
 }
 
-function store(gameId) {
+function persistAll() {
-  const game = GameCommon.get(gameId)
+  const games = GameCommon.getAllGames()
+  for (const gameId of Object.keys(games)) {
+    const game = games[gameId]
     fs.writeFileSync('./../data/' + gameId + '.json', JSON.stringify({
       puzzle: game.puzzle,
       players: game.players,
     }))
+  }
 }
 
 export default {
   loadAllGames,
   getAllGames: GameCommon.getAllGames,
-  store,
+  persistAll,
   createGame,
   exists: GameCommon.exists,
   addPlayer: GameCommon.addPlayer,

server/WebSocketServer.js

@@ -61,6 +61,10 @@ class WebSocketServer {
     })
   }
 
+  close() {
+    this._websocketserver.close()
+  }
+
   notifyOne(data, socket) {
     socket.send(JSON.stringify(data))
   }

server/index.js

@@ -109,30 +109,6 @@ wss.on('close', async ({socket}) => {
   }
 })
 
-const ensurePlayerGame = (gameId, clientId, socket) => {
-  Game.addPlayer(gameId, clientId)
-  Game.addSocket(gameId, socket)
-  Game.store(gameId)
-  const game = Game.get(gameId)
-  notify(
-    [Protocol.EV_SERVER_INIT, {
-      puzzle: game.puzzle,
-      players: game.players,
-      evtInfos: game.evtInfos,
-    }],
-    [socket]
-  )
-}
-
-const handlePlayerInput = (gameId, clientId, clientSeq, clientEvtData) => {
-  const changes = Game.handleInput(gameId, clientId, clientEvtData)
-  Game.store(gameId)
-  notify(
-    [Protocol.EV_SERVER_EVENT, clientId, clientSeq, changes],
-    Game.getSockets(gameId)
-  )
-}
-
 wss.on('message', async ({socket, data}) => {
   try {
     const proto = socket.protocol.split('|')
@@ -145,14 +121,39 @@ wss.on('message', async ({socket, data}) => {
         if (!Game.exists(gameId)) {
           throw `[game ${gameId} does not exist... ]`
         }
-        ensurePlayerGame(gameId, clientId, socket)
+        Game.addPlayer(gameId, clientId)
+        Game.addSocket(gameId, socket)
+        const game = Game.get(gameId)
+        notify(
+          [Protocol.EV_SERVER_INIT, {
+            puzzle: game.puzzle,
+            players: game.players,
+            evtInfos: game.evtInfos,
+          }],
+          [socket]
+        )
       } break;
 
       case Protocol.EV_CLIENT_EVENT: {
         const clientSeq = msg[1]
         const clientEvtData = msg[2]
-        ensurePlayerGame(gameId, clientId, socket)
+        Game.addPlayer(gameId, clientId)
-        handlePlayerInput(gameId, clientId, clientSeq, clientEvtData)
+        Game.addSocket(gameId, socket)
+
+        const game = Game.get(gameId)
+        notify(
+          [Protocol.EV_SERVER_INIT, {
+            puzzle: game.puzzle,
+            players: game.players,
+            evtInfos: game.evtInfos,
+          }],
+          [socket]
+        )
+        const changes = Game.handleInput(gameId, clientId, clientEvtData)
+        notify(
+          [Protocol.EV_SERVER_EVENT, clientId, clientSeq, changes],
+          Game.getSockets(gameId)
+        )
       } break;
     }
   } catch (e) {
@@ -161,5 +162,35 @@ wss.on('message', async ({socket, data}) => {
 })
 
 Game.loadAllGames()
-app.listen(port, hostname, () => console.log(`server running on http://${hostname}:${port}`))
+const server = app.listen(
+  port,
+  hostname,
+  () => console.log(`server running on http://${hostname}:${port}`)
+)
 wss.listen()
+
+// persist games in fixed interval
+setInterval(() => {
+  console.log('Persisting games...');
+  Game.persistAll()
+}, config.persistence.interval)
+
+const gracefulShutdown = (signal) => {
+  console.log(`${signal} received...`)
+  Game.persistAll()
+  server.close()
+  wss.close()
+}
+
+// used by nodemon
+process.once('SIGUSR2', function () {
+  gracefulShutdown('SIGUSR2')
+});
+
+process.once('SIGINT', function (code) {
+  gracefulShutdown('SIGINT')
+});
+
+process.once('SIGTERM', function (code) {
+  gracefulShutdown('SIGTERM')
+});