fix(data-migrations): transaction creation

ballerine-io · Aug 6, 2024 · 753fa8b · 753fa8b
1 parent 25540d5
commit 753fa8b
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 37 deletions.
diff --git a/services/workflows-service/scripts/alerts/generate-alerts.ts b/services/workflows-service/scripts/alerts/generate-alerts.ts
@@ -842,7 +842,7 @@ const generateFakeAlert = (
 };
 
 export const seedTransactionsAlerts = async (
- prisma: PrismaClient,
+ prisma: PrismaClient | PrismaTransaction,
  {
  project,
  businessIds,

diff --git a/services/workflows-service/scripts/alerts/generate-transactions.ts b/services/workflows-service/scripts/alerts/generate-transactions.ts
@@ -14,6 +14,7 @@ import {
 } from '@prisma/client';
 import { faker } from '@faker-js/faker';
 import { generateBusiness, generateEndUser } from '../generate-end-user';
+import { PrismaTransaction } from '@/types';
 
 const PaymentChannel = {
  online: 'online',
@@ -24,55 +25,54 @@ const PaymentChannel = {
 };
 
 export const generateTransactions = async (
- prismaClient: PrismaClient,
+ prismaClient: PrismaClient | PrismaTransaction,
  {
  projectId,
  }: {
  projectId: string;
  },
 ) => {
  // Create counterparties and collect their IDs
- const counterpartyIds = await prismaClient.$transaction(async prisma => {
- const businessCounterparties: string[] = [];
- const endUserCounterparties: string[] = [];
 
- for (let i = 0; i < 100; i++) {
- const correlationId = faker.datatype.uuid();
- const counterparty = await prisma.counterparty.create({
- data: {
- correlationId: correlationId,
- project: { connect: { id: projectId } },
- business: {
- create: generateBusiness({
- correlationId,
- projectId,
- }),
- },
+ const businessCounterparties: string[] = [];
+ const endUserCounterparties: string[] = [];
+
+ for (let i = 0; i < 100; i++) {
+ const correlationId = faker.datatype.uuid();
+ const counterparty = await prismaClient.counterparty.create({
+ data: {
+ correlationId: correlationId,
+ project: { connect: { id: projectId } },
+ business: {
+ create: generateBusiness({
+ correlationId,
+ projectId,
+ }),
  },
- });
+ },
+ });
 
- businessCounterparties.push(counterparty.id);
- }
- for (let i = 0; i < 100; i++) {
- const correlationId = faker.datatype.uuid();
- const counterparty = await prisma.counterparty.create({
- data: {
- correlationId: correlationId,
- project: { connect: { id: projectId } },
- endUser: {
- create: generateEndUser({
- correlationId,
- projectId,
- }),
- },
+ businessCounterparties.push(counterparty.id);
+ }
+ for (let i = 0; i < 100; i++) {
+ const correlationId = faker.datatype.uuid();
+ const counterparty = await prismaClient.counterparty.create({
+ data: {
+ correlationId: correlationId,
+ project: { connect: { id: projectId } },
+ endUser: {
+ create: generateEndUser({
+ correlationId,
+ projectId,
+ }),
  },
- });
+ },
+ });
 
-  endUserCounterparties.push(counterparty.id);
-  }
+ endUserCounterparties.push(counterparty.id);
+ }
 
- return { endUserCounterparties, businessCounterparties };
- });
+ const counterpartyIds = { endUserCounterparties, businessCounterparties };
 
  const ids: Array<{
  counterpartyOriginatorId?: string;

diff --git a/services/workflows-service/src/data-migration/scripts/migrate.ts b/services/workflows-service/src/data-migration/scripts/migrate.ts
@@ -190,6 +190,7 @@ const main = async () => {
  }, 2000);
  } catch (error: unknown) {
  logger.error('Error during running migration', { error });
+ console.error(error);
 
  if (error instanceof Error || typeof error === 'string') {
  sentryService.captureException(error);