CTxMemPool::removeForBlock now uses RemoveStaged

Coming from btc@7659438a63ef162b4a4f942f86683ae6785f8162
PIVX-Project · Jan 18, 2021 · c30fa16 · c30fa16
1 parent ba73898
commit c30fa16
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/txmempool.cpp b/src/txmempool.cpp
@@ -603,8 +603,12 @@ void CTxMemPool::removeForBlock(const std::vector<CTransactionRef>& vtx, unsigne
             entries.push_back(*i);
     }
     for (const auto& tx : vtx) {
-        std::list<CTransactionRef> dummy;
-        remove(*tx, dummy, false);
+        txiter it = mapTx.find(tx->GetHash());
+        if (it != mapTx.end()) {
+            setEntries stage;
+            stage.insert(it);
+            RemoveStaged(stage);
+        }
         removeConflicts(*tx, conflicts);
         ClearPrioritisation(tx->GetHash());
     }