diff options
Diffstat (limited to 'vendor/anstream/src/lockable.rs')
-rw-r--r-- | vendor/anstream/src/lockable.rs | 48 |
1 files changed, 48 insertions, 0 deletions
diff --git a/vendor/anstream/src/lockable.rs b/vendor/anstream/src/lockable.rs new file mode 100644 index 000000000..fcf20e317 --- /dev/null +++ b/vendor/anstream/src/lockable.rs @@ -0,0 +1,48 @@ +#[cfg(all(windows, feature = "wincon"))] +use crate::RawStream; + +/// Explicitly lock a [`std::io::Write`]able +pub trait Lockable { + type Locked; + + /// Get exclusive access to the `AutoStream` + /// + /// Why? + /// - Faster performance when writing in a loop + /// - Avoid other threads interleaving output with the current thread + fn lock(self) -> Self::Locked; +} + +impl Lockable for std::io::Stdout { + type Locked = std::io::StdoutLock<'static>; + + #[inline] + fn lock(self) -> Self::Locked { + #[allow(clippy::needless_borrow)] // Its needed to avoid recursion + (&self).lock() + } +} + +impl Lockable for std::io::Stderr { + type Locked = std::io::StderrLock<'static>; + + #[inline] + fn lock(self) -> Self::Locked { + #[allow(clippy::needless_borrow)] // Its needed to avoid recursion + (&self).lock() + } +} + +#[cfg(all(windows, feature = "wincon"))] +impl<S> Lockable for anstyle_wincon::Console<S> +where + S: RawStream + Lockable, + <S as Lockable>::Locked: RawStream, +{ + type Locked = anstyle_wincon::Console<<S as Lockable>::Locked>; + + #[inline] + fn lock(self) -> Self::Locked { + self.map(|s| s.lock()) + } +} |