Add a WorkerLocal abstraction

Author: Zoxc

src/librustc_data_structures/Cargo.toml

@@ -17,6 +17,7 @@ cfg-if = "0.1.2"
 stable_deref_trait = "1.0.0"
 parking_lot_core = "0.2.8"
 rustc-rayon = { git = "https://github.com/Zoxc/rayon.git", branch = "fiber" }
+rustc-rayon-core = { git = "https://github.com/Zoxc/rayon.git", branch = "fiber" }
 
 [dependencies.parking_lot]
 version = "0.5"

src/librustc_data_structures/lib.rs

@@ -45,6 +45,7 @@ extern crate parking_lot;
 extern crate cfg_if;
 extern crate stable_deref_trait;
 extern crate rustc_rayon as rayon;
+extern crate rustc_rayon_core as rayon_core;
 
 // See librustc_cratesio_shim/Cargo.toml for a comment explaining this.
 #[allow(unused_extern_crates)]

src/librustc_data_structures/sync.rs

@@ -100,6 +100,33 @@ cfg_if! {
 
         use std::cell::Cell;
 
+        #[derive(Debug)]
+        pub struct WorkerLocal<T>(OneThread<T>);
+
+        impl<T> WorkerLocal<T> {
+            /// Creates a new worker local where the `initial` closure computes the
+            /// value this worker local should take for each thread in the thread pool.
+            #[inline]
+            pub fn new<F: FnMut(usize) -> T>(mut initial: F) -> WorkerLocal<T> {
+                WorkerLocal(OneThread::new(f(0)))
+            }
+
+            /// Returns the worker-local value for each thread
+            #[inline]
+            pub fn into_inner(self) -> Vec<T> {
+                vec![self.0.into_inner()]
+            }
+        }
+
+        impl<T> Deref for WorkerLocal<T> {
+            type Target = T;
+
+            #[inline(always)]
+            fn deref(&self) -> &T {
+                &*self.0
+            }
+        }
+
         #[derive(Debug)]
         pub struct MTLock<T>(T);
 
@@ -203,6 +230,8 @@ cfg_if! {
         use std::thread;
         pub use rayon::{join, scope};
 
+        pub use rayon_core::WorkerLocal;
+
         pub use rayon::iter::ParallelIterator;
         use rayon::iter::IntoParallelIterator;