Add event deserialization benchmark

Signed-off-by: trivernis <trivernis@protonmail.com>
3 years ago · adea767aa0
parent dcd346e0c7
commit adea767aa0
4 changed files with 116 additions and 5 deletions
--- a/Cargo.lock
+++ b/Cargo.lock
@ -92,6 +92,7 @@ dependencies = [
 "clap",
 "criterion-plot",
 "csv",
+ "futures",
 "itertools",
 "lazy_static",
 "num-traits",
@ -104,6 +105,7 @@ dependencies = [
 "serde_derive",
 "serde_json",
 "tinytemplate",
+ "tokio",
 "walkdir",
 ]

@ -189,6 +191,68 @@ version = "1.6.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "e78d4f1cc4ae33bbfc157ed5d5a5ef3bc29227303d595861deb238fcec4e9457"

+[[package]]
+name = "futures"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a12aa0eb539080d55c3f2d45a67c3b58b6b0773c1a3ca2dfec66d58c97fd66ca"
+dependencies = [
+ "futures-channel",
+ "futures-core",
+ "futures-io",
+ "futures-sink",
+ "futures-task",
+ "futures-util",
+]
+
+[[package]]
+name = "futures-channel"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5da6ba8c3bb3c165d3c7319fc1cc8304facf1fb8db99c5de877183c08a273888"
+dependencies = [
+ "futures-core",
+ "futures-sink",
+]
+
+[[package]]
+name = "futures-core"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "88d1c26957f23603395cd326b0ffe64124b818f4449552f960d815cfba83a53d"
+
+[[package]]
+name = "futures-io"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "522de2a0fe3e380f1bc577ba0474108faf3f6b18321dbf60b3b9c39a75073377"
+
+[[package]]
+name = "futures-sink"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "36ea153c13024fe480590b3e3d4cad89a0cfacecc24577b68f86c6ced9c2bc11"
+
+[[package]]
+name = "futures-task"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1d3d00f4eddb73e498a54394f228cd55853bdf059259e8e7bc6e69d408892e99"
+
+[[package]]
+name = "futures-util"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "36568465210a3a6ee45e1f165136d68671471a501e632e9a98d96872222b5481"
+dependencies = [
+ "autocfg",
+ "futures-core",
+ "futures-sink",
+ "futures-task",
+ "pin-project-lite",
+ "pin-utils",
+]
+
 [[package]]
 name = "half"
 version = "1.8.2"
@ -326,6 +390,12 @@ version = "0.2.7"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "8d31d11c69a6b52a174b42bdc0c30e5e11670f90788b2c471c31c1d17d449443"

+[[package]]
+name = "pin-utils"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8b870d8c151b6f2fb93e84a13146138f05d02ed11c7e7c54f8826aaaf7c9f184"
+
 [[package]]
 name = "plotters"
 version = "0.3.1"
--- a/Cargo.toml
+++ b/Cargo.toml
@ -12,6 +12,10 @@ description = "IPC using Rust MessagePack (rmp)"
 name = "serialization_benchmark"
 harness = false

+[[bench]]
+name = "deserialization_benchmark"
+harness = false
+
 [dependencies]
 thiserror = "1.0.30"
 rmp-serde = "0.15.4"
@ -27,8 +31,9 @@ features = ["serde_derive"]
 version = "1.12.0"
 features = ["net", "io-std", "io-util", "sync", "time"]

-[dev-dependencies]
-criterion = "0.3.5"
+[dev-dependencies.criterion]
+version = "0.3.5"
+features = ["async_tokio", "html_reports"]

 [dev-dependencies.tokio]
 version = "1.12.0"
--- a/benches/deserialization_benchmark.rs
+++ b/benches/deserialization_benchmark.rs
@ -0,0 +1,35 @@
+use criterion::Criterion;
+use criterion::{black_box, BenchmarkId, Throughput};
+use criterion::{criterion_group, criterion_main};
+use std::io::Cursor;
+
+use rmp_ipc::event::Event;
+use tokio::runtime::Runtime;
+
+pub const EVENT_NAME: &str = "bench_event";
+
+fn create_event_bytes_reader(data_size: usize) -> Cursor<Vec<u8>> {
+    let bytes = Event::new(EVENT_NAME.to_string(), vec![0u8; data_size], None)
+        .into_bytes()
+        .unwrap();
+    Cursor::new(bytes)
+}
+
+fn event_deserialization(c: &mut Criterion) {
+    let runtime = Runtime::new().unwrap();
+    let mut group = c.benchmark_group("event_deserialization");
+
+    for size in (0..32).map(|i| i * 1024) {
+        group.throughput(Throughput::Bytes(size));
+        group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, &size| {
+            b.to_async(&runtime).iter(|| async {
+                let mut reader = black_box(create_event_bytes_reader(size as usize));
+                Event::from_async_read(&mut reader).await.unwrap()
+            })
+        });
+    }
+    group.finish()
+}
+
+criterion_group!(benches, event_deserialization);
+criterion_main!(benches);
--- a/benches/serialization_benchmark.rs
+++ b/benches/serialization_benchmark.rs
@ -7,12 +7,13 @@ fn create_event(data_size: usize) -> Event {
    Event::new(EVENT_NAME.to_string(), vec![0u8; data_size], None)
 }

-pub fn event_serialization(c: &mut Criterion) {
+fn event_serialization(c: &mut Criterion) {
    let mut group = c.benchmark_group("event_serialization");
-    for size in (0..10).map(|i| i * 1024) {
+
+    for size in (0..32).map(|i| i * 1024) {
        group.throughput(Throughput::Bytes(size));
        group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, &size| {
-            b.iter(|| black_box(create_event(size as usize)).into_bytes())
+            b.iter(|| black_box(create_event(size as usize)).into_bytes().unwrap())
        });
    }
    group.finish();