Export task-center task names to tokio

restatedev · May 1, 2024 · 0dd5cd1 · 0dd5cd1
1 parent ec3b7bb
commit 0dd5cd1
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();