diff --git a/lib/activerecord-reset-pk-sequence.rb b/lib/activerecord-reset-pk-sequence.rb index b8fae5b..7870f4f 100644 --- a/lib/activerecord-reset-pk-sequence.rb +++ b/lib/activerecord-reset-pk-sequence.rb @@ -8,9 +8,9 @@ def self.reset_pk_sequence new_max = maximum(primary_key) || 0 update_seq_sql = "UPDATE sqlite_sequence SET seq = #{new_max} WHERE name = '#{table_name}';" ActiveRecord::Base.connection.execute(update_seq_sql) - when 'Mysql' - new_max = maximum(primary_key) + 1 || 1 - update_seq_sql = "ALTER TABLE '#{table_name}' AUTO_INCREMENT = #{new_max};" + when 'Mysql', 'Mysql2' + new_max = maximum(primary_key).to_i + 1 + update_seq_sql = "ALTER TABLE `#{table_name}` AUTO_INCREMENT = #{new_max};" ActiveRecord::Base.connection.execute(update_seq_sql) when 'PostgreSQL' ActiveRecord::Base.connection.reset_pk_sequence!(table_name)