mirror of
https://github.com/discourse/discourse.git
synced 2024-12-12 11:33:44 +08:00
20 lines
630 B
Ruby
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)
|