fix(runtime): forget op if no current runtime (#338)

* fix(runtime): forget op if no current runtime * feat(runtime): prepare for bug fix
compio-rs · Nov 21, 2024 · f7f0403 · f7f0403
1 parent 3628b0f
commit f7f0403
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/compio-runtime/Cargo.toml b/compio-runtime/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "compio-runtime"
-version = "0.6.0"
+version = "0.6.1"
 description = "High-level runtime for compio"
 categories = ["asynchronous"]
 keywords = ["async", "runtime"]

diff --git a/compio-runtime/src/runtime/op.rs b/compio-runtime/src/runtime/op.rs
@@ -38,7 +38,8 @@ impl<T: OpCode> Future for OpFlagsFuture<T> {
 impl<T: OpCode> Drop for OpFlagsFuture<T> {
     fn drop(&mut self) {
         if let Some(key) = self.key.take() {
-            Runtime::with_current(|r| r.cancel_op(key))
+            // If there's no runtime, it's OK to forget it.
+            Runtime::try_with_current(|r| r.cancel_op(key)).ok();
         }
     }
 }
diff --git a/compio-runtime/src/runtime/time.rs b/compio-runtime/src/runtime/time.rs
@@ -144,7 +144,8 @@ impl Future for TimerFuture {
 
 impl Drop for TimerFuture {
     fn drop(&mut self) {
-        Runtime::with_current(|r| r.cancel_timer(self.key));
+        // If there's no runtime, it's OK to forget it.
+        Runtime::try_with_current(|r| r.cancel_timer(self.key)).ok();
     }
 }