Default method for new check of state before

gchq · Feb 20, 2025 · 5de500d · 5de500d
1 parent 1ee06a0
commit 5de500d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/...c/main/java/sleeper/core/statestore/transactionlog/transaction/StateStoreTransaction.java b/...c/main/java/sleeper/core/statestore/transactionlog/transaction/StateStoreTransaction.java
@@ -15,6 +15,7 @@
  */
 package sleeper.core.statestore.transactionlog.transaction;
 
+import sleeper.core.statestore.StateStore;
 import sleeper.core.statestore.StateStoreException;
 
 import java.time.Instant;
@@ -41,4 +42,12 @@ public interface StateStoreTransaction<T> {
      * @param updateTime the time that the update was added to the state store
      */
     void apply(T state, Instant updateTime);
+
+    /**
+     * Checks the present state of the state store.
+     *
+     * @param stateStore the state store
+     */
+    default void checkBefore(StateStore stateStore) throws StateStoreException {
+    }
 }
diff --git a/...eper/core/statestore/transactionlog/transaction/impl/InitialisePartitionsTransaction.java b/...eper/core/statestore/transactionlog/transaction/impl/InitialisePartitionsTransaction.java
@@ -18,6 +18,8 @@
 import sleeper.core.partition.Partition;
 import sleeper.core.partition.PartitionsFromSplitPoints;
 import sleeper.core.schema.Schema;
+import sleeper.core.statestore.StateStore;
+import sleeper.core.statestore.StateStoreException;
 import sleeper.core.statestore.transactionlog.state.StateStorePartitions;
 import sleeper.core.statestore.transactionlog.transaction.PartitionTransaction;
 
@@ -79,4 +81,9 @@ public boolean equals(Object obj) {
     public String toString() {
         return "InitialisePartitionsTransaction{partitions=" + partitions + "}";
     }
+
+    @Override
+    public void checkBefore(StateStore stateStore) throws StateStoreException {
+    }
+
 }