Contents

require "tempfile"

# BufferIO is used for writing big segments of text. When the segment is bigger than a certain number of bytes,
# the underlying memory buffer will be swapped with a tempfile
class Tracksperanto::BufferIO < DelegateClass(IO)
  include Tracksperanto::Returning
  
  IN_MEMORY_SIZE_LIMIT = 100_000
  
  def initialize
    __setobj__(StringIO.new)
  end
  
  def write(s)
    returning(super) { replace_with_tempfile_if_needed }
  end
  
  def puts(s)
    returning(super) { replace_with_tempfile_if_needed }
  end
  
  def <<(s)
    returning(super) { replace_with_tempfile_if_needed }
  end
  
  def putc(c)
    returning(super) { replace_with_tempfile_if_needed }
  end
  
  def close!
    __getobj__.close! if @tempfile_in
    __setobj__(nil)
  end
  
  private
  
  def replace_with_tempfile_if_needed
    return if @tempfile_in
    io = __getobj__
    if io.pos > IN_MEMORY_SIZE_LIMIT
      @tempfile_in = true
      tf = Tempfile.new("tracksperanto-xbuf")
      io.rewind
      tf.write(io.read) until io.eof?
      __setobj__(tf)
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
tracksperanto-1.8.4	lib/tracksperanto/buffer_io.rb
tracksperanto-1.8.2	lib/tracksperanto/buffer_io.rb