@@ -35,6 +35,11 @@ module CompositePrimaryKeys def composite? false end + + def set_auto_increment_column(column) + cattr_accessor :auto_increment_column + self.auto_increment_column = column + end end module InstanceMethods @@ -103,14 +108,12 @@ module CompositePrimaryKeys quoted_pk_columns = self.class.primary_key.map { |col| connection.quote_column_name(col) } cols = quoted_column_names(attributes_minus_pks) << quoted_pk_columns vals = attributes_minus_pks.values << quoted_id - connection.insert( - "INSERT INTO #{self.class.quoted_table_name} " + - "(#{cols.join(', ')}) " + - "VALUES (#{vals.join(', ')})", - "#{self.class.name} Create", - self.class.primary_key, - self.id - ) + send("#{self.class.auto_increment_column}=", + connection.insert("INSERT INTO #{self.class.quoted_table_name} " \ + "(#{cols.join(', ')}) " \ + "VALUES (#{vals.join(', ')})", + "#{self.class.name} Create")) + @new_record = false return true end lib/composite_primary_keys/base.rb 0efff84e20d818323162859687e4011c2e941856 Tyler McMullen tyler@scribd.com http://github.com/tyler/composite_primary_keys/commit/94f8b5eed7db4484d0e3cf8f260d8fa456118e76 94f8b5eed7db4484d0e3cf8f260d8fa456118e76 2009-05-11T18:34:34-07:00 2009-05-11T18:34:34-07:00 Add set_auto_increment_field so we can update the object's auto-incremented field when inserting. ee4b9daf1d3a2e06d9acc9658ff643cf1fca54f2 Tyler McMullen tyler@scribd.com