stringadmin
35 lines
892 B
JavaScript
35 lines
892 B
JavaScript
const { Pool } = require("pg");
|
|
require("dotenv").config();
|
|
|
|
const pool = new Pool({
|
|
host: process.env.PG_HOST || "localhost",
|
|
port: Number(process.env.PG_PORT) || 5432,
|
|
database: process.env.PG_DATABASE || "omniai",
|
|
user: process.env.PG_USER || "omniai",
|
|
password: process.env.PG_PASSWORD || "",
|
|
max: Number(process.env.PG_POOL_MAX) || 10,
|
|
idleTimeoutMillis: 30000,
|
|
connectionTimeoutMillis: 5000,
|
|
});
|
|
|
|
pool.on("error", (err) => {
|
|
console.error("[db] Unexpected pool error:", err.message);
|
|
});
|
|
|
|
async function withTransaction(fn) {
|
|
const client = await pool.connect();
|
|
try {
|
|
await client.query("BEGIN");
|
|
const result = await fn(client);
|
|
await client.query("COMMIT");
|
|
return result;
|
|
} catch (err) {
|
|
await client.query("ROLLBACK").catch(() => {});
|
|
throw err;
|
|
} finally {
|
|
client.release();
|
|
}
|
|
}
|
|
|
|
module.exports = { pool, withTransaction };
|