[Refactoring] Add GetCoinDepth utility function to CCoinsViewCache

random-zebra · random-zebra · commit ae2af54d985d · 2020-09-05T16:20:18.000+02:00
diff --git a/src/coins.cpp b/src/coins.cpp
@@ -268,6 +268,14 @@ double CCoinsViewCache::GetPriority(const CTransaction& tx, int nHeight, CAmount
     return tx.ComputePriority(dResult);
 }
 
+int CCoinsViewCache::GetCoinDepthAtHeight(const COutPoint& output, int nHeight) const
+{
+    const Coin& coin = AccessCoin(output);
+    if (!coin.IsSpent())
+        return nHeight - coin.nHeight + 1;
+    return -1;
+}
+
 static const size_t MAX_OUTPUTS_PER_BLOCK = MAX_BLOCK_SIZE_CURRENT /  ::GetSerializeSize(CTxOut(), SER_NETWORK, PROTOCOL_VERSION); // TODO: merge with similar definition in undo.h.
 
 const Coin& AccessByTxid(const CCoinsViewCache& view, const uint256& txid)
diff --git a/src/coins.h b/src/coins.h
@@ -285,6 +285,12 @@ class CCoinsViewCache : public CCoinsViewBacked
      */
     double GetPriority(const CTransaction& tx, int nHeight, CAmount &inChainInputValue) const;
 
+    /*
+     * Return the depth of a coin at height nHeight, or -1 if not found
+     */
+    int GetCoinDepthAtHeight(const COutPoint& output, int nHeight) const;
+
+
 private:
     CCoinsMap::iterator FetchCoin(const COutPoint& outpoint) const;
 
