ruby - 如何将 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