38 lines
1.4 KiB
Rust
38 lines
1.4 KiB
Rust
// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
|
|
// Copyright by contributors to this project.
|
|
// SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
use criterion::{BatchSize, BenchmarkId, Criterion};
|
|
use mls_rs::test_utils::benchmarks::{load_group_states, BENCH_CIPHER_SUITE};
|
|
|
|
fn bench(c: &mut Criterion) {
|
|
let group_states = load_group_states();
|
|
let mut bench_group = c.benchmark_group("group_receive_commit");
|
|
|
|
for (i, mut group_states) in group_states.into_iter().enumerate() {
|
|
bench_group.bench_with_input(
|
|
BenchmarkId::new(format!("{BENCH_CIPHER_SUITE:?}"), i),
|
|
&i,
|
|
|b, _| {
|
|
b.iter_batched_ref(
|
|
|| {
|
|
let commit = group_states.sender.commit(Vec::new()).unwrap();
|
|
group_states.sender.clear_pending_commit();
|
|
(commit, group_states.receiver.clone())
|
|
},
|
|
move |(commit, receiver)| {
|
|
receiver
|
|
.process_incoming_message(commit.commit_message.clone())
|
|
.unwrap();
|
|
},
|
|
BatchSize::SmallInput,
|
|
)
|
|
},
|
|
);
|
|
}
|
|
|
|
bench_group.finish();
|
|
}
|
|
|
|
criterion::criterion_group!(benches, bench);
|
|
criterion::criterion_main!(benches);
|