Fix integer overflow in bench for total size calculation (#1474)

apache · Jan 30, 2025 · 6ee955a · 6ee955a
1 parent 8534815
commit 6ee955a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/bench/report/Cargo.toml b/bench/report/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "iggy-benchmark-report"
-version = "0.1.2"
+version = "0.1.3"
 edition = "2021"
 description = "Benchmark report and chart generation library for iggy-bench binary and iggy-benchmarks-dashboard web app"
 license = "Apache-2.0"

diff --git a/bench/report/src/prints.rs b/bench/report/src/prints.rs
@@ -9,8 +9,9 @@ use crate::{
 impl BenchmarkReport {
     pub fn print_summary(&self) {
         let kind = self.params.benchmark_kind;
-        let total_messages = self.params.messages_per_batch * self.params.message_batches;
-        let total_size_bytes = total_messages * self.params.message_size;
+        let total_messages: u64 =
+            self.params.messages_per_batch as u64 * self.params.message_batches as u64;
+        let total_size_bytes: u64 = total_messages * self.params.message_size as u64;
         let streams = self.params.streams;
         let messages_per_batch = self.params.messages_per_batch;
         let message_batches = self.params.message_batches;