Add MappedFutures

Author: StoicDeveloper

futures-util/Cargo.toml

@@ -48,6 +48,7 @@ spin = { version = "0.10.0", optional = true }
 futures = { path = "../futures", features = ["async-await", "thread-pool"] }
 futures-test = { path = "../futures-test" }
 tokio = "0.1.11"
+futures-timer = "3.0.3"
 
 [package.metadata.docs.rs]
 all-features = true

futures-util/src/stream/futures_unordered/futures_mapped/mod.rs

@@ -0,0 +1,23 @@
+//! An unbounded map of futures.
+//!
+//! This module is only available when the `std` or `alloc` feature of this
+//! library is activated, and it is activated by default.
+
+use crate::task::AtomicWaker;
+use alloc::sync::{Arc, Weak};
+use core::cell::UnsafeCell;
+use core::fmt::{self, Debug};
+use core::iter::FromIterator;
+use core::marker::PhantomData;
+use core::mem;
+use core::pin::Pin;
+use core::ptr;
+use core::sync::atomic::Ordering::{AcqRel, Acquire, Relaxed, Release, SeqCst};
+use core::sync::atomic::{AtomicBool, AtomicPtr};
+use futures_core::future::Future;
+use futures_core::stream::{FusedStream, Stream};
+use futures_core::task::{Context, Poll};
+use futures_task::{FutureObj, LocalFutureObj, LocalSpawn, Spawn, SpawnError};
+use std::collections::{HashMap, HashSet};
+
+use super::FuturesUnordered;

futures-util/src/stream/futures_unordered/iter.rs

@@ -1,5 +1,6 @@
 use super::task::Task;
 use super::FuturesUnordered;
+use core::hash::Hash;
 use core::marker::PhantomData;
 use core::pin::Pin;
 use core::ptr;
@@ -13,6 +14,19 @@ pub struct IterPinMut<'a, Fut> {
     pub(super) _marker: PhantomData<&'a mut FuturesUnordered<Fut>>,
 }
 
+/// Mutable iterator over all keys and futures in the map.
+// #[derive(Debug)]
+// pub struct MapIterPinMut<'a, K, Fut> {
+//     pub(super) inner: IterMut<'a, K, Fut>,
+//     // pub(super) task: *const Task<Fut>,
+//     // pub(super) len: usize,
+//     // pub(super) _marker: PhantomData<&'a mut FuturesUnordered<Fut>>,
+// }
+
+/// Immutable iterator over all the keys and futures in the map.
+// #[derive(Debug)]
+// pub struct IterMap<'a, K: Hash + Eq, Fut: Unpin>(pub(super) IterPinRef<'a, K, Fut>);
+
 /// Mutable iterator over all futures in the unordered set.
 #[derive(Debug)]
 pub struct IterMut<'a, Fut: Unpin>(pub(super) IterPinMut<'a, Fut>);
@@ -26,6 +40,20 @@ pub struct IterPinRef<'a, Fut> {
     pub(super) _marker: PhantomData<&'a FuturesUnordered<Fut>>,
 }
 
+/// Immutable iterator over all keys and futures in the map.
+// #[derive(Debug)]
+// pub struct MapIterPinRef<'a, K, Fut> {
+//     inner: HashMapIter<'a, K, *const Task<Fut>>,
+//     // pub(super) task: *const Task<Fut>,
+//     // pub(super) len: usize,
+//     // pub(super) pending_next_all: *mut Task<Fut>,
+//     // pub(super) _marker: PhantomData<&'a FuturesUnordered<Fut>>,
+// }
+
+/// Immutable iterator over all keys and futures in the map.
+// #[derive(Debug)]
+// pub struct MapIter<'a, K, Fut>(pub(super) MapIterPinRef<'a, K, Fut>);
+
 /// Immutable iterator over all the futures in the unordered set.
 #[derive(Debug)]
 pub struct Iter<'a, Fut: Unpin>(pub(super) IterPinRef<'a, Fut>);