Inside Bitcoin

1// Copyright (c) 2022-present 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>
9#include <primitives/transaction.h>
10#include <script/script.h>
11#include <test/util/setup_common.h>
12#include <util/check.h>
13#include <util/translation.h>
14#include <wallet/context.h>
15#include <wallet/db.h>
16#include <wallet/test/util.h>
17#include <wallet/transaction.h>
18#include <wallet/wallet.h>
19#include <wallet/walletutil.h>
20
21#include <cstdint>
22#include <memory>
23#include <optional>
24#include <string>
25#include <utility>
26#include <vector>
27
28namespace wallet{
29static void AddTx(CWallet& wallet)
30{
31    CMutableTransaction mtx;
32    mtx.vout.emplace_back(COIN, GetScriptForDestination(*Assert(wallet.GetNewDestination(OutputType::BECH32, ""))));
33    mtx.vin.emplace_back();
34
35    wallet.AddToWallet(MakeTransactionRef(mtx), TxStateInactive{});
36}
37
38static void WalletLoadingDescriptors(benchmark::Bench& bench)
39{
40    const auto test_setup = MakeNoLogFileContext<TestingSetup>();
41
42    WalletContext context;
43    context.args = &test_setup->m_args;
44    context.chain = test_setup->m_node.chain.get();
45
46    // Setup the wallet
47    // Loading the wallet will also create it
48    uint64_t create_flags = WALLET_FLAG_DESCRIPTORS;
49    DatabaseStatus status;
50    DatabaseOptions options;
51    options.require_format = DatabaseFormat::SQLITE;
52    options.require_create = true;
53    bilingual_str error;
54    auto database = MakeWalletDatabase("", options, status, error);
55    auto wallet = TestCreateWallet(std::move(database), context, create_flags);
56
57    // Generate a bunch of transactions and addresses to put into the wallet
58    for (int i = 0; i < 1000; ++i) {
59        AddTx(*wallet);
60    }
61
62    options.require_create = false;
63    options.require_existing = true;
64
65    bench.epochs(5)
66        .setup([&] {
67            TestUnloadWallet(std::move(wallet));
68            database = MakeWalletDatabase("", options, status, error);
69        })
70        .run([&] {
71            wallet = TestLoadWallet(std::move(database), context);
72        });
73
74    // Cleanup
75    TestUnloadWallet(std::move(wallet));
76}
77
78BENCHMARK(WalletLoadingDescriptors);
79} // namespace wallet
80