astatin3/unshell
1
0
Fork 0
mirror of https://github.com/Astatin3/unshell.git synced 2026-06-08 22:38:01 -06:00
Code Issues Packages Projects Releases Wiki Activity

Update protocol_bench.rs

Browse Source
This commit is contained in:
Michael Mikovsky
2026-04-25 12:45:46 -06:00
parent 1f70f99ba6
commit 16164354e3
1 changed files with 45 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/protocol_bench.rs
+33 -11
View File
@@ -1,7 +1,9 @@
use std::hint::black_box; use std::hint::black_box;
use std::time::Instant; use std::time::Instant;
use unshell::protocol::tree::{ChildRoute, Endpoint, Ingress, LeafSpec, LocalEvent, ProtocolEndpoint}; use unshell::protocol::tree::{
ChildRoute, Endpoint, Ingress, LeafSpec, LocalEvent, ProtocolEndpoint,
};
use unshell::protocol::{CallMessage, PacketHeader, PacketType, decode_frame, encode_packet}; use unshell::protocol::{CallMessage, PacketHeader, PacketType, decode_frame, encode_packet};
const SAMPLES: usize = 500; const SAMPLES: usize = 500;
@@ -21,7 +23,10 @@ fn main() {
bench_hook_data_receive(), bench_hook_data_receive(),
]; ];
println!("{:32} {:>14} {:>14} {:>14}", "benchmark", "mean ns/op", "stddev", "samples"); println!(
"{:32} {:>14} {:>14} {:>14}",
"benchmark", "mean ns/op", "stddev", "samples"
);
for bench in benches { for bench in benches {
println!( println!(
"{:32} {:>14.2} {:>14.2} {:>14}", "{:32} {:>14.2} {:>14.2} {:>14}",
@@ -52,7 +57,8 @@ fn bench_encode_call() -> BenchResult {
}; };
run_bench("encode_call", || { run_bench("encode_call", || {
let frame = encode_packet(black_box(&header), black_box(&message)).expect("encode should work"); let frame =
encode_packet(black_box(&header), black_box(&message)).expect("encode should work");
black_box(frame.len()); black_box(frame.len());
}) })
} }
@@ -80,24 +86,39 @@ fn bench_decode_call() -> BenchResult {
} }
fn bench_forward_call_receive() -> BenchResult { fn bench_forward_call_receive() -> BenchResult {
run_prebuilt_bench("forward_call_receive", build_forward_call_cases, |(mut root, frame)| { run_prebuilt_bench(
let outcome = root.receive(&Ingress::Local, frame).expect("forward receive should work"); "forward_call_receive",
build_forward_call_cases,
|(mut root, frame)| {
let outcome = root
.receive(&Ingress::Local, frame)
.expect("forward receive should work");
black_box(outcome.forward.is_some()); black_box(outcome.forward.is_some());
}) },
)
} }
fn bench_local_call_receive() -> BenchResult { fn bench_local_call_receive() -> BenchResult {
run_prebuilt_bench("local_call_receive", build_local_call_cases, |(mut endpoint, frame)| { run_prebuilt_bench(
let outcome = endpoint.receive(&Ingress::Parent, frame).expect("local call should work"); "local_call_receive",
build_local_call_cases,
|(mut endpoint, frame)| {
let outcome = endpoint
.receive(&Ingress::Parent, frame)
.expect("local call should work");
match black_box(outcome.event) { match black_box(outcome.event) {
Some(LocalEvent::Call { .. }) => {} Some(LocalEvent::Call { .. }) => {}
other => panic!("expected local call event, got {other:?}"), other => panic!("expected local call event, got {other:?}"),
} }
}) },
)
} }
fn bench_hook_data_receive() -> BenchResult { fn bench_hook_data_receive() -> BenchResult {
run_prebuilt_bench("hook_data_receive", build_hook_data_cases, |(mut host, frame)| { run_prebuilt_bench(
"hook_data_receive",
build_hook_data_cases,
|(mut host, frame)| {
let outcome = host let outcome = host
.receive(&Ingress::Child(path(&["worker"])), frame) .receive(&Ingress::Child(path(&["worker"])), frame)
.expect("hook data should work"); .expect("hook data should work");
@@ -105,7 +126,8 @@ fn bench_hook_data_receive() -> BenchResult {
Some(LocalEvent::Data { .. }) => {} Some(LocalEvent::Data { .. }) => {}
other => panic!("expected local data event, got {other:?}"), other => panic!("expected local data event, got {other:?}"),
} }
}) },
)
} }
fn run_bench(name: &'static str, mut op: impl FnMut()) -> BenchResult { fn run_bench(name: &'static str, mut op: impl FnMut()) -> BenchResult {