handle the case of set_current_task_id being called after CONTEXT is …

…destroyed
tokio-rs · Nov 9, 2022 · a6b34b9 · a6b34b9
1 parent b6cae2e
commit a6b34b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tokio/src/runtime/context.rs b/tokio/src/runtime/context.rs
@@ -105,7 +105,7 @@ cfg_rt! {
     pub(crate) struct DisallowBlockInPlaceGuard(bool);
 
     pub(crate) fn set_current_task_id(id: Option<Id>) {
-        CONTEXT.with(|ctx| ctx.current_task_id.replace(id));
+        let _ = CONTEXT.try_with(|ctx| ctx.current_task_id.replace(id));
     }
 
     #[track_caller]