Add framing benches

sv2/framing-sv2/Cargo.toml

@@ -19,12 +19,18 @@ buffer_sv2 = { path = "../buffer-sv2", optional=true, version = "^2.0.0" }
 noise_sv2 = { path = "../noise-sv2", version = "^1.0.0" }
 
 [dev-dependencies]
-noise_sv2 = { path = "../noise-sv2", version = "^1.0.0" }
+criterion = { version = "0.3.0", features = ["html_reports"] }
 rand = "0.8.3"
 secp256k1 = { version = "0.28.2", default-features = false, features =["alloc","rand","rand-std"] }
+rayon-core = "=1.12.1"
 
 [features]
 with_buffer_pool = ["binary_sv2/with_buffer_pool", "buffer_sv2"]
 
 [package.metadata.docs.rs]
 features = ["with_buffer_pool"]
+
+
+[[bench]]
+name = "framing"
+harness = false

sv2/framing-sv2/benches/framing.rs

@@ -0,0 +1,117 @@
+use criterion::{black_box, criterion_group, criterion_main, BenchmarkId, Criterion};
+use framing_sv2::framing::Sv2Frame;
+
+#[cfg(not(feature = "with_buffer_pool"))]
+type Slice = Vec<u8>;
+
+#[cfg(feature = "with_buffer_pool")]
+type Slice = buffer_sv2::Slice;
+
+const PAYLOAD_SIZES: &[usize] = &[64, 1024, 16 * 1024, 60 * 1024];
+
+#[cfg(feature = "with_buffer_pool")]
+const BACKEND: &str = "buffer_pool";
+
+#[cfg(not(feature = "with_buffer_pool"))]
+const BACKEND: &str = "vec";
+
+fn payload(size: usize) -> Vec<u8> {
+    vec![2u8; size]
+}
+
+fn bench_from_message(c: &mut Criterion) {
+    let mut group = c.benchmark_group(format!("sv2frame::from_message::{BACKEND}"));
+
+    for &size in PAYLOAD_SIZES {
+        group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, &size| {
+            b.iter(|| {
+                Sv2Frame::<Vec<u8>, Slice>::from_message(black_box(payload(size)), 1, 0, false)
+                    .unwrap()
+            })
+        });
+    }
+
+    group.finish();
+}
+
+fn bench_serialize_fresh(c: &mut Criterion) {
+    let mut group = c.benchmark_group(format!("sv2frame::serialize_fresh::{BACKEND}"));
+
+    for &size in PAYLOAD_SIZES {
+        let frame = Sv2Frame::<Vec<u8>, Slice>::from_message(payload(size), 1, 0, false).unwrap();
+
+        let mut buf = vec![0u8; frame.encoded_length()];
+
+        group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, _| {
+            b.iter(|| {
+                frame.clone().serialize(black_box(&mut buf)).unwrap();
+            })
+        });
+    }
+
+    group.finish();
+}
+
+fn bench_serialize_fast(c: &mut Criterion) {
+    let mut group = c.benchmark_group(format!("sv2frame::serialize_fast::{BACKEND}"));
+
+    for &size in PAYLOAD_SIZES {
+        let frame = Sv2Frame::<Vec<u8>, Slice>::from_message(payload(size), 1, 0, false).unwrap();
+
+        let mut buf = vec![0u8; frame.encoded_length()];
+        frame.clone().serialize(&mut buf).unwrap();
+
+        group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, _| {
+            b.iter(|| {
+                frame.clone().serialize(black_box(&mut buf)).unwrap();
+            })
+        });
+    }
+
+    group.finish();
+}
+
+fn bench_from_bytes(c: &mut Criterion) {
+    let mut group = c.benchmark_group(format!("sv2frame::from_bytes::{BACKEND}"));
+
+    for &size in PAYLOAD_SIZES {
+        let frame = Sv2Frame::<Vec<u8>, Slice>::from_message(payload(size), 1, 0, false).unwrap();
+
+        let mut buf = vec![0u8; frame.encoded_length()];
+        frame.serialize(&mut buf).unwrap();
+
+        group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, _| {
+            b.iter(|| Sv2Frame::<Vec<u8>, _>::from_bytes(black_box(buf.clone())).unwrap())
+        });
+    }
+
+    group.finish();
+}
+
+fn bench_size_hint(c: &mut Criterion) {
+    let mut group = c.benchmark_group(format!("sv2frame::size_hint::{BACKEND}"));
+
+    for &size in PAYLOAD_SIZES {
+        let frame = Sv2Frame::<Vec<u8>, Slice>::from_message(payload(size), 1, 0, false).unwrap();
+
+        let mut buf = vec![0u8; frame.encoded_length()];
+        frame.serialize(&mut buf).unwrap();
+
+        group.bench_with_input(BenchmarkId::from_parameter(size), &size, |b, _| {
+            b.iter(|| black_box(Sv2Frame::<Vec<u8>, Slice>::size_hint(&buf)))
+        });
+    }
+
+    group.finish();
+}
+
+criterion_group!(
+    framing,
+    bench_from_message,
+    bench_serialize_fresh,
+    bench_serialize_fast,
+    bench_from_bytes,
+    bench_size_hint,
+);
+
+criterion_main!(framing);