Eliminate duplicate email addresses. Fixes #388

ldbc · May 3, 2022 · 59d0032 · 59d0032
1 parent 2131160
commit 59d0032
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/main/java/ldbc/snb/datagen/generator/generators/PersonGenerator.java b/src/main/java/ldbc/snb/datagen/generator/generators/PersonGenerator.java
@@ -151,11 +151,14 @@ private Person generatePerson() {
         base = base.replaceAll("\\p{InCombiningDiacriticalMarks}+", "");
         base = base.replaceAll(" ", ".");
         base = base.replaceAll("[.]+", ".");
-        for (int i = 0; i < numEmails; i++) {
+        while (person.getEmails().size() < numEmails) {
             String email = base + "" + person.getAccountId() + "@" +
                     Dictionaries.emails.getRandomEmail(randomFarm.get(RandomGeneratorFarm.Aspect.TOP_EMAIL),
                             randomFarm.get(RandomGeneratorFarm.Aspect.EMAIL));
-            person.getEmails().add(email);
+            // avoid duplicates
+            if (!person.getEmails().contains(email)) {
+                person.getEmails().add(email);
+            }
         }
 
         // Set class year