diff --git a/src/server/db/index.ts b/src/server/db/index.ts index fc8e6be..21201f1 100644 --- a/src/server/db/index.ts +++ b/src/server/db/index.ts @@ -10,11 +10,16 @@ export const createDB = () => export type DB = ReturnType -export const getDB = (() => { - let db: DB | null = null +let _db: DB | null = null - return (): DB => { - db ??= createDB() - return db +export const getDB = (): DB => { + _db ??= createDB() + return _db +} + +export const closeDB = async () => { + if (_db) { + await _db.$client.end() + _db = null } -})() +}