fix: access TASKS in block_in_place

neondatabase · Nov 28, 2022 · afe537f · afe537f
1 parent 376e5af
commit afe537f
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/pageserver/src/task_mgr.rs b/pageserver/src/task_mgr.rs
@@ -335,11 +335,13 @@ async fn task_finish(
     shutdown_process_on_error: bool,
 ) {
     // Remove our entry from the global hashmap.
-    let task = TASKS
-        .lock()
-        .unwrap()
-        .remove(&task_id)
-        .expect("no task in registry");
+    let task = tokio::task::block_in_place(|| {
+        TASKS
+            .lock()
+            .unwrap()
+            .remove(&task_id)
+            .expect("no task in registry")
+    });
 
     let mut shutdown_process = false;
     {