Ibraheem Ahmed
53633392c3
## Summary Continuation of https://github.com/astral-sh/uv/pull/3493. This gives us more flexibility in case we decide to move away from `rayon` in the future.
41 lines
1.0 KiB
Rust
41 lines
1.0 KiB
Rust
use std::num::NonZeroUsize;
|
|
|
|
/// Concurrency limit settings.
|
|
#[derive(Copy, Clone, Debug)]
|
|
pub struct Concurrency {
|
|
/// The maximum number of concurrent downloads.
|
|
///
|
|
/// Note this value must be non-zero.
|
|
pub downloads: usize,
|
|
/// The maximum number of concurrent builds.
|
|
///
|
|
/// Note this value must be non-zero.
|
|
pub builds: usize,
|
|
/// The maximum number of concurrent installs.
|
|
///
|
|
/// Note this value must be non-zero.
|
|
pub installs: usize,
|
|
}
|
|
|
|
impl Default for Concurrency {
|
|
fn default() -> Self {
|
|
Concurrency {
|
|
downloads: Concurrency::DEFAULT_DOWNLOADS,
|
|
builds: Concurrency::threads(),
|
|
installs: Concurrency::threads(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Concurrency {
|
|
// The default concurrent downloads limit.
|
|
pub const DEFAULT_DOWNLOADS: usize = 50;
|
|
|
|
// The default concurrent builds and install limit.
|
|
pub fn threads() -> usize {
|
|
std::thread::available_parallelism()
|
|
.map(NonZeroUsize::get)
|
|
.unwrap_or(1)
|
|
}
|
|
}
|