Inside Bitcoin

1// Copyright (c) 2023-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 <bench/bench.h>
6#include <bench/data/block413567.raw.h>
7#include <flatfile.h>
8#include <node/blockstorage.h>
9#include <primitives/block.h>
10#include <primitives/transaction.h>
11#include <serialize.h>
12#include <streams.h>
13#include <sync.h>
14#include <test/util/setup_common.h>
15#include <util/check.h>
16#include <validation.h>
17
18#include <memory>
19#include <span>
20
21static CBlock CreateTestBlock()
22{
23    CBlock block;
24    SpanReader{benchmark::data::block413567} >> TX_WITH_WITNESS(block);
25    return block;
26}
27
28static void WriteBlockBench(benchmark::Bench& bench)
29{
30    const auto testing_setup{MakeNoLogFileContext<const TestingSetup>(ChainType::MAIN)};
31    auto& blockman{testing_setup->m_node.chainman->m_blockman};
32    const CBlock block{CreateTestBlock()};
33    bench.run([&] {
34        LOCK(::cs_main);
35        const auto pos{blockman.WriteBlock(block, 413'567)};
36        assert(!pos.IsNull());
37    });
38}
39
40static void ReadBlockBench(benchmark::Bench& bench)
41{
42    const auto testing_setup{MakeNoLogFileContext<const TestingSetup>(ChainType::MAIN)};
43    auto& blockman{testing_setup->m_node.chainman->m_blockman};
44    const auto& test_block{CreateTestBlock()};
45    const auto& expected_hash{test_block.GetHash()};
46    const auto& pos{WITH_LOCK(::cs_main, return blockman.WriteBlock(test_block, 413'567))};
47    bench.run([&] {
48        CBlock block;
49        const auto success{blockman.ReadBlock(block, pos, expected_hash)};
50        assert(success);
51    });
52}
53
54static void ReadRawBlockBench(benchmark::Bench& bench)
55{
56    const auto testing_setup{MakeNoLogFileContext<const TestingSetup>(ChainType::MAIN)};
57    auto& blockman{testing_setup->m_node.chainman->m_blockman};
58    const auto pos{WITH_LOCK(::cs_main, return blockman.WriteBlock(CreateTestBlock(), 413'567))};
59    bench.run([&] {
60        const auto res{blockman.ReadRawBlock(pos)};
61        assert(res);
62    });
63}
64
65BENCHMARK(WriteBlockBench);
66BENCHMARK(ReadBlockBench);
67BENCHMARK(ReadRawBlockBench);
68