INSERT all columns

lib/active_record_bulk_insert.rb

@@ -14,26 +14,29 @@ def self.bulk_insert_in_batches(attrs, options = {})
   def self.bulk_insert(attrs, options = {})
     return [] if attrs.empty?
 
-    use_provided_primary_key = options.fetch(:use_provided_primary_key, false)
-    attributes = _resolve_record(attrs.first, options).keys.join(", ")
+
+    to_import = columns
+    to_import = to_import.reject { |column| column.name == primary_key } unless options.fetch(:use_provided_primary_key, false)
 
     invalid = []
     if options.fetch(:validate, false)
       attrs, invalid = attrs.partition { |record| _validate(record) }
     end
 
     values_sql = attrs.map do |record|
-      quoted = _resolve_record(record, options).map {|k, v|
-        column = try(:column_for_attribute, k)
-        v = connection.type_cast_from_column(column, v) if column
+      attributes = _resolve_record(record, options)
+      quoted = to_import.map { |column|
+        k = column.name
+        v = attributes[k] || attributes[k.to_sym]
+        v = connection.type_cast_from_column(column, v)
         connection.quote(v)
       }
       "(#{quoted.join(', ')})"
     end.join(",")
 
     sql = <<-SQL
       INSERT INTO #{quoted_table_name}
-        (#{attributes})
+        (#{to_import.map(&:name).join(", ")})
       VALUES
         #{values_sql}
     SQL
@@ -42,10 +45,9 @@ def self.bulk_insert(attrs, options = {})
   end
 
   def self._resolve_record(record, options)
-    time = ActiveRecord::Base.default_timezone == :utc ? Time.now.utc : Time.now
+     time = ActiveRecord::Base.default_timezone == :utc ? Time.now.utc : Time.now
     _record = record.is_a?(ActiveRecord::Base) ? record.attributes : record
     _record.merge!("created_at" => time, "updated_at" => time) unless options.fetch(:disable_timestamps, false)
-    _record = _record.except(primary_key).except(primary_key.to_sym) unless options.fetch(:use_provided_primary_key, false)
     _record
   end
 

spec/sample_record_spec.rb

@@ -16,6 +16,14 @@
       expect {SampleRecord.bulk_insert(records)}.to change{SampleRecord.count}.by(records.size)
     end
 
+    it "inserts records with different attributes" do
+      records = [
+        { :name => "Foo", :age => 30 },
+        { :name => "Bar" }
+      ]
+      expect {SampleRecord.bulk_insert(records)}.to change{SampleRecord.count}.by(records.size)
+    end
+
     it "inserts multiple records into the DB in a single insert statement" do
       records = 10.times.map { |i| {:age => 4, :name => "Foo#{i}"} }