Return value from the TypeOrm transaction callback (#2096)

safe-global · Nov 6, 2024 · cea811a · cea811a
1 parent a1c7f31
commit cea811a
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/datasources/db/v2/postgres-database.service.ts b/src/datasources/db/v2/postgres-database.service.ts
@@ -84,13 +84,13 @@ export class PostgresDatabaseService {
     return this.dataSource.getRepository<T>(entity);
   }
 
-  public async transaction(
-    callback: (transactionManager: EntityManager) => Promise<void>,
-  ): Promise<void> {
+  public async transaction<T>(
+    callback: (transactionManager: EntityManager) => Promise<T>,
+  ): Promise<T> {
     return this.dataSource.transaction(
-      async (transactionalEntityManager): Promise<void> => {
+      async (transactionalEntityManager): Promise<T> => {
         this.transactionManager = transactionalEntityManager;
-        await callback(this.transactionManager);
+        return await callback(this.transactionManager);
       },
     );
   }