From aa3accb6b394b2b4164061e6e962228ab8285ca6 Mon Sep 17 00:00:00 2001
From: Linnnus <linnnus@users.noreply.github.com>
Date: Sat, 22 Feb 2025 16:05:11 +0100
Subject: Move hooks into their own modules

---
 app/src/lib/server/db.ts | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

(limited to 'app/src/lib/server/db.ts')

diff --git a/app/src/lib/server/db.ts b/app/src/lib/server/db.ts
index db001b7..b35f853 100644
--- a/app/src/lib/server/db.ts
+++ b/app/src/lib/server/db.ts
@@ -1,5 +1,6 @@
 import pg from "pg";
 import { env } from "$env/dynamic/private";
+import type { Handle } from "@sveltejs/kit";
 
 const pool = new pg.Pool({
 	database: env.POSTGRES_DB || "postgres",
@@ -15,6 +16,13 @@ pool.on("error", (err, client) => {
 	console.error("Database error: ", err, client);
 });
 
-export function getDbClient(): Promise<pg.PoolClient> {
-	return pool.connect();
-}
+export const dbHandle = (async ({ event, resolve }) => {
+	const dbClient = await pool.connect();
+	event.locals.dbClient = dbClient;
+
+	try {
+		return await resolve(event);
+	} finally {
+		dbClient.release();
+	}
+}) satisfies Handle;
-- 
cgit v1.2.3