Merge pull request #726 from frsechet/fix/aws-difference

close connection on error in $connect action handler
dherault · Jul 2, 2019 · b9b1d72 · b9b1d72
2 parents 27e1e05 + b83c123
commit b9b1d72
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/ApiGatewayWebSocket.js b/src/ApiGatewayWebSocket.js
@@ -98,6 +98,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);
       };