free spawn function (#343)

* feat: add spawn() free function using TelemetryHandle::current

* feat: add spawn() free function using TelemetryHandle::current

Author: jlizen

dial9-tokio-telemetry/src/lib.rs

@@ -39,4 +39,4 @@ pub use current_config::{
     Dial9Config, Dial9ConfigBuilder, Dial9ConfigBuilderError, ValidationError,
 };
 pub use dial9_macro::main;
-pub use telemetry::{TelemetryRuntimeError, TracedRuntime};
+pub use telemetry::{TelemetryRuntimeError, TracedRuntime, spawn};

dial9-tokio-telemetry/src/telemetry/mod.rs

@@ -25,7 +25,7 @@ pub use format::{
 pub use recorder::{
     HasTracePath, NoTracePath, RuntimeTelemetryHandle, TelemetryCore, TelemetryCoreBuilder,
     TelemetryGuard, TelemetryHandle, TelemetryRuntimeError, TraceRuntimeCoreBuilder, TracedRuntime,
-    TracedRuntimeBuilder, current_worker_id,
+    TracedRuntimeBuilder, current_worker_id, spawn,
 };
 pub use task_metadata::{TaskId, UNKNOWN_TASK_ID};
 pub use writer::{NullWriter, RotatingWriter, TraceWriter};

dial9-tokio-telemetry/src/telemetry/recorder/mod.rs

@@ -513,6 +513,27 @@ impl TelemetryHandle {
     }
 }
 
+/// Spawn a traced task on the current tokio runtime.
+///
+/// Like [`tokio::spawn`], but wraps the future with wake-event tracking
+/// when called from a thread owned by a dial9 runtime. On other threads,
+/// falls back to plain [`tokio::spawn`].
+///
+/// Equivalent to [`TelemetryHandle::current().spawn(future)`](TelemetryHandle::spawn).
+///
+/// # Panics
+///
+/// Panics if called from outside a tokio runtime context (same
+/// as [`tokio::spawn`]).
+#[track_caller]
+pub fn spawn<F>(future: F) -> tokio::task::JoinHandle<F::Output>
+where
+    F: std::future::Future + Send + 'static,
+    F::Output: Send + 'static,
+{
+    TelemetryHandle::current().spawn(future)
+}
+
 /// RAII guard that sets `INSTRUMENTED_SPAWN` to `true` on creation and
 /// resets it to `false` on drop, even if `tokio::spawn` panics.
 struct InstrumentedSpawnGuard;