Enhanced remove_stream frag map unwrap in WorkerNodeManager

risingwavelabs · Feb 7, 2024 · 9873e9e · 9873e9e
1 parent 6004031
commit 9873e9e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/frontend/src/scheduler/worker_node_manager.rs b/src/frontend/src/scheduler/worker_node_manager.rs
@@ -220,8 +220,10 @@ impl WorkerNodeManager {
 
     pub fn remove_streaming_fragment_mapping(&self, fragment_id: &FragmentId) {
         let mut guard = self.inner.write().unwrap();
-        guard.streaming_fragment_vnode_mapping.remove(fragment_id);
-        // todo
+        guard
+            .streaming_fragment_vnode_mapping
+            .remove(fragment_id)
+            .unwrap();
     }
 
     /// Returns fragment's vnode mapping for serving.