21 lines
557 B
JavaScript
21 lines
557 B
JavaScript
const { Pool } = require('pg');
|
|
require('dotenv').config();
|
|
|
|
const pool = new Pool({
|
|
user: process.env.DB_USER,
|
|
host: process.env.DB_HOST,
|
|
database: process.env.DB_DATABASE || process.env.DB_NAME,
|
|
password: String(process.env.DB_PASSWORD || ''),
|
|
port: process.env.DB_PORT,
|
|
});
|
|
|
|
// 监听连接池错误，防止空闲客户端错误导致程序崩溃
|
|
pool.on('error', (err, client) => {
|
|
console.error('Unexpected error on idle client', err);
|
|
});
|
|
|
|
module.exports = {
|
|
query: (text, params) => pool.query(text, params),
|
|
pool
|
|
};
|