feat: select! macro for no-std environments

futures-util/src/async_await/mod.rs

@@ -34,10 +34,8 @@ mod stream_select_mod;
 #[cfg(feature = "async-await-macro")]
 pub use self::stream_select_mod::*;
 
-#[cfg(feature = "std")]
 #[cfg(feature = "async-await-macro")]
 mod random;
-#[cfg(feature = "std")]
 #[cfg(feature = "async-await-macro")]
 pub use self::random::*;
 

futures-util/src/async_await/random.rs

@@ -1,11 +1,3 @@
-use std::{
-    cell::Cell,
-    collections::hash_map::DefaultHasher,
-    hash::Hasher,
-    num::Wrapping,
-    sync::atomic::{AtomicUsize, Ordering},
-};
-
 // Based on [Fisher–Yates shuffle].
 //
 // [Fisher–Yates shuffle]: https://en.wikipedia.org/wiki/Fisher–Yates_shuffle
@@ -24,7 +16,16 @@ fn gen_index(n: usize) -> usize {
 /// Pseudorandom number generator based on [xorshift*].
 ///
 /// [xorshift*]: https://en.wikipedia.org/wiki/Xorshift#xorshift*
+#[cfg(feature = "std")]
 fn random() -> u64 {
+    use std::{
+        cell::Cell,
+        collections::hash_map::DefaultHasher,
+        hash::Hasher,
+        num::Wrapping,
+        sync::atomic::{AtomicUsize, Ordering},
+    };
+
     std::thread_local! {
         static RNG: Cell<Wrapping<u64>> = Cell::new(Wrapping(prng_seed()));
     }
@@ -52,3 +53,26 @@ fn random() -> u64 {
         x.0.wrapping_mul(0x2545_f491_4f6c_dd1d)
     })
 }
+
+#[cfg(not(feature = "std"))]
+fn random() -> u64 {
+    use core::sync::atomic::{AtomicUsize, Ordering};
+
+    static RNG: AtomicUsize = AtomicUsize::new(1);
+
+    let mut x = RNG.load(Ordering::Relaxed);
+
+    if core::mem::size_of::<usize>() == 4 {
+        x ^= x << 13;
+        x ^= x >> 17;
+        x ^= x << 5;
+    } else if core::mem::size_of::<usize>() == 8 {
+        x ^= x >> 12;
+        x ^= x << 25;
+        x ^= x >> 27;
+        x = x.wrapping_mul(0x2545_f491_4f6c_dd1d);
+    }
+
+    RNG.store(x, Ordering::Relaxed);
+    x as u64
+}

futures-util/src/async_await/select_mod.rs

@@ -305,15 +305,13 @@ macro_rules! document_select_macro {
     };
 }
 
-#[cfg(feature = "std")]
 #[doc(hidden)]
 pub use futures_macro::select_internal;
 
 #[doc(hidden)]
 pub use futures_macro::select_biased_internal;
 
 document_select_macro! {
-    #[cfg(feature = "std")]
     #[macro_export]
     macro_rules! select {
         ($($tokens:tt)*) => {{

futures/src/lib.rs

@@ -120,7 +120,6 @@ pub use futures_util::{AsyncBufReadExt, AsyncReadExt, AsyncSeekExt, AsyncWriteEx
 // Macro reexports
 pub use futures_core::ready; // Readiness propagation
 pub use futures_util::pin_mut;
-#[cfg(feature = "std")]
 #[cfg(feature = "async-await")]
 pub use futures_util::select;
 #[cfg(feature = "async-await")]