Let the OutputProxy hold Stdout/Stderr objects, so we do not have to aquire them each write!()

This commit is contained in:
Matthias Beyer 2018-03-04 15:41:32 +01:00
parent 763f3fab86
commit 0b593d6635
2 changed files with 12 additions and 12 deletions

View file

@ -31,24 +31,24 @@ use std::io::Write;
/// a "sink" which does not write to either.
///
pub enum OutputProxy {
Out,
Err,
Out(::std::io::Stdout),
Err(::std::io::Stderr),
Sink,
}
impl Write for OutputProxy {
fn write(&mut self, buf: &[u8]) -> ::std::io::Result<usize> {
match *self {
OutputProxy::Out => ::std::io::stdout().write(buf),
OutputProxy::Err => ::std::io::stderr().write(buf),
OutputProxy::Out(ref mut r) => r.write(buf),
OutputProxy::Err(ref mut r) => r.write(buf),
OutputProxy::Sink => Ok(0),
}
}
fn flush(&mut self) -> ::std::io::Result<()> {
match *self {
OutputProxy::Out => ::std::io::stdout().flush(),
OutputProxy::Err => ::std::io::stderr().flush(),
OutputProxy::Out(ref mut r) => r.flush(),
OutputProxy::Err(ref mut r) => r.flush(),
OutputProxy::Sink => Ok(()),
}
}
@ -58,8 +58,8 @@ impl Write for OutputProxy {
impl Debug for OutputProxy {
fn fmt(&self, f: &mut ::std::fmt::Formatter) -> Result<(), ::std::fmt::Error> {
match *self {
OutputProxy::Out => write!(f, "OutputProxy(Stdout)"),
OutputProxy::Err => write!(f, "OutputProxy(Stderr)"),
OutputProxy::Out(..) => write!(f, "OutputProxy(Stdout)"),
OutputProxy::Err(..) => write!(f, "OutputProxy(Stderr)"),
OutputProxy::Sink => write!(f, "OutputProxy(Sink)"),
}
}

View file

@ -447,14 +447,14 @@ impl<'a> Runtime<'a> {
pub fn stdout(&self) -> OutputProxy {
if self.stdout_is_tty {
OutputProxy::Out
OutputProxy::Out(::std::io::stdout())
} else {
OutputProxy::Err
OutputProxy::Err(::std::io::stderr())
}
}
pub fn stderr(&self) -> OutputProxy {
OutputProxy::Err
OutputProxy::Err(::std::io::stderr())
}
pub fn stdin(&self) -> Option<Stdin> {