fix: solve tls drop panic when task queue drop (#257)

bytedance · Mar 20, 2024 · c677de3 · c677de3
1 parent a1ce0ef
commit c677de3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/monoio/src/task/harness.rs b/monoio/src/task/harness.rs
@@ -180,7 +180,7 @@ where
                         ctx.unpark_thread(owner_id);
                     }
                     None => {
-                        crate::runtime::DEFAULT_CTX.with(|default_ctx| {
+                        let _ = crate::runtime::DEFAULT_CTX.try_with(|default_ctx| {
                             crate::runtime::CURRENT.set(default_ctx, || {
                                 crate::runtime::CURRENT.with(|ctx| {
                                     ctx.send_waker(owner_id, waker);
@@ -237,7 +237,7 @@ where
                         ctx.unpark_thread(owner_id);
                     }
                     None => {
-                        crate::runtime::DEFAULT_CTX.with(|default_ctx| {
+                        let _ = crate::runtime::DEFAULT_CTX.try_with(|default_ctx| {
                             crate::runtime::CURRENT.set(default_ctx, || {
                                 crate::runtime::CURRENT.with(|ctx| {
                                     ctx.send_waker(owner_id, waker);