Ruby：如何从文件或 STDIN 读取可能是 gzip 压缩的数据？

Question

我想从输入文件或 STDIN 读取数据 - 输入数据可能经过 gzip 压缩。

对于文件，这可以使用 Zlib::GzipReader 来完成，如下所示：

require 'zlib'

ios = File.open(file, mode='r')

begin
  ios = Zlib::GzipReader.new(ios)
rescue
  ios.rewind
end

ios.each_line { |line| puts line }

但是，我无法从 STDIN 正确检测到压缩数据：

require 'zlib'

if STDIN.tty?
  # do nothing
else
  ios = STDIN

  begin
    ios = Zlib::GzipReader.new(ios)
  rescue
    ios.rewind
  end
end

ios.each_line { |line| puts line }

上面的内容适用于 STDIN 中的 gzip 压缩数据，但常规数据会导致这样的结果：

./test.rb:14:in `rewind': Illegal seek - <STDIN> (Errno::ESPIPE)
        from ./test.rb:14:in `rescue in <main>'
        from ./test.rb:11:in `<main>'

所以，如果我不能倒带 STDIN，如何测试 STDIN 中的数据是否被压缩？

干杯，

马丁

Answer 1

将数据从 STDIN 加载到临时文件中，然后才解析它

require 'tempfile'

tf = Tempfile.new('tmp')

while $stdin.gets do
   tf.puts $_
end

tf.rewind