fix(apigatewayv2): trigger on websocket connect and disconnect is not working

packages/@aws-cdk/aws-apigatewayv2-integrations-alpha/lib/websocket/lambda.ts

@@ -33,7 +33,7 @@ export class WebSocketLambdaIntegration extends WebSocketRouteIntegration {
       sourceArn: Stack.of(route).formatArn({
         service: 'execute-api',
         resource: route.webSocketApi.apiId,
-        resourceName: `*/*${route.routeKey}`,
+        resourceName: `*${route.routeKey}`,
       }),
     });
 

packages/@aws-cdk/aws-apigatewayv2-integrations-alpha/test/websocket/integ.lambda-connect-disconnect-trigger.ts

@@ -0,0 +1,76 @@
+import { WebSocketApi, WebSocketStage } from '@aws-cdk/aws-apigatewayv2-alpha';
+import * as cdk from 'aws-cdk-lib';
+import { WebSocketLambdaIntegration } from '../../lib';
+import * as lambda from 'aws-cdk-lib/aws-lambda';
+import * as path from 'path';
+import * as dynamodb from 'aws-cdk-lib/aws-dynamodb';
+
+const app = new cdk.App();
+
+const stack = new cdk.Stack(app, 'integ-apigwv2-lambda-connect-integration');
+const webSocketTableName = 'WebSocketConnections';
+
+const connectFunction = new lambda.Function(stack, 'Connect Function', {
+  functionName: 'process_connect_requests',
+  runtime: lambda.Runtime.NODEJS_14_X,
+  handler: 'index.handler',
+  code: lambda.Code.fromAsset(path.join(__dirname, 'lambdas', 'connect')),
+  timeout: cdk.Duration.seconds(5),
+  environment: {
+    TABLE_NAME: webSocketTableName,
+  },
+});
+
+const disconnectFunction = new lambda.Function(
+  stack,
+  'Disconnect Function',
+  {
+    functionName: 'process_disconnect_requests',
+    runtime: lambda.Runtime.NODEJS_14_X,
+    handler: 'index.handler',
+    code: lambda.Code.fromAsset(
+      path.join(__dirname, 'lambdas', 'disconnect'),
+    ),
+    timeout: cdk.Duration.seconds(5),
+    environment: {
+      TABLE_NAME: webSocketTableName,
+    },
+  },
+);
+
+const webSocketLogTable = new dynamodb.Table(stack, 'WebSocket Log Table', {
+  tableName: webSocketTableName,
+  partitionKey: {
+    name: 'ConnectionId',
+    type: dynamodb.AttributeType.STRING,
+  },
+  removalPolicy: cdk.RemovalPolicy.DESTROY, // not recommended for production
+});
+webSocketLogTable.grantReadWriteData(connectFunction);
+webSocketLogTable.grantReadWriteData(disconnectFunction);
+
+const webSocketConnectIntegration = new WebSocketLambdaIntegration(
+  'ConnectIntegration',
+  connectFunction,
+);
+const webSocketDisconnectIntegration = new WebSocketLambdaIntegration(
+  'DisconnectIntegration',
+  disconnectFunction,
+);
+
+const webSocketApi = new WebSocketApi(stack, 'WebSocket API', {
+  apiName: 'webSocket',
+  routeSelectionExpression: '$request.body.action',
+  connectRouteOptions: { integration: webSocketConnectIntegration },
+  disconnectRouteOptions: { integration: webSocketDisconnectIntegration },
+});
+
+new WebSocketStage(
+  stack,
+  'Production Stage',
+  {
+    webSocketApi: webSocketApi,
+    stageName: 'prod',
+    autoDeploy: true,
+  },
+);

packages/@aws-cdk/aws-apigatewayv2-integrations-alpha/test/websocket/lambdas/connect/package.json

@@ -0,0 +1,11 @@
+{
+  "name": "connect",
+  "version": "1.0.0",
+  "description": "connect example for WebSockets on API Gateway",
+  "main": "index.js",
+  "author": "SAM CLI",
+  "license": "MIT",
+  "dependencies": {
+    "aws-sdk": "^2.690.0"
+  }
+}

packages/@aws-cdk/aws-apigatewayv2-integrations-alpha/test/websocket/lambdas/disconnect/package.json

@@ -0,0 +1,11 @@
+{
+  "name": "disconnect",
+  "version": "1.0.0",
+  "description": "disconnect example for WebSockets on API Gateway",
+  "main": "index.js",
+  "author": "SAM CLI",
+  "license": "MIT",
+  "dependencies": {
+    "aws-sdk": "^2.690.0"
+  }
+}