[catnap] Enhancement: Try tracking of pending_ops for async_close()

microsoft · Nov 30, 2023 · acbb42b · acbb42b
1 parent dbd0a32
commit acbb42b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/rust/catnap/mod.rs b/src/rust/catnap/mod.rs
@@ -317,8 +317,10 @@ impl SharedCatnapLibOS {
         let coroutine_constructor = || -> Result<TaskHandle, Fail> {
             let task_name: String = format!("Catnap::close for qd={:?}", qd);
             let yielder: Yielder = Yielder::new();
+            let yielder_handle = yielder.get_handle();
             let coroutine: Pin<Box<Operation>> = Box::pin(self.clone().close_coroutine(qd, yielder));
-            self.runtime.insert_coroutine(&task_name, coroutine)
+            self.runtime
+                .insert_coroutine_with_tracking(&task_name, coroutine, yielder_handle, qd)
         };
 
         queue.async_close(coroutine_constructor)