discourse/lib/freedom_patches/seed_fu.rb

20 lines
630 B
Ruby

# 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)