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|
        yield dumper
        dumper.close
      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

1 entries across 1 versions & 1 rubygems

Version	Path
rbbt-util-5.14.0	lib/rbbt/tsv/dumper.rb