aboutsummaryrefslogtreecommitdiff
path: root/src/lib/db.js
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib/db.js')
-rw-r--r--src/lib/db.js51
1 files changed, 51 insertions, 0 deletions
diff --git a/src/lib/db.js b/src/lib/db.js
new file mode 100644
index 0000000..d6adcdf
--- /dev/null
+++ b/src/lib/db.js
@@ -0,0 +1,51 @@
+import pg from 'pg';
+import { database } from '$lib/config';
+
+const { Pool } = pg;
+
+const pool = new Pool(database);
+
+export async function query(text, params) {
+ const client = await pool.connect();
+ try {
+ const result = await client.query(text, params);
+ return result;
+ } finally {
+ client.release();
+ }
+}
+
+export async function queryWithAbort(text, params, signal) {
+ if (signal?.aborted) {
+ throw new Error('Query aborted');
+ }
+
+ const client = await pool.connect();
+
+ const queryPromise = client.query(text, params);
+
+ if (signal) {
+ const abortPromise = new Promise((_, reject) => {
+ signal.addEventListener('abort', () => {
+ client.release();
+ reject(new Error('Query aborted'));
+ }, { once: true });
+ });
+
+ try {
+ const result = await Promise.race([queryPromise, abortPromise]);
+ return result;
+ } finally {
+ client.release();
+ }
+ }
+
+ try {
+ const result = await queryPromise;
+ return result;
+ } finally {
+ client.release();
+ }
+}
+
+export default pool; \ No newline at end of file