getStates moved to interface level

2 files changed, 8 insertions, 2 deletions

diff --git a/model-data/src/main/java/org/eclipse/viatra/solver/data/map/VersionedMapStore.java b/model-data/src/main/java/org/eclipse/viatra/solver/data/map/VersionedMapStore.java
index 60d869ec..0ff0773f 100644
--- a/model-data/src/main/java/org/eclipse/viatra/solver/data/map/VersionedMapStore.java
+++ b/model-data/src/main/java/org/eclipse/viatra/solver/data/map/VersionedMapStore.java
@@ -1,10 +1,14 @@
 package org.eclipse.viatra.solver.data.map;
 
+import java.util.Set;
+
 public interface VersionedMapStore<K, V> {
         
         public VersionedMap<K, V> createMap();
 
         public VersionedMap<K, V> createMap(long state);
+        
+        public Set<Long> getStates();
 
         public DiffCursor<K,V> getDiffCursor(long fromState, long toState);
 }
\ No newline at end of file
diff --git a/model-data/src/main/java/org/eclipse/viatra/solver/data/map/VersionedMapStoreImpl.java b/model-data/src/main/java/org/eclipse/viatra/solver/data/map/VersionedMapStoreImpl.java
index ab55e4bc..9ee4cfdf 100644
--- a/model-data/src/main/java/org/eclipse/viatra/solver/data/map/VersionedMapStoreImpl.java
+++ b/model-data/src/main/java/org/eclipse/viatra/solver/data/map/VersionedMapStoreImpl.java
@@ -76,8 +76,9 @@ public class VersionedMapStoreImpl<K, V> implements VersionedMapStore<K, V> {
                         ContinousHashProvider<K> hashProvider, V defaultValue) {
                 return createSharedVersionedMapStores(amount, hashProvider, defaultValue, new VersionedMapStoreConfiguration());
         }
-
-        synchronized Set<Long> getStates() {
+        
+        @Override
+        public synchronized Set<Long> getStates() {
                 return states.keySet();
         }
 
@@ -91,6 +92,7 @@ public class VersionedMapStoreImpl<K, V> implements VersionedMapStore<K, V> {
                 ImmutableNode<K, V> data = revert(state);
                 return new VersionedMapImpl<>(this, hashProvider, defaultValue, data);
         }
+        
 
         public synchronized ImmutableNode<K, V> revert(long state) {
                 if (states.containsKey(state)) {