28 lines
666 B
JavaScript
28 lines
666 B
JavaScript
const { Pool } = require('pg');
|
|
require('dotenv').config();
|
|
|
|
const pool = new Pool({
|
|
host: process.env.DB_HOST || 'localhost',
|
|
port: parseInt(process.env.DB_PORT || '5432'),
|
|
database: process.env.DB_NAME || 'company_finance',
|
|
user: process.env.DB_USER || 'postgres',
|
|
password: process.env.DB_PASSWORD,
|
|
max: 20,
|
|
idleTimeoutMillis: 30000,
|
|
connectionTimeoutMillis: 2000,
|
|
});
|
|
|
|
pool.on('connect', () => {
|
|
console.log('✅ PostgreSQL 数据库连接成功');
|
|
});
|
|
|
|
pool.on('error', (err) => {
|
|
console.error('❌ PostgreSQL 连接错误:', err);
|
|
process.exit(-1);
|
|
});
|
|
|
|
module.exports = {
|
|
query: (text, params) => pool.query(text, params),
|
|
pool,
|
|
};
|