CTxMemPool::removeForBlock now uses RemoveStaged

dpc · Mar 14, 2016 · 7659438 · 7659438
1 parent 0735c0c
commit 7659438
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/txmempool.cpp b/src/txmempool.cpp
@@ -564,8 +564,12 @@ void CTxMemPool::removeForBlock(const std::vector<CTransaction>& vtx, unsigned i
     }
     BOOST_FOREACH(const CTransaction& tx, vtx)
     {
-        std::list<CTransaction> 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());
     }