Add a mean to get the current threadpool

rayon-core/src/lib.rs

@@ -109,6 +109,15 @@ pub fn current_num_threads() -> usize {
     crate::registry::Registry::current_num_threads()
 }
 
+/// Returns the current threadpool.
+///
+/// It is a shorthand for [`Threadpool::current()`][tc].
+///
+/// [tc]: struct.ThreadPool.html#method.current
+pub fn current() -> ThreadPool {
+    ThreadPool::current()
+}
+
 /// Error when initializing a thread pool.
 #[derive(Debug)]
 pub struct ThreadPoolBuildError {

rayon-core/src/thread_pool/mod.rs

@@ -252,6 +252,14 @@ impl ThreadPool {
         // We assert that `self.registry` has not terminated.
         unsafe { spawn::spawn_fifo_in(op, &self.registry) }
     }
+
+    /// Returns the current threadpool.
+    pub fn current() -> ThreadPool {
+        let registry = crate::registry::Registry::current();
+
+        registry.increment_terminate_count();
+        ThreadPool { registry }
+    }
 }
 
 impl Drop for ThreadPool {

src/lib.rs

@@ -114,7 +114,7 @@ pub use rayon_core::ThreadBuilder;
 pub use rayon_core::ThreadPool;
 pub use rayon_core::ThreadPoolBuildError;
 pub use rayon_core::ThreadPoolBuilder;
-pub use rayon_core::{current_num_threads, current_thread_index};
+pub use rayon_core::{current, current_num_threads, current_thread_index};
 pub use rayon_core::{join, join_context};
 pub use rayon_core::{scope, Scope};
 pub use rayon_core::{scope_fifo, ScopeFifo};