Module: Fluent::Plugin::Buffer::Chunk::GzipDecompressable

Includes:

Compressable

Defined in:

lib/fluent/plugin/buffer/chunk.rb

Constant Summary

Constants included from Compressable

Compressable::DEFAULT_DECOMPRESSION_SIZE_LIMIT

Instance Method Summary

• #append(data, **kwargs) ⇒ Object
• #open(**kwargs, &block) ⇒ Object
• #read(**kwargs) ⇒ Object
• #write_to(io, **kwargs) ⇒ Object

Methods included from Compressable

#compress, #decompress

Instance Method Details

#append(data, **kwargs) ⇒ Object

# File 'lib/fluent/plugin/buffer/chunk.rb', line 198

def append(data, **kwargs)
  if kwargs[:compress] == :gzip
    io = StringIO.new
    Zlib::GzipWriter.wrap(io) do |gz|
      data.each do |d|
        gz.write d
      end
    end
    concat(io.string, data.size)
  else
    super
  end
end

#open(**kwargs, &block) ⇒ Object

# File 'lib/fluent/plugin/buffer/chunk.rb', line 212

def open(**kwargs, &block)
  if kwargs[:compressed] == :gzip
    super
  else
    super(**kwargs) do |chunk_io|
      output_io = if chunk_io.is_a?(StringIO)
                    StringIO.new
                  else
                    Tempfile.new('decompressed-data')
                  end
      output_io.binmode if output_io.is_a?(Tempfile)
      decompress(input_io: chunk_io, output_io: output_io)
      output_io.seek(0, IO::SEEK_SET)
      yield output_io
    end
  end
end

#read(**kwargs) ⇒ Object

# File 'lib/fluent/plugin/buffer/chunk.rb', line 230

def read(**kwargs)
  if kwargs[:compressed] == :gzip
    super
  else
    decompress(super)
  end
end

#write_to(io, **kwargs) ⇒ Object

# File 'lib/fluent/plugin/buffer/chunk.rb', line 238

def write_to(io, **kwargs)
  open(compressed: :gzip) do |chunk_io|
    if kwargs[:compressed] == :gzip
      IO.copy_stream(chunk_io, io)
    else
      decompress(input_io: chunk_io, output_io: io)
    end
  end
end