若想将AWS API Gateway WebSocket用于实时数据传输，需要遵循以下步骤。

1.在AWS控制台中创建新的WebSocket API。

2.创建新的WebSocket资源并设置其“Route Selection Expression”为 $request.body.action

3.设置新的WebSocket部署。

4.将Lambda函数绑定至WebSocket API，使API存活时间较长。需要确保函数的执行时间足够长，以保持WebSocket连接处于打开状态。

5.在Lambda函数中，使用generateURL方法生成连接的URL，并将其返回给客户端。

以下是使用Node.js在Lambda函数中生成WebSocket连接URL的示例代码：

const AWS = require('aws-sdk'); const apigatewaymanagementapi = new AWS.ApiGatewayManagementApi({endpoint: 'API_GATEWAY_ENDPOINT'});

exports.handler = async (event, context) => { const connectionId = event.requestContext.connectionId; const endpoint = event.requestContext.domainName + '/' + event.requestContext.stage; const postData = JSON.parse(event.body).data;

try { await postToConnection(endpoint, connectionId, postData); return { statusCode: 200, body: 'Data sent.' }; } catch (e) { return { statusCode: 500, body: e.stack }; } };

function postToConnection(endpoint, connectionId, postData) { return apigatewaymanagementapi.postToConnection({ ConnectionId: connectionId, Data: postData, Endpoint: endpoint }).promise(); }

请注意将代码示例中的“API_GATEWAY_ENDPOINT”替换为您的API Gateway WebSocket的端点。