Implement functionality for checkBefore

gchq · Feb 20, 2025 · 2f2b034 · 2f2b034
1 parent 695bda6
commit 2f2b034
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/java/core/src/main/java/sleeper/core/statestore/DelegatingStateStore.java b/java/core/src/main/java/sleeper/core/statestore/DelegatingStateStore.java
@@ -190,11 +190,13 @@ public void fixPartitionUpdateTime(Instant now) {
 
     @Override
     public void addFilesTransaction(AddTransactionRequest request) {
+        request.getTransaction().checkBefore(this);
         fileReferenceStore.addFilesTransaction(request);
     }
 
     @Override
     public void addPartitionsTransaction(AddTransactionRequest request) {
+        request.getTransaction().checkBefore(this);
         partitionStore.addPartitionsTransaction(request);
     }
 }
diff --git a/...eper/core/statestore/transactionlog/transaction/impl/InitialisePartitionsTransaction.java b/...eper/core/statestore/transactionlog/transaction/impl/InitialisePartitionsTransaction.java
@@ -62,6 +62,9 @@ public void apply(StateStorePartitions stateStorePartitions, Instant updateTime)
 
     @Override
     public void checkBefore(StateStore stateStore) throws StateStoreException {
+        if (!stateStore.hasNoFiles()) {
+            throw new StateStoreException("Cannot initialise state store when files are present");
+        }
     }
 
     @Override