Changeset View
Changeset View
Standalone View
Standalone View
src/bench/wallet_balance.cpp
- This file was added.
// 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 <chainparams.h> // For Params() | |||||
#include <config.h> | |||||
#include <interfaces/chain.h> | |||||
#include <key_io.h> | |||||
#include <optional.h> | |||||
#include <test/util.h> | |||||
#include <validationinterface.h> | |||||
#include <wallet/wallet.h> | |||||
struct WalletTestingSetup { | |||||
std::unique_ptr<interfaces::Chain> m_chain = interfaces::MakeChain(); | |||||
CWallet m_wallet; | |||||
WalletTestingSetup() | |||||
: m_wallet{Params(), m_chain.get(), WalletLocation(), | |||||
WalletDatabase::CreateMock()} {} | |||||
void handleNotifications() { | |||||
m_wallet.m_chain_notifications_handler = | |||||
m_chain->handleNotifications(m_wallet); | |||||
} | |||||
}; | |||||
static void WalletBalance(benchmark::State &state, const bool set_dirty, | |||||
const bool add_watchonly, const bool add_mine) { | |||||
const auto &ADDRESS_WATCHONLY = ADDRESS_BCHREG_UNSPENDABLE; | |||||
WalletTestingSetup wallet_t{}; | |||||
auto &wallet = wallet_t.m_wallet; | |||||
{ | |||||
bool first_run; | |||||
if (wallet.LoadWallet(first_run) != DBErrors::LOAD_OK) { | |||||
assert(false); | |||||
} | |||||
wallet_t.handleNotifications(); | |||||
} | |||||
const Config &config = GetConfig(); | |||||
const Optional<std::string> address_mine{ | |||||
add_mine ? Optional<std::string>{getnewaddress(config, wallet)} | |||||
: nullopt}; | |||||
if (add_watchonly) { | |||||
importaddress(wallet, ADDRESS_WATCHONLY); | |||||
} | |||||
for (int i = 0; i < 100; ++i) { | |||||
generatetoaddress(config, address_mine.get_value_or(ADDRESS_WATCHONLY)); | |||||
generatetoaddress(config, ADDRESS_WATCHONLY); | |||||
} | |||||
SyncWithValidationInterfaceQueue(); | |||||
// Cache | |||||
auto bal = wallet.GetBalance(); | |||||
while (state.KeepRunning()) { | |||||
if (set_dirty) { | |||||
wallet.MarkDirty(); | |||||
} | |||||
bal = wallet.GetBalance(); | |||||
if (add_mine) { | |||||
assert(bal.m_mine_trusted > Amount::zero()); | |||||
} | |||||
if (add_watchonly) { | |||||
assert(bal.m_watchonly_trusted > Amount::zero()); | |||||
} | |||||
} | |||||
} | |||||
static void WalletBalanceDirty(benchmark::State &state) { | |||||
WalletBalance(state, /* set_dirty */ true, /* add_watchonly */ true, | |||||
/* add_mine */ true); | |||||
} | |||||
static void WalletBalanceClean(benchmark::State &state) { | |||||
WalletBalance(state, /* set_dirty */ false, /* add_watchonly */ true, | |||||
/* add_mine */ true); | |||||
} | |||||
static void WalletBalanceMine(benchmark::State &state) { | |||||
WalletBalance(state, /* set_dirty */ false, /* add_watchonly */ false, | |||||
/* add_mine */ true); | |||||
} | |||||
static void WalletBalanceWatch(benchmark::State &state) { | |||||
WalletBalance(state, /* set_dirty */ false, /* add_watchonly */ true, | |||||
/* add_mine */ false); | |||||
} | |||||
BENCHMARK(WalletBalanceDirty, 2500); | |||||
BENCHMARK(WalletBalanceClean, 8000); | |||||
BENCHMARK(WalletBalanceMine, 16000); | |||||
BENCHMARK(WalletBalanceWatch, 8000); |