Tweaking timeouts and adding try/catch

Author: Perlkonig

api/sockets/connectHandler.ts

@@ -18,30 +18,35 @@ const ddb = DynamoDBDocumentClient.from(new DynamoDBClient({}));
 
 export const handler = async (event: WebSocketConnectEvent) => {
   console.log("Connect event:", JSON.stringify(event));
-  const { connectionId, authorizer } = event.requestContext;
-
-  if (!authorizer?.userId) {
-    console.error("Missing userId in authorizer context");
-    return { statusCode: 401 };
+  try {
+    const { connectionId, authorizer } = event.requestContext;
+
+    if (!authorizer?.userId) {
+        console.error("Missing userId in authorizer context");
+        return { statusCode: 401 };
+    }
+
+    const userId = authorizer.userId;
+
+    await ddb.send(
+        new PutCommand({
+        TableName: process.env.ABSTRACT_PLAY_TABLE!,
+        Item: {
+            pk: "wsConnections",
+            sk: connectionId,
+
+            connectionId,
+            userId,
+
+            // Optional TTL for auto-cleanup
+            ttl: Math.floor(Date.now() / 1000) + 3600,
+        },
+        })
+    );
+  } catch (ex) {
+    console.log("Connect error:", JSON.stringify(ex));
+    return { statusCode: 500 };
   }
 
-  const userId = authorizer.userId;
-
-  await ddb.send(
-    new PutCommand({
-      TableName: process.env.ABSTRACT_PLAY_TABLE!,
-      Item: {
-        pk: "wsConnections",
-        sk: connectionId,
-
-        connectionId,
-        userId,
-
-        // Optional TTL for auto-cleanup
-        ttl: Math.floor(Date.now() / 1000) + 3600,
-      },
-    })
-  );
-
   return { statusCode: 200 };
 };

api/sockets/disconnectHandler.ts

@@ -30,6 +30,7 @@ const translateConfig = { marshallOptions, unmarshallOptions };
 const ddbDocClient = DynamoDBDocumentClient.from(clnt, translateConfig);
 
 export const handler = async (event: WebSocketDisconnectEvent) => {
+  console.log("Disconnect event:", JSON.stringify(event));
   const { connectionId } = event.requestContext;
 
   try {

serverless.yml

@@ -314,6 +314,7 @@ functions:
   connect:
     handler: api/sockets/connectHandler.handler
     logRetentionInDays: 7
+    timeout: 30
     events:
       - websocket:
           route: $connect
@@ -323,6 +324,7 @@ functions:
 
   disconnect:
     handler: api/sockets/disconnectHandler.handler
+    timeout: 30
     logRetentionInDays: 7
     events:
       - websocket:
@@ -331,6 +333,7 @@ functions:
   messageHandler:
     handler: api/sockets/messageHandler.handler
     logRetentionInDays: 7
+    timeout: 30
     environment:
       DOMAIN_NAME: ${self:custom.websocketDomain}
       STAGE: ${self:provider.stage}