如何让 Zlib 从 Ruby 中的 S3 流中解压缩？

Question

Ruby Zlib::GzipReader 应该通过一个类似 IO 的对象创建（必须有一个行为与 IO#read 相同的读取方法）。

我的问题是我无法从 AWS::S3 lib 获取这个类似 IO 的对象。 据我所知，获得流的唯一方法是将块传递给S3Object#stream。

我已经试过了：

Zlib::GzipReader.new(AWS::S3::S3Object.stream('file', 'bucket'))
# Wich gaves me error: undefined method `read' for #<AWS::S3::S3Object::Value:0x000000017cbe78>

有人知道我该如何实现吗？

Answer 1

一个简单的解决方案是将下载的数据写入StringIO，然后将其读回：

require 'stringio'

io = StringIO.new
io.write AWS::S3::S3Object.value('file', 'bucket')
io.rewind

gz = Zlib::GzipReader.new(io)
data = gz.read
gz.close

# do something with data ...

更复杂的方法是在流仍在下载时开始对 gzip 压缩的数据进行膨胀，这可以通过 IO.pipe 来实现。类似这样的东西：

reader, writer = IO.pipe

fork do
  reader.close
  AWS::S3::S3Object.stream('file', 'bucket') do |chunk|
    writer.write chunk
  end
end

writer.close

gz = Zlib::GzipReader.new(reader)
while line = gz.gets
  # do something with line ...
end

gz.close

您也可以使用Thread 代替fork：

reader, writer = IO.pipe

thread = Thread.new do
  AWS::S3::S3Object.stream('file', 'bucket') do |chunk|
    writer.write chunk
  end
  writer.close
end

gz = Zlib::GzipReader.new(reader)
while line = gz.gets
  # do something with line
end

gz.close
thread.join