当前端终止请求时，如何取消API Gateway触发的Lambda调用

核心原因 #

API Gateway触发Lambda时，前端取消请求后，API Gateway不会立刻终止Lambda进程，而是会等待Lambda执行完成（除非配置了超时）。更关键的是，你的数据库查询没有正确绑定到Lambda传入的context取消信号上，导致context触发取消时，查询仍在持续运行。

具体修复方案 #

1. 确认数据库驱动支持context取消 #

主流Go数据库驱动（如database/sql标准驱动、pgx等）都支持通过context传递取消信号。如果当前使用的驱动不支持，需要升级或更换兼容的驱动版本。

2. 把Lambda的context传递到所有数据库操作 #

执行数据库查询时，必须将Lambda处理函数接收的context.Context传入查询方法，不能使用默认的context.Background()。示例代码：

func handler(ctx context.Context, request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
    // 错误写法：用空context，无法响应取消信号
    // rows, err := db.Query("SELECT * FROM large_dataset")
    
    // 正确写法：传入Lambda的context
    rows, err := db.QueryContext(ctx, "SELECT * FROM large_dataset")
    if err != nil {
        return events.APIGatewayProxyResponse{}, err
    }
    defer rows.Close()

    // 处理查询结果逻辑...
    return events.APIGatewayProxyResponse{StatusCode: 200}, nil
}

3. 捕获并处理context取消错误 #

当context被取消时，数据库查询会返回context.Canceled错误，需要在代码中捕获这个错误并提前终止Lambda执行：

rows, err := db.QueryContext(ctx, "SELECT * FROM large_dataset")
if err != nil {
    if errors.Is(err, context.Canceled) {
        // 前端已取消请求，直接返回，无需继续处理
        return events.APIGatewayProxyResponse{StatusCode: 499}, nil
    }
    return events.APIGatewayProxyResponse{}, err
}

4. 调整API Gateway配置 #

• 在API Gateway的集成设置中，合理设置超时时间，避免Lambda无限制运行。
• 确保启用"Lambda proxy integration"，保证客户端断开连接的信号能正确传递到Lambda。

5. 理解Lambda执行模型限制 #

Lambda基于容器运行，即使context被取消，进程不会立刻终止，但通过正确绑定context，数据库查询会被中断，避免不必要的资源消耗。你无法强制Lambda立刻停止，但可以终止耗时的数据库操作。

内容的提问来源于stack exchange，提问作者Ori Zerah