本文探讨了在 Node.js 环境下，针对每个用户拥有独立数据库的 MySQL 实例，如何高效地管理数据库连接。我们将分析 mysql Node.js 包提供的两种主要连接管理方式：使用单个连接池配合 changeUser 方法，以及使用 PoolCluster 为每个数据库创建独立的连接池。同时，我们还会讨论不使用连接池的场景，并提供基于表名添加数据库名前缀的连接方案，帮助开发者根据实际应用场景选择最佳的数据库连接管理策略。
在构建 API 或 Web 服务时，针对每个用户拥有独立数据库的 MySQL 实例，数据库连接管理是一个需要仔细考虑的问题。mysql Node.js 包提供了多种连接管理方式，本文将深入探讨这些方式，并分析其优缺点，帮助开发者选择最适合自己应用的方案。
1. 单个连接池配合 changeUser 方法
这种方式创建一个全局的连接池，所有用户共享这个连接池。当需要连接到特定用户的数据库时，使用 connection.changeUser() 方法切换数据库。
示例代码：
const mysql = require('mysql');
const pool = mysql.createPool({
connectionLimit: 10,
user: process.env.DB_USER,
password: process.env.DB_PASSWORD,
port: 3306, // default
host: process.env.SQL_INSTANCE_HOST,
});
async function connectToUserDatabase(userDatabase) {
const connection = await pool.getConnection();
try {
await new Promise((resolve, reject) => {
connection.changeUser({ database: userDatabase }, (err) => {
if (err) {
reject(new Error(`Could not connect to database: ${err}`));
} else {
resolve();
}
});
});
return connection;
} catch (error) {
connection.release();
throw error;
}
}
// 使用示例
async function queryUserDatabase(userDatabase, query) {
let connection;
try {
connection = await connectToUserDatabase(userDatabase);
const [rows, fields] = await connection.execute(query);
return rows;
} catch (error) {
console.error("Error querying database:", error);
throw error;
} finally {
if (connection) {
connection.release();
}
}
}
// 示例调用
queryUserDatabase('USER1', 'SELECT * FROM mytable')
.then(results => console.log(results))
.catch(err => console.error(err));
优点：
- 资源利用率高：所有用户共享一个连接池，可以减少连接资源的开销。
- 配置简单：只需要配置一个连接池。
缺点：
- 安全性：如果连接池配置错误，可能会导致用户访问到其他用户的数据库。
- 性能：频繁切换数据库可能会影响性能，尤其在高并发场景下。
2. PoolCluster 为每个数据库创建独立的连接池
PoolCluster 允许为不同的数据库创建独立的连接池，每个连接池拥有自己的配置。
示例代码：
const mysql = require('mysql');
const poolCluster = mysql.createPoolCluster();
poolCluster.add('USER1', {
connectionLimit: 10,
user: process.env.DB_USER,
password: process.env.DB_PASSWORD,
port: 3306,
host: process.env.SQL_INSTANCE_HOST,
database: 'USER1'
});
poolCluster.add('USER2', {
connectionLimit: 10,
user: process.env.DB_USER,
password: process.env.DB_PASSWORD,
port: 3306,
host: process.env.SQL_INSTANCE_HOST,
database: 'USER2'
});
// 使用示例
async function queryUserDatabase(userKey, query) {
return new Promise((resolve, reject) => {
poolCluster.getConnection(userKey, (err, connection) => {
if (err) {
reject(err);
return;
}
connection.query(query, (error, results, fields) => {
connection.release(); // Always release the connection!
if (error) {
reject(error);
return;
}
resolve(results);
});
});
});
}
// 示例调用
queryUserDatabase('USER1', 'SELECT * FROM mytable')
.then(results => console.log(results))
.catch(err => console.error(err));优点：
- 安全性：每个数据库拥有独立的连接池，可以避免用户访问到其他用户的数据库。
- 性能：避免了频繁切换数据库的开销，可以提高性能。
缺点：
- 资源开销大：每个数据库都需要创建一个连接池，会增加资源开销。
- 配置复杂：需要为每个数据库配置连接池。
3. 不使用连接池
如果用户对 API 的访问频率很低，可以考虑不使用连接池，每次请求都创建一个新的连接。
MVM mall 网上购物系统
下载
采用 php+mysql 数据库方式运行的强大网上商店系统，执行效率高速度快，支持多语言，模板和代码分离，轻松创建属于自己的个性化用户界面 v3.5更新： 1).进一步静态化了活动商品. 2).提供了一些重要UFT-8转换文件 3).修复了除了网银在线支付其它支付显示错误的问题. 4).修改了LOGO广告管理,增加LOGO链接后主页LOGO路径错误的问题 5).修改了公告无法发布的问题,可能是打压
示例代码：
const mysql = require('mysql');
async function queryUserDatabase(userDatabase, query) {
const connection = mysql.createConnection({
user: process.env.DB_USER,
password: process.env.DB_PASSWORD,
port: 3306,
host: process.env.SQL_INSTANCE_HOST,
database: userDatabase
});
return new Promise((resolve, reject) => {
connection.connect(err => {
if (err) {
reject(err);
return;
}
connection.query(query, (error, results, fields) => {
connection.end(); // Close the connection!
if (error) {
reject(error);
return;
}
resolve(results);
});
});
});
}
// 示例调用
queryUserDatabase('USER1', 'SELECT * FROM mytable')
.then(results => console.log(results))
.catch(err => console.error(err));优点：
- 简单：不需要配置连接池。
- 资源利用率：当用户访问频率低时，能避免连接池占用资源。
缺点：
- 性能：每次请求都需要创建和关闭连接，会增加延迟。
- 不适合高并发场景。
4. 使用表名前缀
如果可以使用相同的用户名/密码访问所有数据库，可以在查询时使用表名前缀来指定数据库。
示例代码：
const mysql = require('mysql');
const pool = mysql.createPool({
connectionLimit: 10,
user: process.env.DB_USER,
password: process.env.DB_PASSWORD,
port: 3306, // default
host: process.env.SQL_INSTANCE_HOST,
});
async function queryUserDatabase(userDatabase, query) {
const connection = await pool.getConnection();
try {
const prefixedQuery = query.replace(/tablename/g, `${userDatabase}.tablename`); // Replace tablename with database.tablename
const [rows, fields] = await connection.execute(prefixedQuery);
return rows;
} catch (error) {
console.error("Error querying database:", error);
throw error;
} finally {
connection.release();
}
}
// 示例调用
queryUserDatabase('USER1', 'SELECT * FROM tablename')
.then(results => console.log(results))
.catch(err => console.error(err));优点：
- 简单：只需要配置一个连接池。
- 性能：避免了切换数据库的开销。
缺点：
- 需要修改所有查询语句。
- 安全性：需要确保用户只能访问到自己的表。
总结
选择哪种连接管理方式取决于具体的应用场景。
- 如果用户访问频率高，且对性能要求较高，建议使用 PoolCluster。
- 如果用户访问频率不高，可以使用单个连接池配合 changeUser 方法，或者不使用连接池。
- 如果可以使用相同的用户名/密码访问所有数据库，可以使用表名前缀。
在选择连接管理方式时，还需要考虑安全性、资源开销和配置复杂性等因素。
注意事项：
- 在生产环境中，应该使用环境变量来配置数据库连接信息，避免将敏感信息硬编码到代码中。
- 应该对数据库连接进行错误处理，避免程序崩溃。
- 应该定期检查数据库连接池的连接状态，及时释放无效连接。
- 始终在使用完连接后释放连接，避免连接泄漏。
