如何在Python Chalice中实现API Gateway与SQS的直接集成（无需Lambda函数）

要实现API Gateway直接将请求转发到SQS队列（跳过你的Lambda中转逻辑），你可以利用Chalice对AWS服务集成的原生支持。下面是具体的实现方案和代码示例：

核心思路 #

Chalice默认的@app.route会创建Lambda集成，但通过指定integration='aws'参数，你可以配置API Gateway直接与SQS服务集成，让请求绕过Lambda直接发送到SQS，同时通过IAM策略确保API Gateway拥有访问SQS的权限。

实现步骤与代码 #

1. 确认Chalice版本 #

确保你使用的是Chalice 1.12及以上版本，这个版本开始支持AWS服务集成的配置。

2. 编写Chalice应用代码 #

替换你原来的Lambda中转代码，改用直接集成配置：

import os
import json
from chalice import Chalice, IAMPolicy

app = Chalice(app_name='sqs-direct-api')

# 定义允许API Gateway向指定SQS队列发送消息的IAM策略
sqs_send_policy = IAMPolicy(
    Version='2012-10-17',
    Statement=[{
        'Effect': 'Allow',
        'Action': 'sqs:SendMessage',
        'Resource': f"arn:aws:sqs:{app.region}:{app.account_id}:{os.environ['INPUT_QUEUE_NAME']}"
    }]
)

# 配置直接集成到SQS的POST路由
@app.route(
    '/',
    methods=['POST'],
    integration='aws',
    integration_kwargs={
        # SQS集成的URI格式：arn:aws:apigateway:<区域>:sqs:path/<账号ID>/<队列名称>
        'uri': f"arn:aws:apigateway:{app.region}:sqs:path/{app.account_id}/{os.environ['INPUT_QUEUE_NAME']}",
        'http_method': 'POST',
        # 指定API Gateway用来访问SQS的IAM凭据（这里用Chalice生成的策略ARN）
        'credentials': sqs_send_policy.to_arn(),
        # 请求模板：将API Gateway的JSON请求体转换为SQS要求的表单格式
        'request_templates': {
            'application/json': json.dumps({
                'Action': 'SendMessage',
                'MessageBody': "$input.body"
            })
        },
        # 响应模板：格式化SQS返回的结果，返回给客户端
        'response_templates': {
            'application/json': json.dumps({
                'status': 'success',
                'message_id': "$input.path('$.MessageId')"
            })
        },
        # 设置请求头，匹配SQS API要求的Content-Type
        'request_parameters': {
            'integration.request.header.Content-Type': "'application/x-www-form-urlencoded'"
        }
    }
)
def index():
    # 注意：这个函数不会被执行，只是Chalice路由配置的占位符
    return {}

3. 配置环境变量 #

在.chalice/config.json中添加你的队列名称环境变量：

{
  "version": "2.0",
  "app_name": "sqs-direct-api",
  "environment_variables": {
    "INPUT_QUEUE_NAME": "你的SQS队列名称"
  }
}

4. 部署应用 #

运行部署命令：

chalice deploy

关键配置说明 #

• integration='aws'：告诉Chalice这是AWS服务集成，而非默认的Lambda集成。
• uri：指定API Gateway要集成的SQS资源地址，格式严格遵循AWS的规范。
• credentials：授予API Gateway访问SQS的权限，这里通过IAMPolicy动态生成策略，你也可以使用预先创建的IAM角色ARN（需确保角色信任API Gateway服务）。
• 请求模板：SQS的SendMessage API要求使用application/x-www-form-urlencoded格式，所以我们需要将JSON请求体转换为对应的表单参数。
• 响应模板：将SQS返回的原始响应（包含MessageId等信息）格式化为友好的JSON响应返回给客户端。

验证集成 #

部署完成后，你可以直接向API Gateway的POST端点发送JSON请求，请求会被直接转发到SQS队列，无需经过Lambda函数。

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