store: Reset the end of VidBatcher when changing the size

lutter · lutter · commit f3cbad21580c · 2025-04-01T18:10:49.000-07:00
diff --git a/store/postgres/src/vid_batcher.rs b/store/postgres/src/vid_batcher.rs
@@ -237,6 +237,10 @@ impl VidBatcher {
 
     pub(crate) fn set_batch_size(&mut self, size: usize) {
         self.batch_size.size = size as i64;
+        self.end = match &self.ogive {
+            Some(ogive) => ogive.next_point(self.start, size as usize).unwrap(),
+            None => self.start + size as i64,
+        };
     }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -237,6 +237,10 @@ impl VidBatcher {`
`237`	`237`
`238`	`238`	`pub(crate) fn set_batch_size(&mut self, size: usize) {`
`239`	`239`	`self.batch_size.size = size as i64;`
	`240`	`+ self.end = match &self.ogive {`
	`241`	`+ Some(ogive) => ogive.next_point(self.start, size as usize).unwrap(),`
	`242`	`+ None => self.start + size as i64,`
	`243`	`+ };`
`240`	`244`	`}`
`241`	`245`	`}`
`242`	`246`