[Fix #2113] Support group containing different process instance ids

apache · Oct 14, 2024 · bc68df2 · bc68df2
1 parent da4cee6
commit bc68df2
Showing 1 changed file with 6 additions and 8 deletions.
diff --git a/...a-common/src/main/java/org/kie/kogito/index/jpa/storage/ProcessInstanceEntityStorage.java b/...a-common/src/main/java/org/kie/kogito/index/jpa/storage/ProcessInstanceEntityStorage.java
@@ -20,7 +20,8 @@
 
 import java.time.ZonedDateTime;
 import java.util.ArrayList;
-import java.util.Iterator;
+import java.util.HashMap;
+import java.util.Map;
 import java.util.Set;
 
 import org.kie.kogito.event.process.MultipleProcessInstanceDataEvent;
@@ -68,13 +69,10 @@ public ProcessInstanceEntityStorage(ProcessInstanceEntityRepository repository,
 
     @Override
     @Transactional
-    public void indexGroup(MultipleProcessInstanceDataEvent event) {
-        Iterator<ProcessInstanceDataEvent<?>> iter = event.getData().iterator();
-        ProcessInstanceDataEvent<?> firstEvent = iter.next();
-        ProcessInstanceEntity pi = findOrInit(firstEvent);
-        indexEvent(pi, firstEvent);
-        while (iter.hasNext()) {
-            indexEvent(pi, iter.next());
+    public void indexGroup(MultipleProcessInstanceDataEvent events) {
+        Map<String, ProcessInstanceEntity> piMap = new HashMap<>();
+        for (ProcessInstanceDataEvent<?> event : events.getData()) {
+            indexEvent(piMap.computeIfAbsent(event.getKogitoProcessInstanceId(), id -> findOrInit(event)), event);
         }
         repository.flush();
     }