feat(runtime): enable current_thread_id on nightly (#548)

Author: Berrysoft

compio-runtime/Cargo.toml

@@ -67,6 +67,9 @@ time = []
 # Enable it to always notify the driver when a task schedules.
 notify-always = []
 
+current_thread_id = []
+nightly = ["current_thread_id"]
+
 [[test]]
 name = "custom_loop"
 required-features = ["event", "time"]

compio-runtime/src/lib.rs

@@ -9,6 +9,7 @@
 //! ```
 
 #![cfg_attr(docsrs, feature(doc_cfg))]
+#![cfg_attr(feature = "current_thread_id", feature(current_thread_id))]
 #![warn(missing_docs)]
 
 mod affinity;

compio-runtime/src/runtime/send_wrapper.rs

@@ -7,16 +7,31 @@
 // option. This file may not be copied, modified, or distributed
 // except according to those terms.
 
+#[cfg(feature = "current_thread_id")]
+use std::thread::current_id;
 use std::{
-    cell::Cell,
     mem::{self, ManuallyDrop},
     thread::{self, ThreadId},
 };
 
-thread_local! {
-    static THREAD_ID: Cell<ThreadId> = Cell::new(thread::current().id());
+#[cfg(not(feature = "current_thread_id"))]
+mod imp {
+    use std::{
+        cell::Cell,
+        thread::{self, ThreadId},
+    };
+    thread_local! {
+        static THREAD_ID: Cell<ThreadId> = Cell::new(thread::current().id());
+    }
+
+    pub fn current_id() -> ThreadId {
+        THREAD_ID.get()
+    }
 }
 
+#[cfg(not(feature = "current_thread_id"))]
+use imp::current_id;
+
 /// A wrapper that copied from `send_wrapper` crate, with our own optimizations.
 pub struct SendWrapper<T> {
     data: ManuallyDrop<T>,
@@ -30,15 +45,15 @@ impl<T> SendWrapper<T> {
     pub fn new(data: T) -> SendWrapper<T> {
         SendWrapper {
             data: ManuallyDrop::new(data),
-            thread_id: THREAD_ID.get(),
+            thread_id: current_id(),
         }
     }
 
     /// Returns `true` if the value can be safely accessed from within the
     /// current thread.
     #[inline]
     pub fn valid(&self) -> bool {
-        self.thread_id == THREAD_ID.get()
+        self.thread_id == current_id()
     }
 
     /// Returns a reference to the contained value.

compio/Cargo.toml

@@ -121,6 +121,7 @@ ring = ["compio-tls?/ring", "compio-quic?/ring", "compio-ws?/ring"]
 
 # Nightly features
 allocator_api = ["compio-buf/allocator_api", "compio-io?/allocator_api"]
+current_thread_id = ["compio-runtime/current_thread_id"]
 lazy_cell = ["compio-signal?/lazy_cell"]
 linux_pidfd = ["compio-process?/linux_pidfd"]
 once_cell_try = [
@@ -139,6 +140,7 @@ try_trait_v2 = ["compio-buf/try_trait_v2"]
 windows_by_handle = ["compio-fs?/windows_by_handle"]
 nightly = [
     "allocator_api",
+    "current_thread_id",
     "lazy_cell",
     "linux_pidfd",
     "once_cell_try",