FIX: Ensure id sequences are not reset during db:migrate (#14184)

The seed-fu gem resets the sequence on all the tables it touches. In some situations, this can cause primary keys to be re-used. This commit introduces a freedom patch which ensures seed-fu only touches the sequence when it is less than the id of one of the seeded records.
This commit is contained in:
David Taylor 2021-08-30 12:31:22 +01:00 committed by GitHub
parent 08dce4f477
commit ed688bec8c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 38 additions and 0 deletions

View File

@ -0,0 +1,19 @@
# frozen_string_literal: true
SeedFu::Seeder.prepend(Module.new do
def update_id_sequence
max_seeded_id = @data.filter_map { |d| d["id"] }.max
seq = @model_class.connection.execute(<<~SQL)
SELECT last_value
FROM #{@model_class.sequence_name}
SQL
last_seq_value = seq.first["last_value"]
if max_seeded_id && last_seq_value < max_seeded_id
# Update the sequence to start from the highest existing id
@model_class.connection.reset_pk_sequence!(@model_class.table_name)
else
# The sequence is already higher than any of our seeded ids - better not touch it
end
end
end)

View File

@ -0,0 +1,19 @@
# frozen_string_literal: true
require 'rails_helper'
describe "seed-fu patch" do
it "does not modify a sequence on an existing table" do
u = User.create!(username: "test1", email: "test1@example.com")
uid1 = u.id
UserDestroyer.new(Discourse.system_user).destroy(u)
SeedFu.quiet = true
SeedFu.seed
# Ensure sequence was not reset. A new user should have
# id one greater than the last user
u2 = User.create!(username: "test1", email: "test1@example.com")
expect(u2.id).to eq(uid1 + 1)
end
end