封装nodejs使用mysql2的连接池写法
时间: 2023-07-21 11:03:13 浏览: 273
以下是封装nodejs使用mysql2的连接池的写法:
```javascript
const mysql = require('mysql2/promise');
class MysqlPool {
constructor(config) {
this.pool = mysql.createPool(config);
}
async execute(sql, values) {
const connection = await this.pool.getConnection();
try {
const [rows] = await connection.execute(sql, values);
return rows;
} catch (err) {
throw err;
} finally {
connection.release();
}
}
async beginTransaction() {
const connection = await this.pool.getConnection();
try {
await connection.beginTransaction();
return connection;
} catch (err) {
connection.release();
throw err;
}
}
async commitTransaction(connection) {
try {
await connection.commit();
connection.release();
} catch (err) {
await connection.rollback();
connection.release();
throw err;
}
}
async rollbackTransaction(connection) {
try {
await connection.rollback();
connection.release();
} catch (err) {
connection.release();
throw err;
}
}
}
module.exports = MysqlPool;
```
在上面的代码中,我们使用了 `mysql2/promise` 模块创建了一个连接池,然后封装了一些常用的方法,包括 `execute`、 `beginTransaction`、 `commitTransaction` 和 `rollbackTransaction`。其中,`execute` 方法用于执行 SQL 语句,`beginTransaction` 方法用于开启事务,`commitTransaction` 方法用于提交事务,`rollbackTransaction` 方法用于回滚事务。
使用时,只需要实例化 `MysqlPool` 类,然后调用相应的方法即可。例如:
```javascript
const MysqlPool = require('./mysql-pool');
const config = {
host: 'localhost',
user: 'root',
password: '123456',
database: 'test',
waitForConnections: true,
connectionLimit: 10,
queueLimit: 0,
};
const pool = new MysqlPool(config);
async function getUsers() {
const sql = 'SELECT * FROM users';
const result = await pool.execute(sql);
return result;
}
getUsers().then((result) => {
console.log(result);
}).catch((err) => {
console.error(err);
});
```
上面的代码中,我们首先创建了一个 `config` 对象,包含了数据库连接的配置信息。然后实例化了一个 `MysqlPool` 对象,并且调用了 `execute` 方法执行了一条 SQL 查询语句。最后输出查询结果或者错误信息。
阅读全文