38 lines
792 B
JavaScript
38 lines
792 B
JavaScript
const { Pool } = require('pg');
|
|
require('dotenv').config();
|
|
|
|
const pool = new Pool({
|
|
host: process.env.PGHOST,
|
|
user: process.env.PGUSER,
|
|
database: process.env.PGDATABASE,
|
|
password: process.env.PGPASSWORD,
|
|
port: process.env.PGPORT,
|
|
});
|
|
|
|
async function connectDB() {
|
|
try {
|
|
await pool.connect();
|
|
console.log('Database connected.')
|
|
} catch (error) {
|
|
console.error('Failed to connect to the database: ',error)
|
|
process.exit(1)
|
|
}
|
|
}
|
|
|
|
function closeDB() {
|
|
try {
|
|
pool.end();
|
|
console.log('Database connection closed.')
|
|
} catch (error) {
|
|
console.error('Error closing the database: ', error )
|
|
}
|
|
}
|
|
|
|
|
|
|
|
module.exports = {
|
|
query: (text, params) => pool.query(text, params),
|
|
closeDB,
|
|
connectDB
|
|
};
|