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

Comment copier la structure d'une table à une autre avec des contraintes de clé étrangère dans psql ?

Il n'y a pas d'option pour créer automatiquement des clés étrangères dans CREATE TABLE ... LIKE ... .

Pour la documentation :

En pratique, c'est facile avec les outils GUI. Par exemple, dans PgAdmin III :

  • copier la déclaration (DDL) de source_table à l'outil de requête (ctrl-e),
  • modifier la déclaration,
  • exécuter sql.

Dans un script SQL, vous pouvez utiliser la fonction suivante. Hypothèse importante :les clés étrangères de la table source ont des noms corrects, c'est-à-dire que leurs noms contiennent le nom de la table source (ce qui est une situation typique).

create or replace function create_table_like(source_table text, new_table text)
returns void language plpgsql
as $$
declare
    rec record;
begin
    execute format(
        'create table %s (like %s including all)',
        new_table, source_table);
    for rec in
        select oid, conname
        from pg_constraint
        where contype = 'f' 
        and conrelid = source_table::regclass
    loop
        execute format(
            'alter table %s add constraint %s %s',
            new_table,
            replace(rec.conname, source_table, new_table),
            pg_get_constraintdef(rec.oid));
    end loop;
end $$;

Exemple d'utilisation :

create table base_table (base_id int primary key);
create table source_table (id int primary key, base_id int references base_table);

select create_table_like('source_table', 'new_table');

\d new_table

   Table "public.new_table"
 Column  |  Type   | Modifiers 
---------+---------+-----------
 id      | integer | not null
 base_id | integer | 
Indexes:
    "new_table_pkey" PRIMARY KEY, btree (id)
Foreign-key constraints:
    "new_table_base_id_fkey" FOREIGN KEY (base_id) REFERENCES base_table(base_id)