Add method to remove a tx from CCoinsViewCache if it is unchanged

TheBlueMatt · TheBlueMatt · commit 74d0f9026284 · 2015-12-01T15:50:39.000-08:00
diff --git a/src/coins.cpp b/src/coins.cpp
@@ -206,6 +206,15 @@ bool CCoinsViewCache::Flush() {
     return fOk;
 }
 
+void CCoinsViewCache::Uncache(const uint256& hash)
+{
+    CCoinsMap::iterator it = cacheCoins.find(hash);
+    if (it != cacheCoins.end() && it->second.flags == 0) {
+        cachedCoinsUsage -= it->second.coins.DynamicMemoryUsage();
+        cacheCoins.erase(it);
+    }
+}
+
 unsigned int CCoinsViewCache::GetCacheSize() const {
     return cacheCoins.size();
 }
diff --git a/src/coins.h b/src/coins.h
@@ -437,6 +437,12 @@ class CCoinsViewCache : public CCoinsViewBacked
      */
     bool Flush();
 
+    /**
+     * Removes the transaction with the given hash from the cache, if it is
+     * not modified.
+     */
+    void Uncache(const uint256 &txid);
+
     //! Calculate the size of the cache (in number of transactions)
     unsigned int GetCacheSize() const;
 
