postgresql - 使用主键和外键构建树

Question

我对 postgres 还很陌生，尤其是 ltree。在网上搜索 ltree 时，我找到了通过链接字符构建树的示例。但我想使用主键和外键。

因此我构建了下表:

create table fragment(
        id serial primary key,
        description text,
        path ltree
    );
create index tree_path_idx on fragment using gist (path);

我想要 1.3.5，而不是 A.B.G。在线示例中的根是这样添加的:

insert into fragment (description, path) values ('A', 'A');

我想要主键而不是 A(我当时不知道)。有办法做到这一点吗？

添加 child 时我遇到了同样的问题:

insert into tree (letter, path) values ('B', '0.??');

我知道父级的 ID，但不知道我想要附加的子级的 ID。

有办法做到这一点还是我完全偏离了轨道？

非常感谢!

Answer 1

您可以创建一个触发器，在每次插入之前修改path。例如，使用此设置:

DROP TABLE IF EXISTS fragment;
CREATE TABLE fragment(
    id serial primary key
    , description text
    , path ltree
);
CREATE INDEX tree_path_idx ON fragment USING gist (path);

定义触发器:

CREATE OR REPLACE FUNCTION before_insert_on_fragment()
RETURNS TRIGGER LANGUAGE plpgsql AS $$
BEGIN
    new.path := new.path ||  new.id::text;
    return new;
END $$;

DROP TRIGGER IF EXISTS before_insert_on_fragment ON fragment;
CREATE TRIGGER before_insert_on_fragment
BEFORE INSERT ON fragment
FOR EACH ROW EXECUTE PROCEDURE before_insert_on_fragment();

测试触发器:

INSERT INTO fragment (description, path) VALUES ('A', '');
SELECT * FROM fragment;
-- | id | description | path |
-- |----+-------------+------|
-- |  1 | A           |    1 |

现在在 id = 1 下插入 B:

INSERT INTO fragment (description, path) VALUES ('B', (SELECT path FROM fragment WHERE id=1));
SELECT * FROM fragment;

-- | id | description | path |
-- |----+-------------+------|
-- |  1 | A           |    1 |
-- |  2 | B           |  1.2 |

在 B 下插入 C:

INSERT INTO fragment (description, path) VALUES ('C', (SELECT path FROM fragment WHERE description='B'));
SELECT * FROM fragment;

-- | id | description |  path |
-- |----+-------------+-------|
-- |  1 | A           |     1 |
-- |  2 | B           |   1.2 |
-- |  3 | C           | 1.2.3 |