node: Added vector based coin lookup

TomBriar · Mar 7, 2024 · 593d58d · 593d58d
1 parent ebba0f0
commit 593d58d
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 0 deletions.
diff --git a/src/node/coin.cpp b/src/node/coin.cpp
@@ -23,4 +23,23 @@ void FindCoins(const NodeContext& node, std::map<COutPoint, Coin>& coins)
         }
     }
 }
+
+std::vector<Coin> FindCoins(const node::NodeContext& node, const std::vector<COutPoint>& outpoints)
+{
+    assert(node.mempool);
+    assert(node.chainman);
+    LOCK2(cs_main, node.mempool->cs);
+    CCoinsViewCache& chain_view = node.chainman->ActiveChainstate().CoinsTip();
+    CCoinsViewMemPool mempool_view(&chain_view, *node.mempool);
+    std::vector<Coin> coins;
+    for (auto& outpoint : outpoints) {
+        Coin coin;
+        if (!mempool_view.GetCoin(outpoint, coin)) {
+            // Either the coin is not in the CCoinsViewCache or is spent. Clear it.
+            coin.Clear();
+        }
+        coins.push_back(coin);
+    }
+    return coins;
+}
 } // namespace node
diff --git a/src/node/coin.h b/src/node/coin.h
@@ -6,6 +6,7 @@
 #define BITCOIN_NODE_COIN_H
 
 #include <map>
+#include <vector>
 
 class COutPoint;
 class Coin;
@@ -22,6 +23,18 @@ struct NodeContext;
  * @param[in,out] coins map to fill
  */
 void FindCoins(const node::NodeContext& node, std::map<COutPoint, Coin>& coins);
+
+/**
+ * Look up unspent output information. Returns coins in the mempool and in the
+ * current chain UTXO set. Iterates through all the outpoints and
+ * returns the corresponding coins.
+ *
+ * @param[in] node The node context to use for lookup
+ * @param[in] outpoints A Vector containing outpoints
+ */
+std::vector<Coin> FindCoins(const node::NodeContext& node, const std::vector<COutPoint>& outpoints);
 } // namespace node
 
+
+
 #endif // BITCOIN_NODE_COIN_H