Fix off-by-one capacity error when sizing target containers (#450)

Signed-off-by: Moritz Hoffmann <[email protected]>
TimelyDataflow · Dec 20, 2023 · 8c23284 · 8c23284
1 parent d25f911
commit 8c23284
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git 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
@@ -86,6 +86,10 @@ where
                                 vals += 1;
                             }
                             upds += 1;
+                        } else {
+                            keys += 1;
+                            vals += 1;
+                            upds += 1;
                         }
                         prev_keyval = Some((key, val));
                     }