From c572f0b95eb2c878bbf7ac95efecb9b904126719 Mon Sep 17 00:00:00 2001
From: GloopShlugger <88001854+GloopShlugger@users.noreply.github.com>
Date: Thu, 31 Aug 2023 00:31:51 +0200
Subject: [PATCH] add pot to benchmarks (#53)

Co-authored-by: GloopShlugger <Olli>
---
 Cargo.toml       |  2 ++
 benches/bench.rs | 14 ++++++++++++++
 src/bench_pot.rs | 33 +++++++++++++++++++++++++++++++++
 src/lib.rs       |  2 ++
 4 files changed, 51 insertions(+)
 create mode 100644 src/bench_pot.rs

diff --git a/Cargo.toml b/Cargo.toml
index f29add6..8e73d27 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -70,6 +70,7 @@ nanoserde = {version = "0.1.33", optional = true }
 parity-scale-codec = { version = "3.5.0", features = ["full"], optional = true }
 parity-scale-codec-derive = { version = "3.1.4", optional = true }
 postcard = { version = "1.0.4", features = ["alloc"], optional = true }
+pot = { version = "3.0.0", optional = true }
 prost = { version = "0.11.9", optional = true }
 rand = "0.8.5"
 rkyv = { version = "0.7.41", features = ["validation"], optional = true }
@@ -107,6 +108,7 @@ default = [
   "nanoserde",
   "scale",
   "postcard",
+  "pot",
   "prost",
   "prost-build",
   "rkyv",
diff --git a/benches/bench.rs b/benches/bench.rs
index 08ecad4..0c3cc70 100644
--- a/benches/bench.rs
+++ b/benches/bench.rs
@@ -30,6 +30,8 @@ use rust_serialization_benchmark::bench_nanoserde;
 use rust_serialization_benchmark::bench_parity_scale_codec;
 #[cfg(feature = "postcard")]
 use rust_serialization_benchmark::bench_postcard;
+#[cfg(feature = "pot")]
+use rust_serialization_benchmark::bench_pot;
 #[cfg(feature = "prost")]
 use rust_serialization_benchmark::bench_prost;
 #[cfg(feature = "rkyv")]
@@ -158,6 +160,9 @@ fn bench_log(c: &mut Criterion) {
     #[cfg(feature = "postcard")]
     bench_postcard::bench(BENCH, c, &data);
 
+    #[cfg(feature = "pot")]
+    bench_pot::bench(BENCH, c, &data);
+
     #[cfg(feature = "prost")]
     bench_prost::bench(BENCH, c, &data);
 
@@ -314,6 +319,9 @@ fn bench_mesh(c: &mut Criterion) {
     #[cfg(feature = "postcard")]
     bench_postcard::bench(BENCH, c, &data);
 
+    #[cfg(feature = "pot")]
+    bench_pot::bench(BENCH, c, &data);
+
     #[cfg(feature = "prost")]
     bench_prost::bench(BENCH, c, &data);
 
@@ -461,6 +469,9 @@ fn bench_minecraft_savedata(c: &mut Criterion) {
     #[cfg(feature = "postcard")]
     bench_postcard::bench(BENCH, c, &data);
 
+    #[cfg(feature = "pot")]
+    bench_pot::bench(BENCH, c, &data);
+
     #[cfg(feature = "prost")]
     bench_prost::bench(BENCH, c, &data);
 
@@ -612,6 +623,9 @@ fn bench_mk48(c: &mut Criterion) {
     #[cfg(feature = "postcard")]
     bench_postcard::bench(BENCH, c, &data);
 
+    #[cfg(feature = "pot")]
+    bench_pot::bench(BENCH, c, &data);
+
     #[cfg(feature = "prost")]
     bench_prost::bench(BENCH, c, &data);
 
diff --git a/src/bench_pot.rs b/src/bench_pot.rs
new file mode 100644
index 0000000..7412108
--- /dev/null
+++ b/src/bench_pot.rs
@@ -0,0 +1,33 @@
+use criterion::{black_box, Criterion};
+use serde::{Deserialize, Serialize};
+
+pub fn bench<T>(name: &'static str, c: &mut Criterion, data: &T)
+where
+    T: Serialize + for<'de> Deserialize<'de>,
+{
+    const BUFFER_LEN: usize = 70_000_000;
+
+    let mut group = c.benchmark_group(format!("{}/pot", name));
+
+    let mut serialize_buffer = vec![0; BUFFER_LEN];
+    group.bench_function("serialize", |b| {
+        b.iter(|| {
+            black_box(
+                pot::to_writer(black_box(&data), black_box(serialize_buffer.as_mut_slice()))
+                    .unwrap(),
+            );
+        })
+    });
+
+    let deserialize_buffer = pot::to_vec(&data).unwrap();
+
+    group.bench_function("deserialize", |b| {
+        b.iter(|| {
+            black_box(pot::from_slice::<T>(black_box(&deserialize_buffer)).unwrap());
+        })
+    });
+
+    crate::bench_size(name, "pot", deserialize_buffer.as_slice());
+
+    group.finish();
+}
diff --git a/src/lib.rs b/src/lib.rs
index 251d66d..08fca17 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -28,6 +28,8 @@ pub mod bench_nanoserde;
 pub mod bench_parity_scale_codec;
 #[cfg(feature = "postcard")]
 pub mod bench_postcard;
+#[cfg(feature = "pot")]
+pub mod bench_pot;
 #[cfg(feature = "prost")]
 pub mod bench_prost;
 #[cfg(feature = "rkyv")]