summaryrefslogtreecommitdiffstats
path: root/vendor/anstream/src/lockable.rs
blob: fcf20e317528763cfdb3fe2b629e49214d7563bb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
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())
    }
}