1 //!
2 //! [Under construction](https://github.com/rayon-rs/rayon/issues/231)
3 //!
4 //! ## Restricting multiple versions
5 //!
6 //! In order to ensure proper coordination between threadpools, and especially
7 //! to make sure there's only one global threadpool, `rayon-core` is actively
8 //! restricted from building multiple versions of itself into a single target.
9 //! You may see a build error like this in violation:
10 //!
11 //! ```text
12 //! error: native library `rayon-core` is being linked to by more
13 //! than one package, and can only be linked to by one package
14 //! ```
15 //!
16 //! While we strive to keep `rayon-core` semver-compatible, it's still
17 //! possible to arrive at this situation if different crates have overly
18 //! restrictive tilde or inequality requirements for `rayon-core`.  The
19 //! conflicting requirements will need to be resolved before the build will
20 //! succeed.
21 
22 #![doc(html_root_url = "https://docs.rs/rayon-core/1.9")]
23 #![deny(missing_debug_implementations)]
24 #![deny(missing_docs)]
25 #![deny(unreachable_pub)]
26 #![warn(rust_2018_idioms)]
27 
28 use std::any::Any;
29 use std::env;
30 use std::error::Error;
31 use std::fmt;
32 use std::io;
33 use std::marker::PhantomData;
34 use std::str::FromStr;
35 
36 #[macro_use]
37 mod log;
38 #[macro_use]
39 mod private;
40 
41 mod job;
42 mod join;
43 mod latch;
44 mod registry;
45 mod scope;
46 mod sleep;
47 mod spawn;
48 mod thread_pool;
49 mod unwind;
50 mod util;
51 
52 mod compile_fail;
53 mod test;
54 
55 pub use self::join::{join, join_context};
56 pub use self::registry::ThreadBuilder;
57 pub use self::scope::{scope, Scope};
58 pub use self::scope::{scope_fifo, ScopeFifo};
59 pub use self::spawn::{spawn, spawn_fifo};
60 pub use self::thread_pool::current_thread_has_pending_tasks;
61 pub use self::thread_pool::current_thread_index;
62 pub use self::thread_pool::ThreadPool;
63 
64 use self::registry::{CustomSpawn, DefaultSpawn, ThreadSpawn};
65 
66 /// Returns the number of threads in the current registry. If this
67 /// code is executing within a Rayon thread-pool, then this will be
68 /// the number of threads for the thread-pool of the current
69 /// thread. Otherwise, it will be the number of threads for the global
70 /// thread-pool.
71 ///
72 /// This can be useful when trying to judge how many times to split
73 /// parallel work (the parallel iterator traits use this value
74 /// internally for this purpose).
75 ///
76 /// # Future compatibility note
77 ///
78 /// Note that unless this thread-pool was created with a
79 /// builder that specifies the number of threads, then this
80 /// number may vary over time in future versions (see [the
81 /// `num_threads()` method for details][snt]).
82 ///
83 /// [snt]: struct.ThreadPoolBuilder.html#method.num_threads
current_num_threads() -> usize84 pub fn current_num_threads() -> usize {
85     crate::registry::Registry::current_num_threads()
86 }
87 
88 /// Error when initializing a thread pool.
89 #[derive(Debug)]
90 pub struct ThreadPoolBuildError {
91     kind: ErrorKind,
92 }
93 
94 #[derive(Debug)]
95 enum ErrorKind {
96     GlobalPoolAlreadyInitialized,
97     IOError(io::Error),
98 }
99 
100 /// Used to create a new [`ThreadPool`] or to configure the global rayon thread pool.
101 /// ## Creating a ThreadPool
102 /// The following creates a thread pool with 22 threads.
103 ///
104 /// ```rust
105 /// # use rayon_core as rayon;
106 /// let pool = rayon::ThreadPoolBuilder::new().num_threads(22).build().unwrap();
107 /// ```
108 ///
109 /// To instead configure the global thread pool, use [`build_global()`]:
110 ///
111 /// ```rust
112 /// # use rayon_core as rayon;
113 /// rayon::ThreadPoolBuilder::new().num_threads(22).build_global().unwrap();
114 /// ```
115 ///
116 /// [`ThreadPool`]: struct.ThreadPool.html
117 /// [`build_global()`]: struct.ThreadPoolBuilder.html#method.build_global
118 pub struct ThreadPoolBuilder<S = DefaultSpawn> {
119     /// The number of threads in the rayon thread pool.
120     /// If zero will use the RAYON_NUM_THREADS environment variable.
121     /// If RAYON_NUM_THREADS is invalid or zero will use the default.
122     num_threads: usize,
123 
124     /// Custom closure, if any, to handle a panic that we cannot propagate
125     /// anywhere else.
126     panic_handler: Option<Box<PanicHandler>>,
127 
128     /// Closure to compute the name of a thread.
129     get_thread_name: Option<Box<dyn FnMut(usize) -> String>>,
130 
131     /// The stack size for the created worker threads
132     stack_size: Option<usize>,
133 
134     /// Closure invoked on worker thread start.
135     start_handler: Option<Box<StartHandler>>,
136 
137     /// Closure invoked on worker thread exit.
138     exit_handler: Option<Box<ExitHandler>>,
139 
140     /// Closure invoked to spawn threads.
141     spawn_handler: S,
142 
143     /// If false, worker threads will execute spawned jobs in a
144     /// "depth-first" fashion. If true, they will do a "breadth-first"
145     /// fashion. Depth-first is the default.
146     breadth_first: bool,
147 }
148 
149 /// Contains the rayon thread pool configuration. Use [`ThreadPoolBuilder`] instead.
150 ///
151 /// [`ThreadPoolBuilder`]: struct.ThreadPoolBuilder.html
152 #[deprecated(note = "Use `ThreadPoolBuilder`")]
153 pub struct Configuration {
154     builder: ThreadPoolBuilder,
155 }
156 
157 /// The type for a panic handling closure. Note that this same closure
158 /// may be invoked multiple times in parallel.
159 type PanicHandler = dyn Fn(Box<dyn Any + Send>) + Send + Sync;
160 
161 /// The type for a closure that gets invoked when a thread starts. The
162 /// closure is passed the index of the thread on which it is invoked.
163 /// Note that this same closure may be invoked multiple times in parallel.
164 type StartHandler = dyn Fn(usize) + Send + Sync;
165 
166 /// The type for a closure that gets invoked when a thread exits. The
167 /// closure is passed the index of the thread on which is is invoked.
168 /// Note that this same closure may be invoked multiple times in parallel.
169 type ExitHandler = dyn Fn(usize) + Send + Sync;
170 
171 // NB: We can't `#[derive(Default)]` because `S` is left ambiguous.
172 impl Default for ThreadPoolBuilder {
default() -> Self173     fn default() -> Self {
174         ThreadPoolBuilder {
175             num_threads: 0,
176             panic_handler: None,
177             get_thread_name: None,
178             stack_size: None,
179             start_handler: None,
180             exit_handler: None,
181             spawn_handler: DefaultSpawn,
182             breadth_first: false,
183         }
184     }
185 }
186 
187 impl ThreadPoolBuilder {
188     /// Creates and returns a valid rayon thread pool builder, but does not initialize it.
new() -> Self189     pub fn new() -> Self {
190         Self::default()
191     }
192 }
193 
194 /// Note: the `S: ThreadSpawn` constraint is an internal implementation detail for the
195 /// default spawn and those set by [`spawn_handler`](#method.spawn_handler).
196 impl<S> ThreadPoolBuilder<S>
197 where
198     S: ThreadSpawn,
199 {
200     /// Creates a new `ThreadPool` initialized using this configuration.
build(self) -> Result<ThreadPool, ThreadPoolBuildError>201     pub fn build(self) -> Result<ThreadPool, ThreadPoolBuildError> {
202         ThreadPool::build(self)
203     }
204 
205     /// Initializes the global thread pool. This initialization is
206     /// **optional**.  If you do not call this function, the thread pool
207     /// will be automatically initialized with the default
208     /// configuration. Calling `build_global` is not recommended, except
209     /// in two scenarios:
210     ///
211     /// - You wish to change the default configuration.
212     /// - You are running a benchmark, in which case initializing may
213     ///   yield slightly more consistent results, since the worker threads
214     ///   will already be ready to go even in the first iteration.  But
215     ///   this cost is minimal.
216     ///
217     /// Initialization of the global thread pool happens exactly
218     /// once. Once started, the configuration cannot be
219     /// changed. Therefore, if you call `build_global` a second time, it
220     /// will return an error. An `Ok` result indicates that this
221     /// is the first initialization of the thread pool.
build_global(self) -> Result<(), ThreadPoolBuildError>222     pub fn build_global(self) -> Result<(), ThreadPoolBuildError> {
223         let registry = registry::init_global_registry(self)?;
224         registry.wait_until_primed();
225         Ok(())
226     }
227 }
228 
229 impl ThreadPoolBuilder {
230     /// Creates a scoped `ThreadPool` initialized using this configuration.
231     ///
232     /// This is a convenience function for building a pool using [`crossbeam::scope`]
233     /// to spawn threads in a [`spawn_handler`](#method.spawn_handler).
234     /// The threads in this pool will start by calling `wrapper`, which should
235     /// do initialization and continue by calling `ThreadBuilder::run()`.
236     ///
237     /// [`crossbeam::scope`]: https://docs.rs/crossbeam/0.7/crossbeam/fn.scope.html
238     ///
239     /// # Examples
240     ///
241     /// A scoped pool may be useful in combination with scoped thread-local variables.
242     ///
243     /// ```
244     /// # use rayon_core as rayon;
245     ///
246     /// scoped_tls::scoped_thread_local!(static POOL_DATA: Vec<i32>);
247     ///
248     /// fn main() -> Result<(), rayon::ThreadPoolBuildError> {
249     ///     let pool_data = vec![1, 2, 3];
250     ///
251     ///     // We haven't assigned any TLS data yet.
252     ///     assert!(!POOL_DATA.is_set());
253     ///
254     ///     rayon::ThreadPoolBuilder::new()
255     ///         .build_scoped(
256     ///             // Borrow `pool_data` in TLS for each thread.
257     ///             |thread| POOL_DATA.set(&pool_data, || thread.run()),
258     ///             // Do some work that needs the TLS data.
259     ///             |pool| pool.install(|| assert!(POOL_DATA.is_set())),
260     ///         )?;
261     ///
262     ///     // Once we've returned, `pool_data` is no longer borrowed.
263     ///     drop(pool_data);
264     ///     Ok(())
265     /// }
266     /// ```
build_scoped<W, F, R>(self, wrapper: W, with_pool: F) -> Result<R, ThreadPoolBuildError> where W: Fn(ThreadBuilder) + Sync, F: FnOnce(&ThreadPool) -> R,267     pub fn build_scoped<W, F, R>(self, wrapper: W, with_pool: F) -> Result<R, ThreadPoolBuildError>
268     where
269         W: Fn(ThreadBuilder) + Sync, // expected to call `run()`
270         F: FnOnce(&ThreadPool) -> R,
271     {
272         let result = crossbeam_utils::thread::scope(|scope| {
273             let wrapper = &wrapper;
274             let pool = self
275                 .spawn_handler(|thread| {
276                     let mut builder = scope.builder();
277                     if let Some(name) = thread.name() {
278                         builder = builder.name(name.to_string());
279                     }
280                     if let Some(size) = thread.stack_size() {
281                         builder = builder.stack_size(size);
282                     }
283                     builder.spawn(move |_| wrapper(thread))?;
284                     Ok(())
285                 })
286                 .build()?;
287             Ok(with_pool(&pool))
288         });
289 
290         match result {
291             Ok(result) => result,
292             Err(err) => unwind::resume_unwinding(err),
293         }
294     }
295 }
296 
297 impl<S> ThreadPoolBuilder<S> {
298     /// Sets a custom function for spawning threads.
299     ///
300     /// Note that the threads will not exit until after the pool is dropped. It
301     /// is up to the caller to wait for thread termination if that is important
302     /// for any invariants. For instance, threads created in [`crossbeam::scope`]
303     /// will be joined before that scope returns, and this will block indefinitely
304     /// if the pool is leaked. Furthermore, the global thread pool doesn't terminate
305     /// until the entire process exits!
306     ///
307     /// [`crossbeam::scope`]: https://docs.rs/crossbeam/0.7/crossbeam/fn.scope.html
308     ///
309     /// # Examples
310     ///
311     /// A minimal spawn handler just needs to call `run()` from an independent thread.
312     ///
313     /// ```
314     /// # use rayon_core as rayon;
315     /// fn main() -> Result<(), rayon::ThreadPoolBuildError> {
316     ///     let pool = rayon::ThreadPoolBuilder::new()
317     ///         .spawn_handler(|thread| {
318     ///             std::thread::spawn(|| thread.run());
319     ///             Ok(())
320     ///         })
321     ///         .build()?;
322     ///
323     ///     pool.install(|| println!("Hello from my custom thread!"));
324     ///     Ok(())
325     /// }
326     /// ```
327     ///
328     /// The default spawn handler sets the name and stack size if given, and propagates
329     /// any errors from the thread builder.
330     ///
331     /// ```
332     /// # use rayon_core as rayon;
333     /// fn main() -> Result<(), rayon::ThreadPoolBuildError> {
334     ///     let pool = rayon::ThreadPoolBuilder::new()
335     ///         .spawn_handler(|thread| {
336     ///             let mut b = std::thread::Builder::new();
337     ///             if let Some(name) = thread.name() {
338     ///                 b = b.name(name.to_owned());
339     ///             }
340     ///             if let Some(stack_size) = thread.stack_size() {
341     ///                 b = b.stack_size(stack_size);
342     ///             }
343     ///             b.spawn(|| thread.run())?;
344     ///             Ok(())
345     ///         })
346     ///         .build()?;
347     ///
348     ///     pool.install(|| println!("Hello from my fully custom thread!"));
349     ///     Ok(())
350     /// }
351     /// ```
spawn_handler<F>(self, spawn: F) -> ThreadPoolBuilder<CustomSpawn<F>> where F: FnMut(ThreadBuilder) -> io::Result<()>,352     pub fn spawn_handler<F>(self, spawn: F) -> ThreadPoolBuilder<CustomSpawn<F>>
353     where
354         F: FnMut(ThreadBuilder) -> io::Result<()>,
355     {
356         ThreadPoolBuilder {
357             spawn_handler: CustomSpawn::new(spawn),
358             // ..self
359             num_threads: self.num_threads,
360             panic_handler: self.panic_handler,
361             get_thread_name: self.get_thread_name,
362             stack_size: self.stack_size,
363             start_handler: self.start_handler,
364             exit_handler: self.exit_handler,
365             breadth_first: self.breadth_first,
366         }
367     }
368 
369     /// Returns a reference to the current spawn handler.
get_spawn_handler(&mut self) -> &mut S370     fn get_spawn_handler(&mut self) -> &mut S {
371         &mut self.spawn_handler
372     }
373 
374     /// Get the number of threads that will be used for the thread
375     /// pool. See `num_threads()` for more information.
get_num_threads(&self) -> usize376     fn get_num_threads(&self) -> usize {
377         if self.num_threads > 0 {
378             self.num_threads
379         } else {
380             match env::var("RAYON_NUM_THREADS")
381                 .ok()
382                 .and_then(|s| usize::from_str(&s).ok())
383             {
384                 Some(x) if x > 0 => return x,
385                 Some(x) if x == 0 => return num_cpus::get(),
386                 _ => {}
387             }
388 
389             // Support for deprecated `RAYON_RS_NUM_CPUS`.
390             match env::var("RAYON_RS_NUM_CPUS")
391                 .ok()
392                 .and_then(|s| usize::from_str(&s).ok())
393             {
394                 Some(x) if x > 0 => x,
395                 _ => num_cpus::get(),
396             }
397         }
398     }
399 
400     /// Get the thread name for the thread with the given index.
get_thread_name(&mut self, index: usize) -> Option<String>401     fn get_thread_name(&mut self, index: usize) -> Option<String> {
402         let f = self.get_thread_name.as_mut()?;
403         Some(f(index))
404     }
405 
406     /// Sets a closure which takes a thread index and returns
407     /// the thread's name.
thread_name<F>(mut self, closure: F) -> Self where F: FnMut(usize) -> String + 'static,408     pub fn thread_name<F>(mut self, closure: F) -> Self
409     where
410         F: FnMut(usize) -> String + 'static,
411     {
412         self.get_thread_name = Some(Box::new(closure));
413         self
414     }
415 
416     /// Sets the number of threads to be used in the rayon threadpool.
417     ///
418     /// If you specify a non-zero number of threads using this
419     /// function, then the resulting thread-pools are guaranteed to
420     /// start at most this number of threads.
421     ///
422     /// If `num_threads` is 0, or you do not call this function, then
423     /// the Rayon runtime will select the number of threads
424     /// automatically. At present, this is based on the
425     /// `RAYON_NUM_THREADS` environment variable (if set),
426     /// or the number of logical CPUs (otherwise).
427     /// In the future, however, the default behavior may
428     /// change to dynamically add or remove threads as needed.
429     ///
430     /// **Future compatibility warning:** Given the default behavior
431     /// may change in the future, if you wish to rely on a fixed
432     /// number of threads, you should use this function to specify
433     /// that number. To reproduce the current default behavior, you
434     /// may wish to use the [`num_cpus`
435     /// crate](https://crates.io/crates/num_cpus) to query the number
436     /// of CPUs dynamically.
437     ///
438     /// **Old environment variable:** `RAYON_NUM_THREADS` is a one-to-one
439     /// replacement of the now deprecated `RAYON_RS_NUM_CPUS` environment
440     /// variable. If both variables are specified, `RAYON_NUM_THREADS` will
441     /// be prefered.
num_threads(mut self, num_threads: usize) -> Self442     pub fn num_threads(mut self, num_threads: usize) -> Self {
443         self.num_threads = num_threads;
444         self
445     }
446 
447     /// Returns a copy of the current panic handler.
take_panic_handler(&mut self) -> Option<Box<PanicHandler>>448     fn take_panic_handler(&mut self) -> Option<Box<PanicHandler>> {
449         self.panic_handler.take()
450     }
451 
452     /// Normally, whenever Rayon catches a panic, it tries to
453     /// propagate it to someplace sensible, to try and reflect the
454     /// semantics of sequential execution. But in some cases,
455     /// particularly with the `spawn()` APIs, there is no
456     /// obvious place where we should propagate the panic to.
457     /// In that case, this panic handler is invoked.
458     ///
459     /// If no panic handler is set, the default is to abort the
460     /// process, under the principle that panics should not go
461     /// unobserved.
462     ///
463     /// If the panic handler itself panics, this will abort the
464     /// process. To prevent this, wrap the body of your panic handler
465     /// in a call to `std::panic::catch_unwind()`.
panic_handler<H>(mut self, panic_handler: H) -> Self where H: Fn(Box<dyn Any + Send>) + Send + Sync + 'static,466     pub fn panic_handler<H>(mut self, panic_handler: H) -> Self
467     where
468         H: Fn(Box<dyn Any + Send>) + Send + Sync + 'static,
469     {
470         self.panic_handler = Some(Box::new(panic_handler));
471         self
472     }
473 
474     /// Get the stack size of the worker threads
get_stack_size(&self) -> Option<usize>475     fn get_stack_size(&self) -> Option<usize> {
476         self.stack_size
477     }
478 
479     /// Sets the stack size of the worker threads
stack_size(mut self, stack_size: usize) -> Self480     pub fn stack_size(mut self, stack_size: usize) -> Self {
481         self.stack_size = Some(stack_size);
482         self
483     }
484 
485     /// **(DEPRECATED)** Suggest to worker threads that they execute
486     /// spawned jobs in a "breadth-first" fashion.
487     ///
488     /// Typically, when a worker thread is idle or blocked, it will
489     /// attempt to execute the job from the *top* of its local deque of
490     /// work (i.e., the job most recently spawned). If this flag is set
491     /// to true, however, workers will prefer to execute in a
492     /// *breadth-first* fashion -- that is, they will search for jobs at
493     /// the *bottom* of their local deque. (At present, workers *always*
494     /// steal from the bottom of other worker's deques, regardless of
495     /// the setting of this flag.)
496     ///
497     /// If you think of the tasks as a tree, where a parent task
498     /// spawns its children in the tree, then this flag loosely
499     /// corresponds to doing a breadth-first traversal of the tree,
500     /// whereas the default would be to do a depth-first traversal.
501     ///
502     /// **Note that this is an "execution hint".** Rayon's task
503     /// execution is highly dynamic and the precise order in which
504     /// independent tasks are executed is not intended to be
505     /// guaranteed.
506     ///
507     /// This `breadth_first()` method is now deprecated per [RFC #1],
508     /// and in the future its effect may be removed. Consider using
509     /// [`scope_fifo()`] for a similar effect.
510     ///
511     /// [RFC #1]: https://github.com/rayon-rs/rfcs/blob/master/accepted/rfc0001-scope-scheduling.md
512     /// [`scope_fifo()`]: fn.scope_fifo.html
513     #[deprecated(note = "use `scope_fifo` and `spawn_fifo` for similar effect")]
breadth_first(mut self) -> Self514     pub fn breadth_first(mut self) -> Self {
515         self.breadth_first = true;
516         self
517     }
518 
get_breadth_first(&self) -> bool519     fn get_breadth_first(&self) -> bool {
520         self.breadth_first
521     }
522 
523     /// Takes the current thread start callback, leaving `None`.
take_start_handler(&mut self) -> Option<Box<StartHandler>>524     fn take_start_handler(&mut self) -> Option<Box<StartHandler>> {
525         self.start_handler.take()
526     }
527 
528     /// Sets a callback to be invoked on thread start.
529     ///
530     /// The closure is passed the index of the thread on which it is invoked.
531     /// Note that this same closure may be invoked multiple times in parallel.
532     /// If this closure panics, the panic will be passed to the panic handler.
533     /// If that handler returns, then startup will continue normally.
start_handler<H>(mut self, start_handler: H) -> Self where H: Fn(usize) + Send + Sync + 'static,534     pub fn start_handler<H>(mut self, start_handler: H) -> Self
535     where
536         H: Fn(usize) + Send + Sync + 'static,
537     {
538         self.start_handler = Some(Box::new(start_handler));
539         self
540     }
541 
542     /// Returns a current thread exit callback, leaving `None`.
take_exit_handler(&mut self) -> Option<Box<ExitHandler>>543     fn take_exit_handler(&mut self) -> Option<Box<ExitHandler>> {
544         self.exit_handler.take()
545     }
546 
547     /// Sets a callback to be invoked on thread exit.
548     ///
549     /// The closure is passed the index of the thread on which it is invoked.
550     /// Note that this same closure may be invoked multiple times in parallel.
551     /// If this closure panics, the panic will be passed to the panic handler.
552     /// If that handler returns, then the thread will exit normally.
exit_handler<H>(mut self, exit_handler: H) -> Self where H: Fn(usize) + Send + Sync + 'static,553     pub fn exit_handler<H>(mut self, exit_handler: H) -> Self
554     where
555         H: Fn(usize) + Send + Sync + 'static,
556     {
557         self.exit_handler = Some(Box::new(exit_handler));
558         self
559     }
560 }
561 
562 #[allow(deprecated)]
563 impl Configuration {
564     /// Creates and return a valid rayon thread pool configuration, but does not initialize it.
new() -> Configuration565     pub fn new() -> Configuration {
566         Configuration {
567             builder: ThreadPoolBuilder::new(),
568         }
569     }
570 
571     /// Deprecated in favor of `ThreadPoolBuilder::build`.
build(self) -> Result<ThreadPool, Box<dyn Error + 'static>>572     pub fn build(self) -> Result<ThreadPool, Box<dyn Error + 'static>> {
573         self.builder.build().map_err(Box::from)
574     }
575 
576     /// Deprecated in favor of `ThreadPoolBuilder::thread_name`.
thread_name<F>(mut self, closure: F) -> Self where F: FnMut(usize) -> String + 'static,577     pub fn thread_name<F>(mut self, closure: F) -> Self
578     where
579         F: FnMut(usize) -> String + 'static,
580     {
581         self.builder = self.builder.thread_name(closure);
582         self
583     }
584 
585     /// Deprecated in favor of `ThreadPoolBuilder::num_threads`.
num_threads(mut self, num_threads: usize) -> Configuration586     pub fn num_threads(mut self, num_threads: usize) -> Configuration {
587         self.builder = self.builder.num_threads(num_threads);
588         self
589     }
590 
591     /// Deprecated in favor of `ThreadPoolBuilder::panic_handler`.
panic_handler<H>(mut self, panic_handler: H) -> Configuration where H: Fn(Box<dyn Any + Send>) + Send + Sync + 'static,592     pub fn panic_handler<H>(mut self, panic_handler: H) -> Configuration
593     where
594         H: Fn(Box<dyn Any + Send>) + Send + Sync + 'static,
595     {
596         self.builder = self.builder.panic_handler(panic_handler);
597         self
598     }
599 
600     /// Deprecated in favor of `ThreadPoolBuilder::stack_size`.
stack_size(mut self, stack_size: usize) -> Self601     pub fn stack_size(mut self, stack_size: usize) -> Self {
602         self.builder = self.builder.stack_size(stack_size);
603         self
604     }
605 
606     /// Deprecated in favor of `ThreadPoolBuilder::breadth_first`.
breadth_first(mut self) -> Self607     pub fn breadth_first(mut self) -> Self {
608         self.builder = self.builder.breadth_first();
609         self
610     }
611 
612     /// Deprecated in favor of `ThreadPoolBuilder::start_handler`.
start_handler<H>(mut self, start_handler: H) -> Configuration where H: Fn(usize) + Send + Sync + 'static,613     pub fn start_handler<H>(mut self, start_handler: H) -> Configuration
614     where
615         H: Fn(usize) + Send + Sync + 'static,
616     {
617         self.builder = self.builder.start_handler(start_handler);
618         self
619     }
620 
621     /// Deprecated in favor of `ThreadPoolBuilder::exit_handler`.
exit_handler<H>(mut self, exit_handler: H) -> Configuration where H: Fn(usize) + Send + Sync + 'static,622     pub fn exit_handler<H>(mut self, exit_handler: H) -> Configuration
623     where
624         H: Fn(usize) + Send + Sync + 'static,
625     {
626         self.builder = self.builder.exit_handler(exit_handler);
627         self
628     }
629 
630     /// Returns a ThreadPoolBuilder with identical parameters.
into_builder(self) -> ThreadPoolBuilder631     fn into_builder(self) -> ThreadPoolBuilder {
632         self.builder
633     }
634 }
635 
636 impl ThreadPoolBuildError {
new(kind: ErrorKind) -> ThreadPoolBuildError637     fn new(kind: ErrorKind) -> ThreadPoolBuildError {
638         ThreadPoolBuildError { kind }
639     }
640 }
641 
642 const GLOBAL_POOL_ALREADY_INITIALIZED: &str =
643     "The global thread pool has already been initialized.";
644 
645 impl Error for ThreadPoolBuildError {
646     #[allow(deprecated)]
description(&self) -> &str647     fn description(&self) -> &str {
648         match self.kind {
649             ErrorKind::GlobalPoolAlreadyInitialized => GLOBAL_POOL_ALREADY_INITIALIZED,
650             ErrorKind::IOError(ref e) => e.description(),
651         }
652     }
653 
source(&self) -> Option<&(dyn Error + 'static)>654     fn source(&self) -> Option<&(dyn Error + 'static)> {
655         match &self.kind {
656             ErrorKind::GlobalPoolAlreadyInitialized => None,
657             ErrorKind::IOError(e) => Some(e),
658         }
659     }
660 }
661 
662 impl fmt::Display for ThreadPoolBuildError {
fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result663     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
664         match &self.kind {
665             ErrorKind::GlobalPoolAlreadyInitialized => GLOBAL_POOL_ALREADY_INITIALIZED.fmt(f),
666             ErrorKind::IOError(e) => e.fmt(f),
667         }
668     }
669 }
670 
671 /// Deprecated in favor of `ThreadPoolBuilder::build_global`.
672 #[deprecated(note = "use `ThreadPoolBuilder::build_global`")]
673 #[allow(deprecated)]
initialize(config: Configuration) -> Result<(), Box<dyn Error>>674 pub fn initialize(config: Configuration) -> Result<(), Box<dyn Error>> {
675     config.into_builder().build_global().map_err(Box::from)
676 }
677 
678 impl<S> fmt::Debug for ThreadPoolBuilder<S> {
fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result679     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
680         let ThreadPoolBuilder {
681             ref num_threads,
682             ref get_thread_name,
683             ref panic_handler,
684             ref stack_size,
685             ref start_handler,
686             ref exit_handler,
687             spawn_handler: _,
688             ref breadth_first,
689         } = *self;
690 
691         // Just print `Some(<closure>)` or `None` to the debug
692         // output.
693         struct ClosurePlaceholder;
694         impl fmt::Debug for ClosurePlaceholder {
695             fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
696                 f.write_str("<closure>")
697             }
698         }
699         let get_thread_name = get_thread_name.as_ref().map(|_| ClosurePlaceholder);
700         let panic_handler = panic_handler.as_ref().map(|_| ClosurePlaceholder);
701         let start_handler = start_handler.as_ref().map(|_| ClosurePlaceholder);
702         let exit_handler = exit_handler.as_ref().map(|_| ClosurePlaceholder);
703 
704         f.debug_struct("ThreadPoolBuilder")
705             .field("num_threads", num_threads)
706             .field("get_thread_name", &get_thread_name)
707             .field("panic_handler", &panic_handler)
708             .field("stack_size", &stack_size)
709             .field("start_handler", &start_handler)
710             .field("exit_handler", &exit_handler)
711             .field("breadth_first", &breadth_first)
712             .finish()
713     }
714 }
715 
716 #[allow(deprecated)]
717 impl Default for Configuration {
default() -> Self718     fn default() -> Self {
719         Configuration {
720             builder: Default::default(),
721         }
722     }
723 }
724 
725 #[allow(deprecated)]
726 impl fmt::Debug for Configuration {
fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result727     fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
728         self.builder.fmt(f)
729     }
730 }
731 
732 /// Provides the calling context to a closure called by `join_context`.
733 #[derive(Debug)]
734 pub struct FnContext {
735     migrated: bool,
736 
737     /// disable `Send` and `Sync`, just for a little future-proofing.
738     _marker: PhantomData<*mut ()>,
739 }
740 
741 impl FnContext {
742     #[inline]
new(migrated: bool) -> Self743     fn new(migrated: bool) -> Self {
744         FnContext {
745             migrated,
746             _marker: PhantomData,
747         }
748     }
749 }
750 
751 impl FnContext {
752     /// Returns `true` if the closure was called from a different thread
753     /// than it was provided from.
754     #[inline]
migrated(&self) -> bool755     pub fn migrated(&self) -> bool {
756         self.migrated
757     }
758 }
759