defend against uncertain cursor semantics

MaterializeInc · Aug 4, 2021 · b0df756 · b0df756
1 parent dc76793
commit b0df756
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/dataflow/src/render/context.rs b/src/dataflow/src/render/context.rs
@@ -526,7 +526,9 @@ impl<K: PartialEq, V, T: Timestamp, R, C: Cursor<K, V, T, R>> PendingWork<K, V,
         let mut work: usize = 0;
         let mut session = output.session(&self.capability);
         if let Some(key) = key {
-            self.cursor.seek_key(&self.batch, key);
+            if self.cursor.get_key(&self.batch) != Some(key) {
+                self.cursor.seek_key(&self.batch, key);
+            }
             if self.cursor.get_key(&self.batch) == Some(key) {
                 while let Some(val) = self.cursor.get_val(&self.batch) {
                     self.cursor.map_times(&self.batch, |time, diff| {