Files
pezkuwi-sdk/pezcumulus/test/service/benches/block_import.rs
T
pezkuwichain 4666047395 chore: add Dijital Kurdistan Tech Institute to copyright headers
Updated 4763 files with dual copyright:
- Parity Technologies (UK) Ltd.
- Dijital Kurdistan Tech Institute
2025-12-27 21:28:36 +03:00

102 lines
3.1 KiB
Rust

// This file is part of Pezcumulus.
// Copyright (C) Parity Technologies (UK) Ltd. and Dijital Kurdistan Tech Institute
// SPDX-License-Identifier: Apache-2.0
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
use criterion::{criterion_group, criterion_main, BatchSize, Criterion, Throughput};
use pezsc_block_builder::BlockBuilderBuilder;
use pezsc_client_api::UsageProvider;
use core::time::Duration;
use pezcumulus_primitives_core::ParaId;
use pezsp_api::{Core, ProvideRuntimeApi};
use pezsp_keyring::Sr25519Keyring::{Alice, Bob};
use pezcumulus_test_service::bench_utils as utils;
fn benchmark_block_import(c: &mut Criterion) {
pezsp_tracing::try_init_simple();
let runtime = tokio::runtime::Runtime::new().expect("creating tokio runtime doesn't fail; qed");
let para_id = ParaId::from(pezcumulus_test_runtime::TEYRCHAIN_ID);
let tokio_handle = runtime.handle();
// Create enough accounts to fill the block with transactions.
// Each account should only be included in one transfer.
let (src_accounts, dst_accounts, account_ids) = utils::create_benchmark_accounts();
for bench_parameters in &[(true, Alice), (false, Bob)] {
let node = runtime.block_on(
pezcumulus_test_service::TestNodeBuilder::new(
para_id,
tokio_handle.clone(),
bench_parameters.1,
)
// Preload all accounts with funds for the transfers
.endowed_accounts(account_ids.clone())
.import_proof_recording(bench_parameters.0)
.build(),
);
let client = node.client;
let backend = node.backend;
let (max_transfer_count, extrinsics) =
utils::create_benchmarking_transfer_extrinsics(&client, &src_accounts, &dst_accounts);
let parent_hash = client.usage_info().chain.best_hash;
let mut block_builder = BlockBuilderBuilder::new(&*client)
.on_parent_block(parent_hash)
.fetch_parent_block_number(&*client)
.unwrap()
.build()
.unwrap();
for extrinsic in extrinsics {
block_builder.push(extrinsic).unwrap();
}
let benchmark_block = block_builder.build().unwrap();
let mut group = c.benchmark_group("Block import");
group.sample_size(20);
group.measurement_time(Duration::from_secs(120));
group.throughput(Throughput::Elements(max_transfer_count as u64));
group.bench_function(
format!(
"(transfers = {max_transfer_count}, proof_recording = {}) block import",
bench_parameters.0
),
|b| {
b.iter_batched(
|| {
backend.reset_trie_cache();
benchmark_block.block.clone()
},
|block| {
client.runtime_api().execute_block(parent_hash, block.into()).unwrap();
},
BatchSize::SmallInput,
)
},
);
}
}
criterion_group!(benches, benchmark_block_import);
criterion_main!(benches);