AWS API Gateway WebSocket无法向客户端推送数据求助

核心问题：异步方法未等待 #

你的代码中调用了PostToConnectionAsync异步方法但未使用await等待其完成。Lambda函数会在执行到return语句时直接终止进程，导致WebSocket发送请求还没完成就被中断，因此既没有数据发送成功，也不会捕获到异步抛出的异常。

修正后的代码 #

public async Task<string> FunctionHandler(string input, ILambdaContext context)
{
    try
    {
        // 不推荐硬编码凭证，建议使用Lambda执行角色赋予API Gateway权限
        string SK = "xxxxxx";
        string AK = "xxxx";

        var stream = new MemoryStream(UTF8Encoding.UTF8.GetBytes(input));
        
        var apiClient = new AmazonApiGatewayManagementApiClient(AK, SK, new AmazonApiGatewayManagementApiConfig 
        { 
            ServiceURL = "https://xxxxxx.execute-api.us-east-1.amazonaws.com/Test" 
        });
        
        // 必须await异步方法，确保请求完整执行
        await apiClient.PostToConnectionAsync(new Amazon.ApiGatewayManagementApi.Model.PostToConnectionRequest
        {
            ConnectionId = "fdCqPfd0oAMCJmg=",
            Data = stream
        });

        return input.ToUpper();
    }
    catch(Exception ex)
    {
        context.Logger.LogError(ex, "发送WebSocket消息失败");
        return ex.ToString();
    }
}

关键修改说明 #

1. 方法签名改为异步：将string返回类型改为Task<string>，添加async关键字，支持异步等待
2. 添加await关键字：确保Lambda等待PostToConnectionAsync执行完成后再返回，避免进程提前终止
3. 日志记录异常：通过context.Logger.LogError将异常写入CloudWatch，方便后续排查问题
4. 安全提示：硬编码AK/SK不符合安全最佳实践，建议通过Lambda执行角色赋予execute-api:ManageConnections权限

额外排查方向 #

• 验证ConnectionId有效性：WebSocket连接断开后，对应的ConnectionId会失效，需确认使用的ID来自当前活跃的客户端连接
• 检查IAM权限：确保Lambda角色拥有execute-api:ManageConnections权限，且资源范围覆盖当前API Gateway的WebSocket端点
• 查看CloudWatch日志：修正代码后，若仍有问题，可在CloudWatch中查看Lambda的执行日志，获取具体异常信息（如权限不足、端点错误等）
• 测试端点连通性：直接通过Postman或curl调用API Gateway的@connections端点，验证是否能正常发送数据，排除Lambda代码外的问题

内容的提问来源于stack exchange，提问作者Hardik Parmar