主页 / 问题 / 984765
Accepted
Oleg
Asked:2020-05-23 21:31:43 +0000 UTC2020-05-23 21:31:43 +0000 UTC

查找循环键

  • 772

有一个由一些实体组成的表，它们相互嵌套，一个实体从同一个表parent_id指向另一个实体：id

 entity
╔════╦═════════╦═══════════╗
║ id ║ name    ║ parent_id ║
╠════╬═════════╬═══════════╣
║ 0  ║ "Root"  ║ 0         ║ 
║ 1  ║ "One"   ║ 0         ║
║ 2  ║ "Two"   ║ 1         ║
║ 3  ║ "Three" ║ 4         ║
║ 4  ║ "Four"  ║ 3         ║
╚════╩═════════╩═══════════╝

这些实体可以通过 rest 进行编辑（或创建）。一般来说，查询很简单，归结为以下几点：

insert into entity (name, parent_id) values (${name}, ${parentId});
或
update entity set (name = ${name}, parent_id = ${parentId}) where id = ${id};

因为parentId你可以替换任何数字。那么像上面的例子一样，纯理论的情况是可能的：第三个实体指的是第四个，第四个指的是第三个。因此，我们得到了循环嵌套，并且不再可能从这些实体构建有效的树。

我认为parent_id添加实体时可以检查循环依赖关系，但我不知道如何使用Postgres. 有没有人已经为自己解决了这个问题？

来自@vp_arth 评论的UPD想法：

在另一个父项下添加或“移动”后，您可以获取该实体的所有父项的树：

  with recursive r as(
      select *
      from entity
      where id = $id
    union
       select et.*
       from entity as et
       join r
         on et.parent_id = r.id
   )
  select * from r

并在那里寻找给定的实体where id = ${entityId}。

问题在于，在这种形式中，如果存在循环嵌套，则此查询将“挂起”。（这是普遍预期的）

sql

1 个回答

  1. Best Answer

    最简单的做法是设置一个限制：

    alter table entity
    add constraint entity_pkey primary key (id),
    add constraint entity_parent_id_fkey foreign key (parent_id)
        references entity (id) match simple
        on update cascade on delete restrict;

    这将避免当条目 withid=3设置时添加不正确条目的情况parent_id=4，而条目 withid=4尚不存在。此外，它还可以帮助您确保数据库的完整性，因为如果删除了一条记录，则也无法构建树。

    但是，这不会让您避免在更新过程中设置不正确的值。

    在这里，我建议您使用一个函数（或存储过程），您可以在其中遍历链parent_id并防止错误更新：

    create or replace function update_entity(arg_id int, arg_name varchar, arg_parent_id int)
  returns table(id int, name varchar, parent_id int) as
$body$
declare
    parent_ids int[] = array(
        with recursive w_entity as (
            select
                e.id,
                e.parent_id
            from entity e
            where e.id = arg_id

            union

            select
                c.id,
                c.parent_id
            from entity c
            join w_entity p on
                c.parent_id = p.id
        )
        select e.id
        from w_entity e
        where e.parent_id is not null
    );
begin
    if arg_parent_id = any(parent_ids) then
        raise exception 'Некорректный parent_id: %', arg_parent_id;
    end if;

    return query
    update entity as e set
        name = arg_name,
        parent_id = arg_parent_id
    where e.id = arg_id
    returning *;
end;
$body$
  language plpgsql volatile;

    此选项的便利之处在于，一切都将在一个事务中完成，此外，调用函数总是更方便、更短：

    select * from update_entity(1, 'some name', 3)

    顺便说一句，所描述的函数返回修改后的记录，因此如果不需要，请删除returning *. 如果你觉得方便，那么请确保描述returns table(id int, name varchar, parent_id int)与表中列的数量和类型相匹配entity，如果在不更改函数描述的情况下更改列，则会出现错误。

    这种方法的缺点是它使迁移复杂化。如果您需要将所有内容移植到另一个 DBMS，那么重写函数是额外的工作。

    试试看。也许这会对你有所帮助。

    祝你好运！

    • 1

相关问题