在AWS API网关中，可以使用自定义域名将API网关中的请求映射到客户自己的域名。但是，在使用自定义域名时，AWS API网关默认会将主机头替换为自定义域名。这可能会导致一些问题，例如：

• 如果您的应用程序期望通过主机头来识别请求，则自定义域名可能会破坏这些期望。
• 一些客户端（例如浏览器）可能会发送HTTP Referrer头部，该头部包含之前访问页面的主机名。如果API网关替换了主机头，这将导致请求失败。

为了解决这个问题，您可以使用Lambda函数自定义请求和响应。 在请求中，您可以保留原始主机头，然后在响应中将其恢复。这样您就可以使用自定义域名，并且继续正常使用主机头。

示例代码：

1. 创建一个Lambda函数并添加以下代码。

import json

def lambda_handler(event, context):
    # Parse the request from API Gateway
    request = json.loads(event['body'])

    # Set the original host header to 'host' key in the request headers
    request['headers']['host'] = event['headers']['host']

    # Do some processing with the request here (if needed)

    # Return the modified request back to API Gateway
    return {
        'statusCode': 200,
        'headers': {},
        'body': json.dumps(request)
    }

2. 在API网关中创建Lambda集成连接，并将Lambda函数与API方法映射。指定Integration Request类型为“Lambda函数”。将Lambda集成设置为使用代理集成，并指定Lambda函数的ARN。找到“生成模型”部分，使用以下代码覆盖默认的主机头：

{
    "type": "object",
    "properties": {
        "host": {"type": "string"}
    }
}

3. 创建另一个Lambda函数，并将其作为API网关响应集成。将该函数与API方法映射。指定“Lambda函数”作为响应类型，并提供Lambda函数的ARN。

import json

def lambda_handler(event, context):
    # Parse the response from Lambda function
    response = json.loads(event['body'])

    # Set the original host header to the 'host' key in the response headers
    response_headers = response.get('headers', {})
    response_headers['host'] = response_headers.pop('Host', None)
    response['headers'] = response_headers

    # Do some processing with the response here (if needed)

    # Return the modified response back to API Gateway
    return {
        'statusCode': response['statusCode'],
        'headers': response.get('headers', {}),
        'body': json.dumps(response.get('body', {}))
    }

现在您已经成功地使用自定义域名，并保留了主机头，避免了由此带来的问题。同时，您的Lambda函数还能够处理传入的请求和响应，这将为您带来更多的灵活性和控制权。