咨询：AWS API Gateway场景下Lambda连接Postgres及维护连接池的方法

Lambda要通过SSH连接Postgres，核心是在Lambda内部建立SSH隧道，通过堡垒机中转访问私有子网内的Postgres，具体实现步骤如下：

• 准备堡垒机：部署一台EC2作为SSH堡垒机，确保它能访问Postgres所在的私有子网，同时Lambda的执行角色有访问堡垒机的权限（如果堡垒机在公有子网，需配置安全组允许Lambda的IP段访问22端口；如果在私有子网，Lambda要部署在同VPC的子网）。
• 依赖准备：Lambda需要安装SSH客户端库（如paramiko）和Postgres驱动（如psycopg2-binary），可以通过Lambda层打包这些依赖，或者直接包含在部署包中。
• 密钥管理：将SSH私钥存储在AWS Secrets Manager中，Lambda执行角色需具备读取Secrets Manager的权限，避免硬编码密钥。
• 代码实现示例：

import paramiko
import psycopg2
from sshtunnel import SSHTunnelForwarder
import boto3

def get_ssh_key():
    secrets_manager = boto3.client('secretsmanager')
    response = secrets_manager.get_secret_value(SecretId='ssh-private-key')
    return response['SecretString']

def lambda_handler(event, context):
    ssh_key = get_ssh_key()
    with SSHTunnelForwarder(
        ('堡垒机IP', 22),
        ssh_username='ec2-user',
        ssh_pkey=paramiko.RSAKey.from_private_key(ssh_key),
        remote_bind_address=('Postgres端点', 5432),
        local_bind_address=('127.0.0.1', 5432)
    ) as tunnel:
        conn = psycopg2.connect(
            dbname='your-db-name',
            user='db-user',
            password='db-password',
            host='127.0.0.1',
            port=5432
        )
        # 执行数据库操作
        cursor = conn.cursor()
        cursor.execute('SELECT 1')
        result = cursor.fetchone()
        conn.close()
        return {'result': result}

• 注意事项：Lambda的执行超时需设置合理值（最长15分钟），避免SSH隧道和数据库操作超时；堡垒机的安全组要严格限制访问源，只允许Lambda的VPC CIDR访问。

API Gateway本身是无状态的请求路由服务，不具备维护数据库连接池的能力，需结合Lambda和其他AWS服务实现连接池管理，可行方案如下：

• 使用RDS Proxy（推荐）：AWS托管的数据库连接池服务，专门适配Lambda的无状态特性。Lambda通过RDS Proxy访问Postgres，RDS Proxy会自动管理连接复用、处理连接失效、控制并发连接数，无需在Lambda或API Gateway中维护连接池。API Gateway只需将请求转发到Lambda，Lambda直接通过RDS Proxy的端点连接数据库即可。
• Lambda全局范围初始化连接池：将数据库连接池的初始化代码放在Lambda函数的全局代码块中（而非lambda_handler内部），这样当Lambda实例被复用时，连接池会被保留，减少冷启动时的连接开销。但需处理连接失效的情况，比如每次请求前检查连接是否可用，不可用则重新初始化连接池。示例：

import psycopg2
from psycopg2 import pool

# 全局初始化连接池
connection_pool = psycopg2.pool.SimpleConnectionPool(
    minconn=1,
    maxconn=10,
    dbname='your-db-name',
    user='db-user',
    password='db-password',
    host='postgres-endpoint'
)

def lambda_handler(event, context):
    conn = None
    try:
        conn = connection_pool.getconn()
        cursor = conn.cursor()
        cursor.execute('SELECT 1')
        result = cursor.fetchone()
        return {'result': result}
    except psycopg2.OperationalError:
        # 连接失效，重新初始化连接池
        global connection_pool
        connection_pool.closeall()
        connection_pool = psycopg2.pool.SimpleConnectionPool(
            minconn=1,
            maxconn=10,
            dbname='your-db-name',
            user='db-user',
            password='db-password',
            host='postgres-endpoint'
        )
        # 重新获取连接执行操作
        conn = connection_pool.getconn()
        cursor = conn.cursor()
        cursor.execute('SELECT 1')
        result = cursor.fetchone()
        return {'result': result}
    finally:
        if conn:
            connection_pool.putconn(conn)

• Lambda容器镜像部署：将FastAPI应用打包成容器镜像部署到Lambda，在FastAPI应用内部维护连接池（比如用SQLAlchemy的QueuePool）。Lambda容器实例被复用时，应用的连接池会保留，但Lambda的实例复用机制不保证连接池一直有效，仍需处理连接失效的逻辑。

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