如何在Ruby中临时重定向stderr?

时间:2010-12-16 09:47:17

标签: ruby stderr

我想在块中使用Ruby脚本暂时重定向stderr,确保在块结束时将其重置为原始值。

我无法在ruby文档中找到如何执行此操作。

4 个答案:

答案 0 :(得分:63)

在Ruby中,$stderr当前使用的输出流称为stderr,而STDERR默认 stderr流。可以很容易地将不同的输出流暂时分配给$stderr

require "stringio"

def capture_stderr
  # The output stream must be an IO-like object. In this case we capture it in
  # an in-memory IO object so we can return the string value. You can assign any
  # IO object here.
  previous_stderr, $stderr = $stderr, StringIO.new
  yield
  $stderr.string
ensure
  # Restore the previous value of stderr (typically equal to STDERR).
  $stderr = previous_stderr
end

现在您可以执行以下操作:

captured_output = capture_stderr do
  # Does not output anything directly.
  $stderr.puts "test"
end

captured_output
#=> "test\n"

同样的原则也适用于$stdoutSTDOUT

答案 1 :(得分:15)

这是一个更抽象的解决方案(归功于David Heinemeier Hansson):

def silence_streams(*streams)
  on_hold = streams.collect { |stream| stream.dup }
  streams.each do |stream|
    stream.reopen(RUBY_PLATFORM =~ /mswin/ ? 'NUL:' : '/dev/null')
    stream.sync = true
  end
  yield
ensure
  streams.each_with_index do |stream, i|
    stream.reopen(on_hold[i])
  end
end

用法:

silence_streams(STDERR) { do_something }

答案 2 :(得分:10)

与@ molf的答案基本相同,具有相同的用法:

require "stringio"
def capture_stderr
  real_stderr, $stderr = $stderr, StringIO.new
  yield
  $stderr.string
ensure
  $stderr = real_stderr
end

它更简洁地使用StringIO,并保留$ stderr,就像调用capture_stderr之前的那样。

答案 3 :(得分:6)

我喜欢StringIO的答案。但是,如果您正在调用外部进程,并且$stderr = StringIO.new不起作用,则可以将stderr写入临时文件:

require 'tempfile'

def capture_stderr
  backup_stderr = STDERR.dup
  begin
    Tempfile.open("captured_stderr") do |f|
      STDERR.reopen(f)
      yield
      f.rewind
      f.read
    end
  ensure
    STDERR.reopen backup_stderr
  end
end