chore: if an error occurs during the $connect action, close the conne…

…ction
dherault · Jul 2, 2019 · b83c123 · b83c123
1 parent 1ba79f2
commit b83c123
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/ApiGatewayWebSocket.js b/src/ApiGatewayWebSocket.js
@@ -94,6 +94,11 @@ module.exports = class ApiGatewayWebSocket {
           ws.send(JSON.stringify({ message:'Internal server error', connectionId, requestId:'1234567890' }));
         }
 
+        // mimic AWS behaviour (close connection) when the $connect action handler throws
+        if (name === '$connect') {
+          ws.close();
+        }
+
         debugLog(`Error in handler of action ${action}`, err);
       };