Super duper extra close the database

packages/db/DittoDatabase.ts

@@ -2,7 +2,7 @@ import type { Kysely } from 'kysely';
 
 import type { DittoTables } from './DittoTables.ts';
 
-export interface DittoDatabase {
+export interface DittoDatabase extends AsyncDisposable {
   readonly kysely: Kysely<DittoTables>;
   readonly poolSize: number;
   readonly availableConnections: number;

packages/db/adapters/DittoPglite.ts

@@ -36,6 +36,10 @@ export class DittoPglite {
       poolSize: 1,
       availableConnections: 1,
       listen,
+      [Symbol.asyncDispose]: async () => {
+        await pglite.close();
+        await kysely.destroy();
+      },
     };
   }
 }

packages/db/adapters/DittoPostgres.ts

@@ -54,6 +54,10 @@ export class DittoPostgres {
         return pg.connections.idle;
       },
       listen,
+      [Symbol.asyncDispose]: async () => {
+        await pg.end();
+        await kysely.destroy();
+      },
     };
   }
 }

packages/ditto/test.ts

@@ -38,7 +38,7 @@ export async function createTestDB(opts?: { pure?: boolean }) {
         await sql`truncate table ${sql.ref(tablename)} cascade`.execute(db.kysely);
       }
 
-      await db.kysely.destroy();
+      await db[Symbol.asyncDispose]();
     },
   };
 }