Bitcoin Core 29.99.0
P2P Digital Currency
wallet_loading.cpp
Go to the documentation of this file.
1// Copyright (c) 2022 The Bitcoin Core developers
2// Distributed under the MIT software license, see the accompanying
3// file COPYING or http://www.opensource.org/licenses/mit-license.php.
4
5#include <addresstype.h>
6#include <bench/bench.h>
7#include <consensus/amount.h>
8#include <outputtype.h>
11#include <util/check.h>
12#include <wallet/context.h>
13#include <wallet/db.h>
14#include <wallet/test/util.h>
15#include <wallet/transaction.h>
16#include <wallet/wallet.h>
17#include <wallet/walletutil.h>
18
19#include <cstdint>
20#include <memory>
21#include <utility>
22#include <vector>
23
24namespace wallet{
25static void AddTx(CWallet& wallet)
26{
28 mtx.vout.emplace_back(COIN, GetScriptForDestination(*Assert(wallet.GetNewDestination(OutputType::BECH32, ""))));
29 mtx.vin.emplace_back();
30
31 wallet.AddToWallet(MakeTransactionRef(mtx), TxStateInactive{});
32}
33
35{
36 const auto test_setup = MakeNoLogFileContext<TestingSetup>();
37
38 WalletContext context;
39 context.args = &test_setup->m_args;
40 context.chain = test_setup->m_node.chain.get();
41
42 // Setup the wallet
43 // Loading the wallet will also create it
44 uint64_t create_flags = WALLET_FLAG_DESCRIPTORS;
45 auto database = CreateMockableWalletDatabase();
46 auto wallet = TestLoadWallet(std::move(database), context, create_flags);
47
48 // Generate a bunch of transactions and addresses to put into the wallet
49 for (int i = 0; i < 1000; ++i) {
50 AddTx(*wallet);
51 }
52
53 database = DuplicateMockDatabase(wallet->GetDatabase());
54
55 // reload the wallet for the actual benchmark
56 TestUnloadWallet(std::move(wallet));
57
58 bench.epochs(5).run([&] {
59 wallet = TestLoadWallet(std::move(database), context, create_flags);
60
61 // Cleanup
62 database = DuplicateMockDatabase(wallet->GetDatabase());
63 TestUnloadWallet(std::move(wallet));
64 });
65}
66
68} // namespace wallet
CScript GetScriptForDestination(const CTxDestination &dest)
Generate a Bitcoin scriptPubKey for the given CTxDestination.
static constexpr CAmount COIN
The amount of satoshis in one BTC.
Definition: amount.h:15
#define Assert(val)
Identity function.
Definition: check.h:106
Main entry point to nanobench's benchmarking facility.
Definition: nanobench.h:627
Bench & run(char const *benchmarkName, Op &&op)
Repeatedly calls op() based on the configuration, and performs measurements.
Definition: nanobench.h:1234
Bench & epochs(size_t numEpochs) noexcept
Controls number of epochs, the number of measurements to perform.
A CWallet maintains a set of transactions and balances, and provides the ability to create new transa...
Definition: wallet.h:300
@ HIGH
Definition: bench.h:48
BENCHMARK(WalletBalanceDirty, benchmark::PriorityLevel::HIGH)
std::shared_ptr< CWallet > TestLoadWallet(std::unique_ptr< WalletDatabase > database, WalletContext &context, uint64_t create_flags)
Definition: util.cpp:51
void TestUnloadWallet(std::shared_ptr< CWallet > &&wallet)
Definition: util.cpp:74
static void AddTx(CWallet &wallet)
std::unique_ptr< WalletDatabase > CreateMockableWalletDatabase(MockableData records)
Definition: util.cpp:187
std::unique_ptr< WalletDatabase > DuplicateMockDatabase(WalletDatabase &database)
Definition: util.cpp:82
@ WALLET_FLAG_DESCRIPTORS
Indicate that this wallet supports DescriptorScriptPubKeyMan.
Definition: walletutil.h:74
static void WalletLoadingDescriptors(benchmark::Bench &bench)
static CTransactionRef MakeTransactionRef(Tx &&txIn)
Definition: transaction.h:424
A mutable version of CTransaction.
Definition: transaction.h:378
std::vector< CTxOut > vout
Definition: transaction.h:380
std::vector< CTxIn > vin
Definition: transaction.h:379
State of transaction not confirmed or conflicting with a known block and not in the mempool.
Definition: transaction.h:58
WalletContext struct containing references to state shared between CWallet instances,...
Definition: context.h:36
interfaces::Chain * chain
Definition: context.h:37
ArgsManager * args
Definition: context.h:39