init uniqueObjectTypesInStore

apache · Aug 4, 2023 · 102c2ef · 102c2ef
1 parent 42a8db3
commit 102c2ef
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.../src/main/java/org/drools/reliability/core/SimpleSerializationReliableRefObjectStore.java b/.../src/main/java/org/drools/reliability/core/SimpleSerializationReliableRefObjectStore.java
@@ -32,6 +32,9 @@ public class SimpleSerializationReliableRefObjectStore extends SimpleSerializati
     public SimpleSerializationReliableRefObjectStore(Storage<Long, StoredObject> storage) {
         super(storage);
         uniqueObjectTypesInStore = new HashMap<>();
+        if (storage.size()>0){
+            storage.values().forEach(sObject -> {this.putIntoObjectTypesList( sObject.getObject() );});
+        }
         this.storage = storage.size()>0 ? updateObjectReferences(storage) : storage;
     }
 
@@ -100,7 +103,7 @@ private List<Field> getReferencedObjects(Object object){
         Field[] fields = object.getClass().getDeclaredFields();
 
         List<Field> fieldsWithTypeInTheStore = Arrays.stream(fields)
-                .filter(field -> uniqueObjectTypesInStore.keySet().contains(field.getType()))
+                .filter(field -> uniqueObjectTypesInStore.containsKey(field.getType()))
                 .collect(Collectors.toList());
         return fieldsWithTypeInTheStore;
     }