Sha256: 77a5a0ebc68d0cac10c120e3ad6af4032ff93b25c2f8a56b6857bda82c85f474

Contents?: true

Size: 1.39 KB

Versions: 5

Compression:

Stored size: 1.39 KB

Contents

# Provides method that can be included on File-type objects (IO, StringIO, Tempfile, etc) to allow stream copying
# and Tempfile conversion.
module IOStream

  # Returns a Tempfile containing the contents of the readable object.
  def to_tempfile
    tempfile = Tempfile.new("stream")
    tempfile.binmode
    self.stream_to(tempfile)
  end

  # Copies one read-able object from one place to another in blocks, obviating the need to load
  # the whole thing into memory. Defaults to 8k blocks. If this module is included in both
  # StringIO and Tempfile, then either can have its data copied anywhere else without typing
  # worries or memory overhead worries. Returns a File if a String is passed in as the destination
  # and returns the IO or Tempfile as passed in if one is sent as the destination.
  def stream_to path_or_file, in_blocks_of = 8192
    dstio = case path_or_file
            when String   then File.new(path_or_file, "wb+")
            when IO       then path_or_file
            when Tempfile then path_or_file
            end
    buffer = ""
    self.rewind
    while self.read(in_blocks_of, buffer) do
      dstio.write(buffer)
    end
    dstio.rewind    
    dstio
  end
end

class IO
  include IOStream
end

class Mash
  include IOStream
end

%w( Tempfile StringIO ).each do |klass|
  if Object.const_defined? klass
    Object.const_get(klass).class_eval do
      include IOStream
    end
  end
end

Version data entries

5 entries across 5 versions & 3 rubygems

Version Path
jeremydurham-merb_paperclip-0.9.12 lib/paperclip/iostream.rb
jeremydurham-merb_paperclip-0.9.13 lib/paperclip/iostream.rb
peterpunk-merb_paperclip-0.9.3 lib/paperclip/iostream.rb
peterpunk-merb_paperclip-0.9.4 lib/paperclip/iostream.rb
rughetto-merb_paperclip-0.9.13 lib/paperclip/iostream.rb