git / moonbite-core / src/bench/coin_selection.cpp · raw

// Copyright (c) 2012-2019 The Bitcoin Core developers// Distributed under the MIT software license, see the accompanying// file COPYING or http://www.opensource.org/licenses/mit-license.php.#include <bench/bench.h>#include <interfaces/chain.h>#include <node/context.h>#include <wallet/coinselection.h>#include <wallet/wallet.h>#include <set>static void addCoin(const CAmount& nValue, const CWallet& wallet, std::vector<std::unique_ptr<CWalletTx>>& wtxs){    static int nextLockTime = 0;    CMutableTransaction tx;    tx.nLockTime = nextLockTime++; // so all transactions get different hashes    tx.vout.resize(1);    tx.vout[0].nValue = nValue;    wtxs.push_back(MakeUnique<CWalletTx>(&wallet, MakeTransactionRef(std::move(tx))));}// Simple benchmark for wallet coin selection. Note that it maybe be necessary// to build up more complicated scenarios in order to get meaningful// measurements of performance. From laanwj, "Wallet coin selection is probably// the hardest, as you need a wider selection of scenarios, just testing the// same one over and over isn't too useful. Generating random isn't useful// either for measurements."// (https://github.com/bitcoin/bitcoin/issues/7883#issuecomment-224807484)static void CoinSelection(benchmark::Bench& bench){    NodeContext node;    auto chain = interfaces::MakeChain(node);    CWallet wallet(chain.get(), "", CreateDummyWalletDatabase());    wallet.SetupLegacyScriptPubKeyMan();    std::vector<std::unique_ptr<CWalletTx>> wtxs;    LOCK(wallet.cs_wallet);    // Add coins.    for (int i = 0; i < 1000; ++i) {        addCoin(1000 * COIN, wallet, wtxs);    }    addCoin(3 * COIN, wallet, wtxs);    // Create groups    std::vector<OutputGroup> groups;    for (const auto& wtx : wtxs) {        COutput output(wtx.get(), 0 /* iIn */, 6 * 24 /* nDepthIn */, true /* spendable */, true /* solvable */, true /* safe */);        groups.emplace_back(output.GetInputCoin(), 6, false, 0, 0);    }    const CoinEligibilityFilter filter_standard(1, 6, 0);    const CoinSelectionParams coin_selection_params(/* use_bnb= */ true, /* change_output_size= */ 34, /* mweb_change_output_weight= */ 0,                                                    /* change_spend_size= */ 148, /* effective_feerate= */ CFeeRate(0),                                                    /* long_term_feerate= */ CFeeRate(0), /* discard_feerate= */ CFeeRate(0),                                                    /* tx_no_inputs_size= */ 0, /* mweb_no_change_weight= */ 0);    bench.run([&] {        std::set<CInputCoin> setCoinsRet;        CAmount nValueRet;        bool bnb_used;        bool success = wallet.SelectCoinsMinConf(1003 * COIN, filter_standard, groups, setCoinsRet, nValueRet, coin_selection_params, bnb_used);        assert(success);        assert(nValueRet == 1003 * COIN);        assert(setCoinsRet.size() == 2);    });}typedef std::set<CInputCoin> CoinSet;static NodeContext testNode;static auto testChain = interfaces::MakeChain(testNode);static CWallet testWallet(testChain.get(), "", CreateDummyWalletDatabase());std::vector<std::unique_ptr<CWalletTx>> wtxn;// Copied from src/wallet/test/coinselector_tests.cppstatic void add_coin(const CAmount& nValue, int nInput, std::vector<OutputGroup>& set){    CMutableTransaction tx;    tx.vout.resize(nInput + 1);    tx.vout[nInput].nValue = nValue;    std::unique_ptr<CWalletTx> wtx = MakeUnique<CWalletTx>(&testWallet, MakeTransactionRef(std::move(tx)));    set.emplace_back(COutput(wtx.get(), nInput, 0, true, true, true).GetInputCoin(), 0, true, 0, 0);    wtxn.emplace_back(std::move(wtx));}// Copied from src/wallet/test/coinselector_tests.cppstatic CAmount make_hard_case(int utxos, std::vector<OutputGroup>& utxo_pool){    utxo_pool.clear();    CAmount target = 0;    for (int i = 0; i < utxos; ++i) {        target += (CAmount)1 << (utxos+i);        add_coin((CAmount)1 << (utxos+i), 2*i, utxo_pool);        add_coin(((CAmount)1 << (utxos+i)) + ((CAmount)1 << (utxos-1-i)), 2*i + 1, utxo_pool);    }    return target;}static void BnBExhaustion(benchmark::Bench& bench){    // Setup    testWallet.SetupLegacyScriptPubKeyMan();    std::vector<OutputGroup> utxo_pool;    CoinSet selection;    CAmount value_ret = 0;    CAmount not_input_fees = 0;    bench.run([&] {        // Benchmark        CAmount target = make_hard_case(17, utxo_pool);        SelectCoinsBnB(utxo_pool, target, 0, selection, value_ret, not_input_fees); // Should exhaust        // Cleanup        utxo_pool.clear();        selection.clear();    });}BENCHMARK(CoinSelection);BENCHMARK(BnBExhaustion);