面试题：MySQL中如何实现多数据库连接及动态选择数据库

Question

假设你有两个MySQL数据库，`db1`和`db2`，请描述如何在你的应用程序（语言不限，以Python为例）中实现对这两个数据库的连接，并且根据业务逻辑动态选择使用其中一个数据库进行查询操作。阐述连接的建立过程以及选择数据库的SQL语句。

Answer 1

1. 连接数据库：
   • 在Python中可以使用pymysql库来连接MySQL数据库。首先安装pymysql库：pip install pymysql。
   • 连接db1和db2的代码示例如下：

import pymysql

# 连接db1
conn1 = pymysql.connect(
    host='your_host',
    user='your_user',
    password='your_password',
    database='db1',
    port=3306
)

# 连接db2
conn2 = pymysql.connect(
    host='your_host',
    user='your_user',
    password='your_password',
    database='db2',
    port=3306
)

2. 动态选择数据库进行查询：
   • 假设根据某个业务逻辑标志use_db1来决定使用哪个数据库。如果use_db1为True，则使用db1；否则使用db2。
   • 示例代码如下：

use_db1 = True  # 这里根据实际业务逻辑决定
if use_db1:
    conn = conn1
else:
    conn = conn2

try:
    with conn.cursor() as cursor:
        # 选择数据库的SQL语句，这里在连接时已经指定了数据库，无需再次选择
        # 如果在连接时未指定数据库，可使用如下SQL语句选择数据库
        # cursor.execute('USE db1' if use_db1 else 'USE db2')
        query = "SELECT * FROM your_table"
        cursor.execute(query)
        result = cursor.fetchall()
        for row in result:
            print(row)
finally:
    conn.close()

• 在这个示例中，连接数据库时已经在pymysql.connect方法中指定了数据库名。如果连接时未指定数据库，可以使用USE语句来选择数据库，如注释部分代码所示。根据业务逻辑决定使用哪个连接对象conn，然后使用该连接对象进行查询操作。最后，操作完成后关闭连接。