diff options
Diffstat (limited to 'third_party/rust/dashmap/src/iter_set.rs')
-rw-r--r-- | third_party/rust/dashmap/src/iter_set.rs | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/third_party/rust/dashmap/src/iter_set.rs b/third_party/rust/dashmap/src/iter_set.rs new file mode 100644 index 0000000000..619a5b500e --- /dev/null +++ b/third_party/rust/dashmap/src/iter_set.rs @@ -0,0 +1,71 @@ +use crate::setref::multiple::RefMulti; +use crate::t::Map; +use core::hash::{BuildHasher, Hash}; + +pub struct OwningIter<K, S> { + inner: crate::iter::OwningIter<K, (), S>, +} + +impl<K: Eq + Hash, S: BuildHasher + Clone> OwningIter<K, S> { + pub(crate) fn new(inner: crate::iter::OwningIter<K, (), S>) -> Self { + Self { inner } + } +} + +impl<K: Eq + Hash, S: BuildHasher + Clone> Iterator for OwningIter<K, S> { + type Item = K; + + fn next(&mut self) -> Option<Self::Item> { + self.inner.next().map(|(k, _)| k) + } +} + +unsafe impl<K, S> Send for OwningIter<K, S> +where + K: Eq + Hash + Send, + S: BuildHasher + Clone + Send, +{ +} + +unsafe impl<K, S> Sync for OwningIter<K, S> +where + K: Eq + Hash + Sync, + S: BuildHasher + Clone + Sync, +{ +} + +pub struct Iter<'a, K, S, M> { + inner: crate::iter::Iter<'a, K, (), S, M>, +} + +unsafe impl<'a, 'i, K, S, M> Send for Iter<'i, K, S, M> +where + K: 'a + Eq + Hash + Send, + S: 'a + BuildHasher + Clone, + M: Map<'a, K, (), S>, +{ +} + +unsafe impl<'a, 'i, K, S, M> Sync for Iter<'i, K, S, M> +where + K: 'a + Eq + Hash + Sync, + S: 'a + BuildHasher + Clone, + M: Map<'a, K, (), S>, +{ +} + +impl<'a, K: Eq + Hash, S: 'a + BuildHasher + Clone, M: Map<'a, K, (), S>> Iter<'a, K, S, M> { + pub(crate) fn new(inner: crate::iter::Iter<'a, K, (), S, M>) -> Self { + Self { inner } + } +} + +impl<'a, K: Eq + Hash, S: 'a + BuildHasher + Clone, M: Map<'a, K, (), S>> Iterator + for Iter<'a, K, S, M> +{ + type Item = RefMulti<'a, K, S>; + + fn next(&mut self) -> Option<Self::Item> { + self.inner.next().map(RefMulti::new) + } +} |