mirror of
https://github.com/chatmail/core.git
synced 2026-04-17 21:46:35 +03:00
ea7fc3a171
Don't count the account creation in the receive emails benchmark Use Criterion's async support See https://bheisler.github.io/criterion.rs/book/user_guide/benchmarking_async.html
85 lines
2.3 KiB
Rust
85 lines
2.3 KiB
Rust
use async_std::{path::PathBuf, task::block_on};
|
|
use criterion::{
|
|
async_executor::AsyncStdExecutor, black_box, criterion_group, criterion_main, BatchSize,
|
|
Criterion,
|
|
};
|
|
use deltachat::{
|
|
config::Config,
|
|
context::Context,
|
|
dc_receive_imf::dc_receive_imf,
|
|
imex::{imex, ImexMode},
|
|
};
|
|
use tempfile::tempdir;
|
|
|
|
async fn recv_all_emails(context: Context) -> Context {
|
|
for i in 0..100 {
|
|
let imf_raw = format!(
|
|
"Subject: Benchmark
|
|
Message-ID: Mr.OssSYnOFkhR.{i}@testrun.org
|
|
Date: Sat, 07 Dec 2019 19:00:27 +0000
|
|
To: alice@example.com
|
|
From: sender@testrun.org
|
|
Chat-Version: 1.0
|
|
Chat-Disposition-Notification-To: sender@testrun.org
|
|
Chat-User-Avatar: 0
|
|
In-Reply-To: Mr.OssSYnOFkhR.{i_dec}@testrun.org
|
|
MIME-Version: 1.0
|
|
|
|
Content-Type: text/plain; charset=utf-8; format=flowed; delsp=no
|
|
|
|
Hello {i}",
|
|
i = i,
|
|
i_dec = i - 1,
|
|
);
|
|
dc_receive_imf(&context, black_box(imf_raw.as_bytes()), "INBOX", false)
|
|
.await
|
|
.unwrap();
|
|
}
|
|
context
|
|
}
|
|
|
|
async fn create_context() -> Context {
|
|
let dir = tempdir().unwrap();
|
|
let dbfile = dir.path().join("db.sqlite");
|
|
let id = 100;
|
|
let context = Context::new(dbfile.into(), id).await.unwrap();
|
|
|
|
let backup: PathBuf = std::env::current_dir()
|
|
.unwrap()
|
|
.join("delta-chat-backup.tar")
|
|
.into();
|
|
if backup.exists().await {
|
|
println!("Importing backup");
|
|
imex(&context, ImexMode::ImportBackup, &backup, None)
|
|
.await
|
|
.unwrap();
|
|
}
|
|
|
|
let addr = "alice@example.com";
|
|
context.set_config(Config::Addr, Some(addr)).await.unwrap();
|
|
context
|
|
.set_config(Config::ConfiguredAddr, Some(addr))
|
|
.await
|
|
.unwrap();
|
|
context
|
|
.set_config(Config::Configured, Some("1"))
|
|
.await
|
|
.unwrap();
|
|
context
|
|
}
|
|
|
|
fn criterion_benchmark(c: &mut Criterion) {
|
|
let mut group = c.benchmark_group("Receive messages");
|
|
group.bench_function("Receive 100 simple text msgs", |b| {
|
|
b.to_async(AsyncStdExecutor).iter_batched(
|
|
|| block_on(create_context()),
|
|
|context| recv_all_emails(black_box(context)),
|
|
BatchSize::LargeInput,
|
|
);
|
|
});
|
|
group.finish();
|
|
}
|
|
|
|
criterion_group!(benches, criterion_benchmark);
|
|
criterion_main!(benches);
|