node/chainstate: Add options for in-memory DBs

dashpay · Dec 16, 2024 · c06e074 · c06e074
1 parent 52bb35d
commit c06e074
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/src/init.cpp b/src/init.cpp
@@ -1881,6 +1881,8 @@ bool AppInitMain(NodeContext& node, interfaces::BlockAndHeaderTipInfo* tip_info)
                                 cache_sizes.block_tree_db,
                                 cache_sizes.coins_db,
                                 cache_sizes.coins,
+                                /*block_tree_db_in_memory=*/false,
+                                /*coins_db_in_memory=*/false,
                                 ShutdownRequested,
                                 []() {
                                     uiInterface.ThreadSafeMessageBox(

diff --git a/src/node/chainstate.cpp b/src/node/chainstate.cpp
@@ -48,6 +48,8 @@ std::optional<ChainstateLoadingError> LoadChainstate(bool fReset,
                                                      int64_t nBlockTreeDBCache,
                                                      int64_t nCoinDBCache,
                                                      int64_t nCoinCacheUsage,
+                                                     bool block_tree_db_in_memory,
+                                                     bool coins_db_in_memory,
                                                      std::function<bool()> shutdown_requested,
                                                      std::function<void()> coins_error_cb)
 {
@@ -73,7 +75,7 @@ std::optional<ChainstateLoadingError> LoadChainstate(bool fReset,
         // new CBlockTreeDB tries to delete the existing file, which
         // fails if it's still open from the previous loop. Close it first:
         pblocktree.reset();
-        pblocktree.reset(new CBlockTreeDB(nBlockTreeDBCache, false, fReset));
+        pblocktree.reset(new CBlockTreeDB(nBlockTreeDBCache, block_tree_db_in_memory, fReset));
 
         // Same logic as above with pblocktree
         dmnman.reset();
@@ -177,7 +179,7 @@ std::optional<ChainstateLoadingError> LoadChainstate(bool fReset,
         for (CChainState* chainstate : chainman.GetAll()) {
             chainstate->InitCoinsDB(
                 /* cache_size_bytes */ nCoinDBCache,
-                /* in_memory */ false,
+                /* in_memory */ coins_db_in_memory,
                 /* should_wipe */ fReset || fReindexChainState);
 
             if (coins_error_cb) {

diff --git a/src/node/chainstate.h b/src/node/chainstate.h
@@ -110,6 +110,8 @@ std::optional<ChainstateLoadingError> LoadChainstate(bool fReset,
                                                      int64_t nBlockTreeDBCache,
                                                      int64_t nCoinDBCache,
                                                      int64_t nCoinCacheUsage,
+                                                     bool block_tree_db_in_memory,
+                                                     bool coins_db_in_memory,
                                                      std::function<bool()> shutdown_requested = nullptr,
                                                      std::function<void()> coins_error_cb = nullptr);