Remove Postgres database service getTransactionRunner (#2097)

safe-global · Nov 7, 2024 · e43375b · e43375b
1 parent 40e72aa
commit e43375b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 14 deletions.
diff --git a/src/datasources/db/v2/__tests__/postgresql-database.service.mock.ts b/src/datasources/db/v2/__tests__/postgresql-database.service.mock.ts
@@ -18,5 +18,4 @@ export const mockPostgresDatabaseService = {
         return callback(mockEntityManager);
       },
     ),
-  getTransactionRunner: jest.fn().mockReturnValue(mockEntityManager),
 } as jest.MockedObjectDeep<PostgresDatabaseService>;
diff --git a/src/datasources/db/v2/postgres-database.service.ts b/src/datasources/db/v2/postgres-database.service.ts
@@ -13,8 +13,6 @@ import {
 
 @Injectable()
 export class PostgresDatabaseService {
-  private transactionManager?: EntityManager = undefined;
-
   public constructor(
     @Inject(LoggingService) private readonly loggingService: ILoggingService,
     @InjectDataSource()
@@ -88,18 +86,9 @@ export class PostgresDatabaseService {
     callback: (transactionManager: EntityManager) => Promise<T>,
   ): Promise<T> {
     return this.dataSource.transaction(
-      async (transactionalEntityManager): Promise<T> => {
-        this.transactionManager = transactionalEntityManager;
-        return await callback(this.transactionManager);
+      async (transactionalEntityManager: EntityManager): Promise<T> => {
+        return await callback(transactionalEntityManager);
       },
     );
   }
-
-  public getTransactionRunner(): EntityManager {
-    if (!this.transactionManager) {
-      throw new Error('Query runner is not initialized...');
-    }
-
-    return this.transactionManager;
-  }
 }