LedgerDB.V2.TestInternals: prune LedgerSeq

This is used in db-analyser only, where everything happens synchronously in a
single thread, so it is fine to immediately prune.

V1 already does this.

Author: amesgen

ouroboros-consensus/src/ouroboros-consensus/Ouroboros/Consensus/Storage/LedgerDB/API.hs

@@ -305,7 +305,14 @@ data TestInternals m l blk = TestInternals
   { wipeLedgerDB :: m ()
   , takeSnapshotNOW :: WhereToTakeSnapshot -> Maybe String -> m ()
   , push :: ExtLedgerState blk DiffMK -> m ()
+  -- ^ Push a ledger state, and prune the 'LedgerDB' w.r.t. the security parameter.
+  --
+  -- This does not modify the set of previously applied points.
   , reapplyThenPushNOW :: blk -> m ()
+  -- ^ Apply block to the tip ledger state (using reapplication), and prune the
+  -- 'LedgerDB' w.r.t. the security parameter.
+  --
+  -- This does not modify the set of previously applied points.
   , truncateSnapshots :: m ()
   , closeLedgerDB :: m ()
   }

ouroboros-consensus/src/ouroboros-consensus/Ouroboros/Consensus/Storage/LedgerDB/V2.hs

@@ -1,3 +1,4 @@
+{-# LANGUAGE BangPatterns #-}
 {-# LANGUAGE CPP #-}
 {-# LANGUAGE DataKinds #-}
 {-# LANGUAGE DeriveAnyClass #-}
@@ -35,6 +36,7 @@ import Data.Set (Set)
 import qualified Data.Set as Set
 import Data.Void
 import Data.Word
+import qualified Debug.Trace as DT
 import GHC.Generics
 import NoThunks.Class
 import Ouroboros.Consensus.Block
@@ -210,8 +212,9 @@ mkInternals bss h =
         eFrk <- newForkerAtTarget h reg VolatileTip
         case eFrk of
           Left{} -> error "Unreachable, Volatile tip MUST be in LedgerDB"
-          Right frk ->
+          Right frk -> do
             forkerPush frk st >> atomically (forkerCommit frk) >> forkerClose frk
+            getEnv h pruneLedgerSeq
     , reapplyThenPushNOW = \blk -> getEnv h $ \env -> withRegistry $ \reg -> do
         eFrk <- newForkerAtTarget h reg VolatileTip
         case eFrk of
@@ -226,6 +229,7 @@ mkInternals bss h =
                     blk
                     (st `withLedgerTables` tables)
             forkerPush frk st' >> atomically (forkerCommit frk) >> forkerClose frk
+            pruneLedgerSeq env
     , wipeLedgerDB = getEnv h $ destroySnapshots . ldbHasFS
     , closeLedgerDB =
         let LDBHandle tvar = h
@@ -244,6 +248,12 @@ mkInternals bss h =
     InMemoryHandleArgs -> InMemory.takeSnapshot
     LSMHandleArgs x -> absurd x
 
+  pruneLedgerSeq :: LedgerDBEnv m (ExtLedgerState blk) blk -> m ()
+  pruneLedgerSeq env =
+    join $ atomically $ stateTVar (ldbSeq env) $ prune (LedgerDbPruneKeeping k)
+   where
+    k = ledgerDbCfgSecParam $ ldbCfg env
+
 -- | Testing only! Truncate all snapshots in the DB.
 implIntTruncateSnapshots :: MonadThrow m => SomeHasFS m -> m ()
 implIntTruncateSnapshots sfs@(SomeHasFS fs) = do
@@ -412,8 +422,11 @@ implTryTakeSnapshot bss env mTime nrBlocks =
 -- periodically flush the accumulated differences to the disk. However, in the
 -- second version there is no need to do so, and because of that, this function
 -- does nothing in this case.
-implTryFlush :: Applicative m => LedgerDBEnv m l blk -> m ()
-implTryFlush _ = pure ()
+implTryFlush :: (MonadSTM m, GetTip l) => LedgerDBEnv m l blk -> m ()
+implTryFlush env = do
+  LedgerSeq ldb <- readTVarIO $ ldbSeq env
+  !_ <- DT.traceM $ "ledger seq len: " <> show (AS.length ldb)
+  pure ()
 
 implCloseDB :: IOLike m => LedgerDBHandle m l blk -> m ()
 implCloseDB (LDBHandle varState) = do