chatmail-core/benches/receive_emails.rs

use std::convert::TryInto;

use async_std::{path::PathBuf, task::block_on};
use criterion::{
    async_executor::AsyncStdExecutor, black_box, criterion_group, criterion_main, BatchSize,
    BenchmarkId, Criterion,
};
use deltachat::{
    config::Config,
    context::Context,
    dc_receive_imf::dc_receive_imf,
    imex::{imex, ImexMode},
};
use tempfile::tempdir;

async fn recv_emails(context: Context, emails: &[&[u8]]) -> Context {
    for (i, bytes) in emails.iter().enumerate() {
        dc_receive_imf(
            &context,
            bytes,
            "INBOX",
            black_box(i.try_into().unwrap()),
            false,
        )
        .await
        .unwrap();
    }
    context
}

async fn recv_all_emails(mut context: Context, needs_move_enabled: bool) -> Context {
    context.disable_needs_move = !needs_move_enabled;

    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,
            imf_raw.as_bytes(),
            "INBOX",
            black_box(i.try_into().unwrap()),
            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("FakeOS".into(), 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)
            .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("from_elem");
    for needs_move_enabled in [false, true] {
        group.bench_with_input(
            BenchmarkId::new("Receive many messages", needs_move_enabled),
            &needs_move_enabled,
            |b, needs_move_enabled| {
                b.to_async(AsyncStdExecutor).iter_batched(
                    || block_on(create_context()),
                    |context| recv_all_emails(black_box(context), *needs_move_enabled),
                    BatchSize::LargeInput,
                );
            },
        );
    }
    group.finish();
}

criterion_group!(benches, criterion_benchmark);
criterion_main!(benches);