New method for Loading transaction or pointer

gchq · Jan 10, 2025 · 67fce2f · 67fce2f
1 parent c7eef11
commit 67fce2f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/java/core/src/main/java/sleeper/core/statestore/transactionlog/TransactionLogEntry.java b/java/core/src/main/java/sleeper/core/statestore/transactionlog/TransactionLogEntry.java
@@ -70,6 +70,14 @@ public Optional<StateStoreTransaction<?>> getTransaction() {
         return Optional.ofNullable(transaction);
     }
 
+    public StateStoreTransaction<?> getTransactionOrLoadFromPointer(TransactionBodyStore bodyStore) {
+        if (transaction != null) {
+            return transaction;
+        } else {
+            return bodyStore.getBody(bodyPointer);
+        }
+    }
+
     @Override
     public int hashCode() {
         return Objects.hash(transactionNumber, transaction);

diff --git a/java/core/src/main/java/sleeper/core/statestore/transactionlog/TransactionLogHead.java b/java/core/src/main/java/sleeper/core/statestore/transactionlog/TransactionLogHead.java
@@ -257,7 +257,7 @@ private void applyTransaction(TransactionLogEntry entry) {
                     entry.getTransactionType());
             return;
         }
-        transactionType.cast(entry.getTransaction().orElseThrow())
+        transactionType.cast(entry.getTransactionOrLoadFromPointer(bodyStore))
                 .apply(state, entry.getUpdateTime());
         lastTransactionNumber = entry.getTransactionNumber();
     }