在 AWS Lambda 中使用 Application Load Balancer (ALB) 时获取实际客户端 IP

在现代云架构中，使用无服务器技术和负载均衡器是相当普遍的做法。在 AWS Lambda 和应用程序负载均衡器（ALB）的组合使用场景中，获取实际客户端的 IP 地址可能会导致一些困惑。因为通过 ALB 进行请求时，原来的客户端 IP 地址通常被 ALB 的 IP 地址替换了。本文将详细探讨如何在这样的架构中获取真实的客户端 IP 地址。

理解 AWS Lambda 与 ALB 的基本架构

在 AWS 中，Lambda 是一种无服务器计算服务，它允许用户运行代码，而无需担心底层基础设施。当使用 ALB 时，Lambda 可以作为其中一个目标，将 HTTP 请求路由到 Lambda 函数。ALB 支持不同类型的负载均衡，并能够对流量进行智能路由。

ALB 的工作原理

ALB 在接收到请求时，会将其转发给配置的后端目标。在使用 Lambda 的情况下，这意味着 ALB 会将请求转发给 Lambda 函数。这一转换过程的核心是请求信息的重新构造，这可能会导致原始客户端 IP 地址丢失。

请求头中传递 IP 地址

为了获取实际的客户端 IP 地址，ALB 会自动在请求的头部添加一个名为 `X-Forwarded-For` 的字段。此字段包含了客户端的原始 IP 地址及代理链中其他服务器的 IP 地址。我们需要在 Lambda 函数中解析这个字段，以提取客户端的真实 IP。

在 Lambda 函数中获取真实 IP 地址

接下来，我们需要在 AWS Lambda 函数中处理请求，并从中获取 `X-Forwarded-For` 头部的真实客户端 IP 地址。以下是一个简单的示例，演示如何在 Python Lambda 函数中实现这一点。

import json

def lambda_handler(event, context):
    # 从事件中获取 HTTP 请求头
    headers = event.get('headers')
    # 提取 X-Forwarded-For 头部
    x_forwarded_for = headers.get('X-Forwarded-For')
    # 解析出客户端 IP 地址
    if x_forwarded_for:
        client_ip = x_forwarded_for.split(',')[0]
    else:
        client_ip = "Unknown IP"
    response = {
        'statusCode': 200,
        'body': json.dumps({
            'client_ip': client_ip
        })
    }
    
    return response

测试与验证客户端 IP 的获取

完成 Lambda 函数的编写后，需要进行测试，以确保能够正确获取和返回客户端的 IP 地址。可以使用以下步骤进行测试：

步骤 1: 创建 ALB 和 Lambda

在 AWS 控制台中创建一个 Application Load Balancer，并将其配置为监听 HTTP 请求。设置 ALB 的目标为刚刚创建的 Lambda 函数。

步骤 2: 发送请求并查看响应

使用工具如 Postman 或 curl，向 ALB 发送 HTTP 请求。查看 Lambda 返回的响应，确保 `client_ip` 字段中显示的是发送请求的真实 IP 地址。

步骤 3: 日志记录

在 AWS Lambda 控制台中查看云日志，以确保 Lambda 函数能正常执行，没有错误发生，并且客户端 IP 地址被正确提取和记录。

总结

在 AWS Lambda 和 Application Load Balancer 的架构中，获取真实客户端 IP 地址并非自动完成。但是，通过 `X-Forwarded-For` 请求头，开发者仍可以轻松提取客户端的真实 IP。掌握这一方法，将有助于实现更为精准的用户行为分析和安全控制。