add Runtime::handle

Signed-off-by: Runji Wang <wangrunji0408@163.com>

CHANGELOG.md

@@ -6,6 +6,12 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## tokio [0.2.27] - 2024-06-05
+
+### Added
+
+- Add `Runtime::handle()`.
+
 ## tokio [0.2.26] - 2024-05-27
 
 ### Added

madsim-tokio/src/sim/runtime.rs

@@ -1,3 +1,5 @@
+// FIXME: create a dedicated handle for tokio runtime instead of reusing madsim handle.
+//        tasks spawned with this handle are not correctly associated with the tokio runtime.
 pub use madsim::runtime::Handle;
 use madsim::task::{AbortHandle, JoinHandle};
 use spin::Mutex;
@@ -42,13 +44,15 @@ impl Builder {
     pub fn build(&mut self) -> io::Result<Runtime> {
         Ok(Runtime {
             abort_handles: Default::default(),
+            handle: Handle::current(),
         })
     }
 }
 
 /// A fake Tokio runtime.
 pub struct Runtime {
     abort_handles: Mutex<Vec<AbortHandle>>,
+    handle: Handle,
 }
 
 #[allow(dead_code)]
@@ -81,6 +85,11 @@ impl Runtime {
         // Madsim runtime is entered by default. No-op here.
         EnterGuard(self)
     }
+
+    /// Returns a handle to the runtime’s spawner.
+    pub fn handle(&self) -> &Handle {
+        &self.handle
+    }
 }
 
 impl Drop for Runtime {