From db2ef58e4b0f61684ac137529b59c3bdbb1d3b31 Mon Sep 17 00:00:00 2001
From: Moritz Hoffmann <antiguru@gmail.com>
Date: Wed, 20 Dec 2023 16:18:24 -0500
Subject: [PATCH] Fix off-by-one capacity error when sizing target containers

Signed-off-by: Moritz Hoffmann <antiguru@gmail.com>
---
 src/trace/implementations/merge_batcher.rs     | 4 ++++
 src/trace/implementations/merge_batcher_col.rs | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/src/trace/implementations/merge_batcher.rs b/src/trace/implementations/merge_batcher.rs
index 2c0feec3f..680aa5306 100644
--- a/src/trace/implementations/merge_batcher.rs
+++ b/src/trace/implementations/merge_batcher.rs
@@ -82,6 +82,10 @@ where
                                 vals += 1;
                             }
                             upds += 1;
+                        } else {
+                            keys += 1;
+                            vals += 1;
+                            upds += 1;
                         }
                         prev_keyval = Some((key, val));
                     }
diff --git a/src/trace/implementations/merge_batcher_col.rs b/src/trace/implementations/merge_batcher_col.rs
index fcd9f5ff8..9c2faab12 100644
--- a/src/trace/implementations/merge_batcher_col.rs
+++ b/src/trace/implementations/merge_batcher_col.rs
@@ -86,6 +86,10 @@ where
                                 vals += 1;
                             }
                             upds += 1;
+                        } else {
+                            keys += 1;
+                            vals += 1;
+                            upds += 1;
                         }
                         prev_keyval = Some((key, val));
                     }