Add par_iter_mut shim

shssoichiro · May 23, 2023 · 5eb3ee0 · 5eb3ee0
1 parent 67f93db
commit 5eb3ee0
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/rayon.rs b/src/rayon.rs
@@ -26,6 +26,14 @@ pub trait IntoParallelRefIterator<'data> {
  fn par_iter(&'data self) -> Self::Iter;
 }
 
+pub trait IntoParallelRefMutIterator<'data> {
+ type Iter: ParallelIterator<Item = Self::Item>;
+ type Item: Send + 'data;
+
+ // Required method
+ fn par_iter_mut(&'data mut self) -> Self::Iter;
+}
+
 impl<I: IntoIterator> IntoParallelIterator for I
 where
  I::Item: Send,
@@ -50,6 +58,18 @@ where
  }
 }
 
+impl<'data, I: 'data + ?Sized> IntoParallelRefMutIterator<'data> for I
+where
+ &'data mut I: IntoParallelIterator,
+{
+ type Iter = <&'data mut I as IntoParallelIterator>::Iter;
+ type Item = <&'data mut I as IntoParallelIterator>::Item;
+
+ fn par_iter_mut(&'data mut self) -> Self::Iter {
+ self.into_par_iter()
+ }
+}
+
 impl<I: Iterator> ParallelIterator for I {}
 
 #[allow(dead_code)]