wip

ldbc · Sep 19, 2024 · 0a554c0 · 0a554c0
1 parent a08d427
commit 0a554c0
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 15 deletions.
diff --git a/...agen/generation/events/LoanSubEvents.java → ...neration/events/LoanActivitiesEvents.java b/...agen/generation/events/LoanSubEvents.java → ...neration/events/LoanActivitiesEvents.java
@@ -19,7 +19,7 @@
 import ldbc.finbench.datagen.generation.DatagenParams;
 import ldbc.finbench.datagen.util.RandomGeneratorFarm;
 
-public class LoanSubEvents implements Serializable {
+public class LoanActivitiesEvents implements Serializable {
     private final RandomGeneratorFarm randomFarm;
     private final Random indexRandom;
     private final Random actionRandom;
@@ -29,7 +29,7 @@ public class LoanSubEvents implements Serializable {
     private final Map<String, AtomicLong> multiplicityMap;
     private List<Account> targetAccounts;
 
-    public LoanSubEvents() {
+    public LoanActivitiesEvents() {
         multiplicityMap = new ConcurrentHashMap<>();
         randomFarm = new RandomGeneratorFarm();
         indexRandom = new Random(DatagenParams.defaultSeed);

diff --git a/src/main/scala/ldbc/finbench/datagen/generation/generators/ActivityGenerator.scala b/src/main/scala/ldbc/finbench/datagen/generation/generators/ActivityGenerator.scala
@@ -175,18 +175,16 @@ class ActivityGenerator()(implicit spark: SparkSession)
         .toList
     )
 
-    val loanSubEvents = new LoanSubEvents
-    val afterLoanActions = loanRDD
-      .mapPartitionsWithIndex((index, loans) => {
-        loanSubEvents
-          .afterLoanApplied(
-            loans.toList.asJava,
-            sampledAccounts.value.asJava,
-            index
-          )
-          .iterator()
-          .asScala
-      })
-    afterLoanActions
+    loanRDD.mapPartitionsWithIndex((index, loans) => {
+      val loanSubEvents = new LoanActivitiesEvents
+      loanSubEvents
+        .afterLoanApplied(
+          loans.toList.asJava,
+          sampledAccounts.value.asJava,
+          index
+        )
+        .iterator()
+        .asScala
+    })
   }
 }