collect opend database and close inside decorator in parallel

assuncaocharles · Dec 3, 2024 · d0df0cc · d0df0cc
1 parent 7b54be2
commit d0df0cc
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/projects/ngx-indexed-db/src/lib/decorators.ts b/projects/ngx-indexed-db/src/lib/decorators.ts
@@ -1,6 +1,6 @@
 import { Observable } from 'rxjs';
 import { finalize } from 'rxjs/operators';
-import { closeDatabase } from './ngx-indexed-db';
+import { closeDatabase, openedDatabases } from './ngx-indexed-db';
 
 export function CloseDbConnection(): MethodDecorator {
   return function (_target: any, _propertyKey: string, descriptor: PropertyDescriptor): PropertyDescriptor {
@@ -11,10 +11,12 @@ export function CloseDbConnection(): MethodDecorator {
       if (result instanceof Observable) {
         return result.pipe(
           finalize(async () => {
-            const db: IDBDatabase | null = this.getDatabaseInstance();
-            if (db) {
+            const promises = openedDatabases.map(async (db: IDBDatabase) => {
               await closeDatabase(db);
-            }
+              console.log('Database connection closed: ', db.name);
+            });
+            await Promise.all(promises);
+            openedDatabases.length = 0;
           })
         );
       }

diff --git a/projects/ngx-indexed-db/src/lib/ngx-indexed-db.ts b/projects/ngx-indexed-db/src/lib/ngx-indexed-db.ts
@@ -1,6 +1,8 @@
 import { ObjectStoreMeta } from './ngx-indexed-db.meta';
 import { Observable, Subscriber } from 'rxjs';
 
+export const openedDatabases: IDBDatabase[] = [];
+
 export function openDatabase(
   indexedDB: IDBFactory,
   dbName: string,
@@ -15,6 +17,7 @@ export function openDatabase(
     let db: IDBDatabase;
     request.onsuccess = (event: Event) => {
       db = request.result;
+      openedDatabases.push(db);
       resolve(db);
     };
     request.onerror = (event: Event) => {