Merge pull request #25326 from ggevay/compute-set_read_policy-fix

Fix `set_read_policy` also in the Compute controller
MaterializeInc · Feb 16, 2024 · d16a9fc · d16a9fc
2 parents de908fa + 6e4ea7c
commit d16a9fc
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/compute-client/src/controller/instance.rs b/src/compute-client/src/controller/instance.rs
@@ -1195,10 +1195,11 @@ where
             let old_capability = &collection.implied_capability;
             let new_capability = new_policy.frontier(collection.write_frontier.borrow());
             if PartialOrder::less_than(old_capability, &new_capability) {
-                let mut update = ChangeBatch::new();
-                update.extend(old_capability.iter().map(|t| (t.clone(), -1)));
-                update.extend(new_capability.iter().map(|t| (t.clone(), 1)));
-                read_capability_changes.insert(id, update);
+                let entry = read_capability_changes
+                    .entry(id)
+                    .or_insert_with(ChangeBatch::new);
+                entry.extend(old_capability.iter().map(|t| (t.clone(), -1)));
+                entry.extend(new_capability.iter().map(|t| (t.clone(), 1)));
                 collection.implied_capability = new_capability;
             }