PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Comment mettre à jour en bloc l'ID de séquence postgreSQL pour toutes les tables

En supposant que toutes les séquences utilisées appartiennent aux colonnes respectives, par ex. via un serial ou identity , vous pouvez l'utiliser pour réinitialiser toutes les séquences (possédées) dans la base de données actuelle.

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;

La première partie sélectionne toutes les séquences appartenant à une colonne. La deuxième partie utilise alors query_to_xml() pour obtenir la valeur maximale de la colonne associée à cette séquence. Et le SELECT final applique ensuite cette valeur maximale à chaque séquence en utilisant setval() .

Vous voudrez peut-être l'exécuter sans le setval() appelez d'abord pour voir si tout est comme vous avez besoin.