Refactor use of state in TransactionLogPartitionStore

java/core/src/main/java/sleeper/core/statestore/transactionlog/TransactionLogPartitionStore.java

@@ -29,7 +29,6 @@
 import java.time.Instant;
 import java.time.ZoneId;
 import java.util.List;
-import java.util.stream.Stream;
 
 import static java.util.stream.Collectors.toUnmodifiableList;
 
@@ -59,17 +58,17 @@ public void clearPartitionData() throws StateStoreException {
 
     @Override
     public List<Partition> getAllPartitions() throws StateStoreException {
-        return partitions().collect(toUnmodifiableList());
+        return partitions().all().stream().toList();
     }
 
     @Override
     public List<Partition> getLeafPartitions() throws StateStoreException {
-        return partitions().filter(Partition::isLeafPartition).collect(toUnmodifiableList());
+        return partitions().all().stream().filter(Partition::isLeafPartition).collect(toUnmodifiableList());
     }
 
     @Override
     public Partition getPartition(String partitionId) throws StateStoreException {
-        return state().byId(partitionId)
+        return partitions().byId(partitionId)
                 .orElseThrow(() -> new StateStoreException("Partition not found: " + partitionId));
     }
 
@@ -106,11 +105,7 @@ void applyEntryFromLog(TransactionLogEntry logEntry, StateListenerBeforeApply li
         head.applyTransactionUpdatingIfNecessary(logEntry, listener);
     }
 
-    private Stream<Partition> partitions() throws StateStoreException {
-        return state().all().stream();
-    }
-
-    private StateStorePartitions state() throws StateStoreException {
+    private StateStorePartitions partitions() throws StateStoreException {
         head.update();
         return head.state();
     }