70 lines
2.0 KiB
Rust
70 lines
2.0 KiB
Rust
//! read many sources
|
|
#[macro_use]
|
|
extern crate criterion;
|
|
|
|
use criterion::Criterion;
|
|
use ethers_core::rand;
|
|
use ethers_solc::{artifacts::Source, project_util::TempProject};
|
|
use rand::{distributions::Alphanumeric, Rng};
|
|
use std::{
|
|
fs::File,
|
|
io::{BufWriter, Write},
|
|
path::{Path, PathBuf},
|
|
};
|
|
|
|
fn read_all_benchmark(c: &mut Criterion) {
|
|
let root = tempfile::tempdir().unwrap();
|
|
let inputs = prepare_contracts(root.path(), 35);
|
|
|
|
let mut group = c.benchmark_group("read many");
|
|
group.bench_function("sequential", |b| {
|
|
b.iter(|| {
|
|
Source::read_all(&inputs).unwrap();
|
|
});
|
|
});
|
|
group.bench_function("parallel", |b| {
|
|
b.iter(|| {
|
|
Source::par_read_all(&inputs).unwrap();
|
|
});
|
|
});
|
|
}
|
|
|
|
fn read_solmate(c: &mut Criterion) {
|
|
let prj = TempProject::checkout("transmissions11/solmate").unwrap();
|
|
let inputs = ethers_solc::utils::source_files(prj.sources_path());
|
|
|
|
let mut group = c.benchmark_group("read solmate");
|
|
group.bench_function("sequential", |b| {
|
|
b.iter(|| {
|
|
Source::read_all(&inputs).unwrap();
|
|
});
|
|
});
|
|
group.bench_function("parallel", |b| {
|
|
b.iter(|| {
|
|
Source::par_read_all(&inputs).unwrap();
|
|
});
|
|
});
|
|
}
|
|
|
|
fn prepare_contracts(root: &Path, num: usize) -> Vec<PathBuf> {
|
|
let mut files = Vec::with_capacity(num);
|
|
for _ in 0..num {
|
|
let path = root.join(format!("file{num}.sol"));
|
|
let f = File::create(&path).unwrap();
|
|
let mut writer = BufWriter::new(f);
|
|
|
|
let mut rng = rand::thread_rng();
|
|
|
|
// let's assume a solidity file is between 2kb and 16kb
|
|
let n: usize = rng.gen_range(4..17);
|
|
let s: String = rng.sample_iter(&Alphanumeric).take(n * 1024).map(char::from).collect();
|
|
writer.write_all(s.as_bytes()).unwrap();
|
|
writer.flush().unwrap();
|
|
files.push(path)
|
|
}
|
|
files
|
|
}
|
|
|
|
criterion_group!(benches, read_all_benchmark, read_solmate);
|
|
criterion_main!(benches);
|