如何将 Ruby 脚本的输出传送到“head”不会出现管道损坏错误

Question

我有一个简单的 Ruby 脚本，如下所示

require 'csv'
while line = STDIN.gets
  array = CSV.parse_line(line)
  puts array[2]
end

但是当我尝试在像这样的 Unix 管道中使用此脚本时，我得到 10 行输出，然后是一个错误：

ruby lib/myscript.rb < data.csv  | head

12080450
12080451
12080517
12081046
12081048
12081050
12081051
12081052
12081054
lib/myscript.rb:4:in `write': Broken pipe - <STDOUT> (Errno::EPIPE)

是否有一种方法可以以阻止的方式编写 Ruby 脚本引发破损管道异常？

Answer 1

head 在读取所需的所有数据后关闭标准输出流。您应该处理异常并停止写入标准输出。一旦标准输出关闭，以下代码将中止循环：

while line = STDIN.gets
  array = CSV.parse_line(line)
  begin
    puts array[2]
  rescue Errno::EPIPE
    break
  end
end

Answer 2

我使用的技巧是将 head 替换为 sed -n 1,10p。

这使管道保持打开状态，因此 ruby​​（或任何其他测试损坏管道并发出警告的程序）不会获取损坏的管道，因此不会发出警告。选择您想要的行数值。

显然，这并不是试图修改您的 Ruby 脚本。几乎可以肯定有一种方法可以在 Ruby 代码中做到这一点。然而，即使您无法选择修改生成消息的程序，“sed 而不是 head”技术也能发挥作用。