Fixed bug in the extraction of blocking keys.

scify · Mar 8, 2018 · b602d84 · b602d84
1 parent 65f16da
commit b602d84
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/jedai-core/src/main/java/org/scify/jedai/blockbuilding/AbstractBlockBuilding.java b/jedai-core/src/main/java/org/scify/jedai/blockbuilding/AbstractBlockBuilding.java
@@ -28,6 +28,7 @@
 
 import java.util.ArrayList;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
 import java.util.Set;
@@ -108,19 +109,24 @@ public double getTotalNoOfEntities() {
     protected void indexEntities(Map<String, TIntList> index, List<EntityProfile> entities) {
         int counter = 0;
         for (EntityProfile profile : entities) {
+            final Set<String> allKeys = new HashSet<>();
             for (Attribute attribute : profile.getAttributes()) {
                 for (String key : getBlockingKeys(attribute.getValue())) {
                     String normalizedKey = key.trim().toLowerCase();
                     if (0 < normalizedKey.length()) {
-                        TIntList entityList = index.get(normalizedKey);
-                        if (entityList == null) {
-                            entityList = new TIntArrayList();
-                            index.put(normalizedKey, entityList);
-                        }
-                        entityList.add(counter);
+                        allKeys.add(normalizedKey);
                     }
                 }
             }
+
+            for (String key : allKeys) {
+                TIntList entityList = index.get(key);
+                if (entityList == null) {
+                    entityList = new TIntArrayList();
+                    index.put(key, entityList);
+                }
+                entityList.add(counter);
+            }
             counter++;
         }
     }