When rebuilding account view include mempool TX in coins view

DeFiCh · Sep 13, 2023 · b3fd350 · b3fd350
1 parent 1dfeb92
commit b3fd350
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/txmempool.cpp b/src/txmempool.cpp
@@ -552,7 +552,14 @@ void CTxMemPool::removeForReorg(const CCoinsViewCache *pcoins, unsigned int nMem
     if (pcustomcsview) {
         accountsViewDirty |= forceRebuildForReorg;
         CTransactionRef ptx{};
-        rebuildAccountsView(nMemPoolHeight, &::ChainstateActive().CoinsTip(), ptx);
+
+        CCoinsView dummy;
+        CCoinsViewCache view(&dummy);
+        CCoinsViewCache& coins_cache = ::ChainstateActive().CoinsTip();
+        CCoinsViewMemPool viewMemPool(&coins_cache, *this);
+        view.SetBackend(viewMemPool);
+
+        rebuildAccountsView(nMemPoolHeight, view, ptx);
     }
 }
 
@@ -619,7 +626,14 @@ void CTxMemPool::removeForBlock(const std::vector<CTransactionRef>& vtx, unsigne
     if (pcustomcsview) {
         accountsViewDirty |= forceRebuildForReorg;
         CTransactionRef ptx{};
-        rebuildAccountsView(nBlockHeight, &::ChainstateActive().CoinsTip(), ptx);
+
+        CCoinsView dummy;
+        CCoinsViewCache view(&dummy);
+        CCoinsViewCache& coins_cache = ::ChainstateActive().CoinsTip();
+        CCoinsViewMemPool viewMemPool(&coins_cache, *this);
+        view.SetBackend(viewMemPool);
+
+        rebuildAccountsView(nBlockHeight, view, ptx);
     }
 
     lastRollingFeeUpdate = GetTime();