如何在单个AWS Lambda函数中连接多个RDS数据库Schema

针对你的场景，这里提供三种可行的实现方式，根据业务需求选择即可：

方法1：维护多套连接配置，创建独立连接实例 #

适合需要同时操作两个Schema，或两个Schema对应不同权限用户的场景。

• 修改config.py，将原有的单个配置扩展为包含多个Schema的字典：

# config.py
dbinfos = {
    'game1': {
        'db': 'your_db_name',
        'user': 'your_db_user',
        'passwd': 'your_db_password',
        'schema': 'game1'
    },
    'game2': {
        'db': 'your_db_name',
        'user': 'your_db_user',
        'passwd': 'your_db_password',
        'schema': 'game2'
    }
}

• 在Lambda业务代码中，封装获取连接的方法，按需初始化对应Schema的连接：

import mysql.connector  # 请根据实际使用的数据库驱动调整（如psycopg2用于PostgreSQL）
from config import dbinfos

def get_db_connection(schema_key):
    config = dbinfos[schema_key]
    conn = mysql.connector.connect(
        host='your_rds_endpoint',
        database=config['db'],
        user=config['user'],
        password=config['passwd']
    )
    # 可选：设置当前连接默认使用指定Schema
    with conn.cursor() as cursor:
        cursor.execute(f"USE {config['schema']}")
    return conn

# 调用示例
# 操作game1 Schema
game1_conn = get_db_connection('game1')
with game1_conn.cursor() as cursor:
    cursor.callproc('your_stored_procedure', args=(param1, param2))
game1_conn.commit()
game1_conn.close()

# 操作game2 Schema
game2_conn = get_db_connection('game2')
with game2_conn.cursor() as cursor:
    cursor.callproc('your_stored_procedure', args=(param1, param2))
game2_conn.commit()
game2_conn.close()

方法2：复用单个连接，调用存储过程时指定Schema前缀 #

如果两个Schema属于同一个RDS实例，且当前数据库用户拥有访问两者的权限，可以复用单个连接，直接通过Schema前缀调用存储过程。

• 简化config.py配置（无需指定Schema）：

# config.py
dbinfo = {
    'db': 'your_db_name',
    'user': 'your_db_user',
    'passwd': 'your_db_password'
}

• 业务代码中创建单个连接，调用存储过程时明确指定Schema：

import mysql.connector
from config import dbinfo

# 初始化单个数据库连接
conn = mysql.connector.connect(
    host='your_rds_endpoint',
    database=dbinfo['db'],
    user=dbinfo['user'],
    password=dbinfo['passwd']
)

# 调用game1的存储过程
with conn.cursor() as cursor:
    cursor.callproc('game1.your_stored_procedure', args=(param1, param2))
conn.commit()

# 调用game2的存储过程
with conn.cursor() as cursor:
    cursor.callproc('game2.your_stored_procedure', args=(param1, param2))
conn.commit()

conn.close()

这种方式能减少连接数，更适合Lambda的连接复用优化（注意Lambda的连接生命周期，避免冷启动带来的资源浪费）。

方法3：动态切换当前连接的默认Schema #

部分数据库支持动态切换当前连接的默认Schema（如MySQL的USE语句、PostgreSQL的SET search_path），适合按顺序操作两个Schema的场景。

• 业务代码示例（以MySQL为例）：

import mysql.connector
from config import dbinfo

conn = mysql.connector.connect(
    host='your_rds_endpoint',
    database=dbinfo['db'],
    user=dbinfo['user'],
    password=dbinfo['passwd']
)

with conn.cursor() as cursor:
    # 切换到game1并调用存储过程
    cursor.execute("USE game1")
    cursor.callproc('your_stored_procedure', args=(param1, param2))
    
    # 切换到game2并调用存储过程
    cursor.execute("USE game2")
    cursor.callproc('your_stored_procedure', args=(param1, param2))

conn.commit()
conn.close()

注意：这种方式是连接级别的设置，在多线程并发场景下需谨慎使用，避免Schema切换导致的操作混乱。

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