数据库扩展双十二活动

import mysql.connector
from mysql.connector import pooling

# 创建主库连接池
master_config = {
    "host": "master_host",
    "user": "master_user",
    "password": "master_password",
    "database": "mydatabase"
}
master_pool = mysql.connector.pooling.MySQLConnectionPool(pool_name="master_pool", pool_size=5, **master_config)

# 创建从库连接池
slave_config = {
    "host": "slave_host",
    "user": "slave_user",
    "password": "slave_password",
    "database": "mydatabase"
}
slave_pool = mysql.connector.pooling.MySQLConnectionPool(pool_name="slave_pool", pool_size=5, **slave_config)

def execute_query(query, params=None, use_slave=False):
    if use_slave:
        connection = slave_pool.get_connection()
    else:
        connection = master_pool.get_connection()
    
    cursor = connection.cursor()
    cursor.execute(query, params)
    result = cursor.fetchall()
    cursor.close()
    connection.close()
    return result

# 示例：执行写操作（使用主库）
execute_query("INSERT INTO users (name, email) VALUES (%s, %s)", ("John Doe", "john@example.com"))

# 示例：执行读操作（使用从库）
users = execute_query("SELECT * FROM users", use_slave=True)
print(users)