Export task-center task names to tokio

restatedev · May 2, 2024 · 2f71abc · 2f71abc
1 parent 9afb910
commit 2f71abc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/crates/core/Cargo.toml b/crates/core/Cargo.toml
@@ -40,7 +40,7 @@ static_assertions = { workspace = true }
 strum = { workspace = true }
 strum_macros = { workspace = true }
 thiserror = { workspace = true }
-tokio = { workspace = true }
+tokio = { workspace = true, features = ["tracing" ] }
 tokio-stream = { workspace = true }
 tokio-util = { workspace = true }
 tracing = { workspace = true }

diff --git a/crates/core/src/task_center.rs b/crates/core/src/task_center.rs
@@ -225,15 +225,19 @@ impl TaskCenter {
         let mut handle_mut = task.join_handle.lock().unwrap();
 
         let task_cloned = Arc::clone(&task);
-        let join_handle = inner.default_runtime_handle.spawn(wrapper(
+        let tokio_task = tokio::task::Builder::new().name(name);
+        let fut = wrapper(
             self.clone(),
             id,
             kind,
             task_cloned,
             cancel,
             metadata,
             future,
-        ));
+        );
+        let join_handle = tokio_task
+            .spawn_on(fut, &inner.default_runtime_handle)
+            .expect("default runtime can spawn tasks");
         *handle_mut = Some(join_handle);
         drop(handle_mut);
         let kind_str: &'static str = kind.into();