Как массово обновить ID последовательности postgreSQL для всех таблиц

Question

Я импортировал файл Postgres SQL на свой сервер, используя TablePlus (SQL клиент), но после того, как я вставил новую строку, я получил такую ​​ошибку:

SQLSTATE [23505] : Уникальное нарушение: 7 ОШИБКА: повторяющееся значение ключа нарушает уникальное ограничение \ "users_pkey \" ДЕТАЛИ: Ключ (id) = (1) уже существует

Я знаю, что это вызвано тем, что значение последовательности равно 0 и необходимо для обновления с помощью кода ниже:

SELECT setval(_sequence_name_, max(id)) FROM _table_name_;

Но для этого требуется так много времени, если я должен писать во все последовательности таблиц (возможно, сотни последовательностей) одну за другой. Так как же обновить сразу все последовательности?

Answer 1

Предполагая, что все используемые последовательности принадлежат соответствующим столбцам, например, через атрибут serial или identity, вы можете использовать это для сброса всех (принадлежащих) последовательностей в текущей базе данных.

with sequences as (
  select *
  from (
    select table_schema,
           table_name,
           column_name,
           pg_get_serial_sequence(format('%I.%I', table_schema, table_name), column_name) as col_sequence
    from information_schema.columns
    where table_schema not in ('pg_catalog', 'information_schema')
  ) t
  where col_sequence is not null
), maxvals as (
  select table_schema, table_name, column_name, col_sequence,
          (xpath('/row/max/text()',
             query_to_xml(format('select max(%I) from %I.%I', column_name, table_schema, table_name), true, true, ''))
          )[1]::text::bigint as max_val
  from sequences
) 
select table_schema, 
       table_name, 
       column_name, 
       col_sequence,
       coalesce(max_val, 0) as max_val,
       setval(col_sequence, coalesce(max_val, 1)) --<< this will change the sequence
from maxvals;

Первая часть выбирает все последовательности, принадлежащие столбцу. Вторая часть затем использует query_to_xml(), чтобы получить максимальное значение для столбца, связанного с этой последовательностью. И последний SELECT затем применяет это максимальное значение к каждой последовательности, используя setval().

Возможно, вы захотите сначала запустить это без вызова setval(), чтобы проверить, все ли так, как вам нужно.

Answer 2

поскольку ответ @a_horse_with_no_name не работает в моем случае (возможно, что-то не так с файлом SQL), я изменил запрос, как показано ниже, который работает в моем случае.

with sequences as (
  select *
  from (
    select table_schema,
           table_name,
           column_name,
           replace(replace(replace(column_default, '::regclass)', ''), '''', ''), 'nextval(', 'public.') as col_sequence
    from information_schema.columns
    where table_schema not in ('pg_catalog', 'information_schema') and column_default ILIKE 'nextval(%'
  ) t
  where col_sequence is not null
), maxvals as (
  select table_schema, table_name, column_name, col_sequence,
          (xpath('/row/max/text()',
             query_to_xml(format('select max(%I) from %I.%I', column_name, table_schema, table_name), true, true, ''))
          )[1]::text::bigint as max_val
  from sequences
) 
select table_schema, 
       table_name, 
       column_name, 
       col_sequence,
       coalesce(max_val, 0) as max_val,
       setval(col_sequence, coalesce(max_val, 1)) --<< this will change the sequence
from maxvals;

Я просто измените pg_get_serial_sequence(format('%I.%I', table_schema, table_name), column_name) as col_sequence на replace(replace(replace(column_default, '::regclass)', ''), '''', ''), 'nextval(', 'public.') as col_sequence.

Возможно, мой запрос не слишком хорош, мне следует использовать регулярное выражение вместо множественной замены. но в моем случае это работает на 100%.

Answer 3

Вы не можете обновить всю последовательность вместе, поскольку каждая последовательность может содержать разные значения, относящиеся к каждой таблице. Вы должны взять максимальное значение из каждой таблицы и обновить его.

SELECT setval(_sequence_name_, max(id)) FROM _table_name_;