将全局 $stdout 重新分配给控制台 - ruby

Question

我正在尝试将 $stdout 设置为临时写入文件，然后再返回文件。

test.rb :
   old_stdout = $stdout    
    $stdout.reopen("mytestfile.out",'w+')
       puts "this goes in mytestfile"
    $stdout= old_stdout
puts "this should be on the console"
    $stdout.reopen("mytestfile1.out",'w+')
       puts "this goes in mytestfile1:"
    $stdout = old_stdout
 puts "this should be back on the console"

这是输出。

ruby test.rb => no output on the console
cat mytestfile.out 
  this goes in mytestfile
  this should be on the console
cat  mytestfile1.out
  this goes in mytestfile1:
  this should be back on the console

我不确定为什么 $stdout 没有重置到控制台？

Answer 1

这个问题可以通过在更改之前在$stdout上调用dup来解决：

old_stdout = $stdout.dup  
$stdout.reopen("mytestfile.out",'w+')
puts "this goes in mytestfile"
$stdout = old_stdout.dup
puts "this should be on the console"
$stdout.reopen("mytestfile1.out",'w+')
puts "this goes in mytestfile1:"
$stdout = old_stdout
puts "this should be back on the console"

输出：

ruby test.rb
# => this should be on the console
# => this should be back on the console
cat mytestfile.out
# => this goes in mytestfile
cat mytestfile1.out
# => this goes in mytestfile1

以下是我通常将此功能打包成函数的方式：

# Runs a block of code while blocking stdout.
# Note that /dev/null should be changed to NUL on Windows.
def silence_stdout(log = '/dev/null')
  old = $stdout.dup
  $stdout.reopen(File.new(log, 'w'))
  yield
  $stdout = old
end

用法：

silence_stdout 'mytestfile.out' do
  puts "this goes in mytestfile"
end

puts "this should be on the console"

silence_stdout 'mytestfile1.out' do
  puts "this goes in mytestfile1"
end

puts "this should be back on the console"

编辑：正如另一张海报提到的，仅在使用纯 Ruby 代码时才需要使用重新打开。上面的函数既适用于纯 Ruby 代码，也适用于例如写入 STDOUT 的 C 扩展。

Answer 2

如果您只是使用 Ruby 代码，则无需使用 reopen。 puts 和其他 Ruby 方法将使用 $stdout 的当前值，因此您可以重新分配它。

old_stdout = $stdout    
$stdout = File.new("mytestfile.out",'w+')
puts "this goes in mytestfile"
$stdout = old_stdout
puts "this should be on the console"
$stdout = File.new("mytestfile1.out",'w+')
puts "this goes in mytestfile1:"
$stdout = old_stdout
puts "this should be back on the console"

如果您正在执行诸如创建子进程（例如使用fork）之类的操作并且希望子进程的输出到其他地方，或者如果您有一个直接写入标准输出的扩展，则您只需要使用reopen使用 Ruby 的 $stdout 全局。

在您的代码中，当您调用 reopen 时，您正在重定向 both $stdout 和 old_stdout，因为它们都只是对同一个 IO 对象的引用，这就是为什么您不是当您将old_stdout 分配回stdout 时，不会将输出返回到控制台。