Contents

class Upsert
  class Buffer
    class << self
      def for(connection, table_name)
        const_get(connection.class.name.gsub(/\W+/, '_')).new connection, table_name
      end
    end

    attr_reader :connection
    attr_reader :table_name
    attr_reader :rows
    attr_writer :async
    
    def initialize(connection, table_name)
      @connection = connection
      @table_name = table_name
      @rows = []
    end

    def async?
      !!@async
    end

    def add(selector, document)
      rows << Row.new(selector, document)
      if sql = chunk
        execute sql
      end
    end

    def clear
      while sql = chunk
        execute sql
      end
    end

    def chunk
      return if rows.empty?
      targets = []
      sql = nil
      begin
        targets << rows.pop
        last_sql = sql
        sql = compose(targets)
      end until rows.empty? or targets.length >= max_targets or sql.length > max_length
      if sql.length > max_length
        raise if last_sql.nil?
        sql = last_sql
        rows << targets.pop
      end
      sql
    end
    
    def cleanup
      clear
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
upsert-0.0.1	lib/upsert/buffer.rb