Contents

module Populator
  class Factory
    def initialize(model_class, amount)
      @model_class = model_class
      @amount = amount.kind_of?(Integer) ? amount : amount.to_a.rand
      @records = []
    end
    
    def run(&block)
      build_records(&block)
      save_records
    end
    
    private
    
    def quoted_column_names
      @model_class.column_names.map do |column_name|
        @model_class.connection.quote_column_name(column_name)
      end
    end
    
    def last_id
      @model_class.connection.select_value("SELECT id FROM #{@model_class.quoted_table_name} ORDER BY id DESC", "#{@model_class.name} Last ID").to_i
    end
    
    def build_records(&block)
      (1..@amount).map do |i|
        record = Record.new(@model_class, last_id+i)
        block.call(record) if block
        @records << record
      end
    end
    
    def save_records
      @model_class.connection.populate(@model_class.quoted_table_name, columns_sql, rows_sql_arr, "#{@model_class.name} Populate")
    end
    
    def columns_sql
      "(#{quoted_column_names.join(', ')})"
    end
    
    def rows_sql_arr
      @records.map do |record|
        quoted_attributes = record.attribute_values.map { |v| @model_class.sanitize(v) }
        "(#{quoted_attributes.join(', ')})"
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
ryanb-populator-0.1.0	lib/populator/factory.rb