Sha256: 89c43c61cb482a821039399b5c5238a9bac61cf4222b89049381fa54b05f54d0
Contents?: true
Size: 1.84 KB
Versions: 33
Compression:
Stored size: 1.84 KB
Contents
module TSV
class Dumper
attr_accessor :in_stream, :stream, :options, :filename
def self.stream(options = {}, filename = nil, &block)
dumper = TSV::Dumper.new options, filename
Thread.new(Thread.current) do |parent|
begin
yield dumper
dumper.close
rescue Exception
raise $!
end
end
dumper.stream
end
def initialize(options, filename = nil)
if TSV === options
@options = options.options.merge(:key_field => options.key_field, :fields => options.fields)
@filename ||= options.filename
else
@options = options
@filename = filename
end
@filename ||= Misc.fingerprint options
@stream, @in_stream = Misc.pipe
end
def self.values_to_s(values, fields = nil)
case values
when nil
if fields.nil? or fields.empty?
"\n"
else
"\t" << ([""] * fields.length) * "\t" << "\n"
end
when Array
"\t" << values.collect{|v| Array === v ? v * "|" : v} * "\t" << "\n"
else
"\t" << values.to_s << "\n"
end
end
def init
options = @options.dup
key_field, fields = Misc.process_options options, :key_field, :fields
str = TSV.header_lines(key_field, fields, options)
Thread.pass while IO.select(nil, [@in_stream],nil,1).nil?
@in_stream.puts str
end
def add(k,v)
@fields ||= @options[:fields]
begin
Thread.pass while IO.select(nil, [@in_stream],nil,1).nil?
@in_stream << k << TSV::Dumper.values_to_s(v, @fields)
rescue IOError
rescue Exception
raise $!
end
end
def close_out
@stream.close unless @stream.closed?
end
def close_in
@in_stream.close unless @in_stream.closed?
end
def close
close_in
end
end
end
Version data entries
33 entries across 33 versions & 1 rubygems