PostgreSQL
 sql >> Base de Dados >  >> RDS >> PostgreSQL

Como criar uma tabela de árvore sem relacionamento cíclico?


Provavelmente, a implementação SQL mais simples e comum de uma árvore é uma tabela de auto-referência, por exemplo:
create table tree(
    id int primary key, 
    parent int references tree(id));

insert into tree values
    (1, null),
    (2, 1),
    (3, 1),
    (4, 2),
    (5, 4);

Você pode percorrer a árvore de cima para baixo com uma consulta recursiva como esta:
with recursive top_down as (
    select id, parent, array[id] as path
    from tree
    where parent is null
union all
    select t.id, t.parent, path || t.id
    from tree t
    join top_down r on t.parent = r.id
)
select *
from top_down;

 id | parent |   path    
----+--------+-----------
  1 |        | {1}
  2 |      1 | {1,2}
  3 |      1 | {1,3}
  4 |      2 | {1,2,4}
  5 |      4 | {1,2,4,5}
(5 rows)

Veja também esta resposta para um exemplo de baixo para cima.

Integridade


Você não pode remover um nó que é pai de outro. A chave estrangeira impede que a árvore seja dividida em partes separadas:
delete from tree
where id = 2;

ERROR:  update or delete on table "tree" violates foreign key constraint "tree_parent_fkey" on table "tree"
DETAIL:  Key (id)=(2) is still referenced from table "tree".    

Opcionalmente, você pode garantir que a árvore tenha apenas uma raiz usando um índice exclusivo parcial:
create unique index tree_one_root_idx on tree ((parent is null)) where parent is null;

insert into tree
values(6, null);

ERROR:  duplicate key value violates unique constraint "tree_one_root_idx"
DETAIL:  Key ((parent IS NULL))=(t) already exists. 

Ciclos


Você pode eliminar a possibilidade de inserir ciclos usando um gatilho. A função verifica se um dos ancestrais de um nó inserido ou atualizado pode ser o próprio nó:
create or replace function before_insert_or_update_on_tree()
returns trigger language plpgsql as $$
declare rec record;
begin
    if exists(
        with recursive bottom_up as (
            select new.id, new.parent, array[]::int[] as path, false as cycle
        union all
            select r.id, t.parent, path || t.id, new.id = any(path)
            from tree t
            join bottom_up r on r.parent = t.id and not cycle
        )
        select *
        from bottom_up
        where cycle or (id = parent))
    then raise exception 'Cycle detected on node %.', new.id;
    end if;
    return new;
end $$;

create trigger before_insert_or_update_on_tree
before insert or update on tree
for each row execute procedure before_insert_or_update_on_tree();

Verificar:
insert into tree values (6, 7), (7, 6);

ERROR:  Cycle detected on node 7.

update tree
set parent = 4
where id = 2;

ERROR:  Cycle detected on node 2.