在 PostgreSQL 函数中使用递归查询

Question

我有一个包含 (id, name, parent_id ) 列的 table 文件夹。作为父文件夹的文件夹是 parent_id = -1。我需要获取每个文件夹的顶级父 ID。以下查询工作正常，但不适用于 spring data jpa.

WITH RECURSIVE nodes_cte AS (
SELECT tn.id, tn.parent_id, tn.name, tn.id top_id
FROM folder AS tn 
WHERE tn.parent_id =-1
UNION ALL
SELECT c.id, c.parent_id, c.name, p.top_id
FROM nodes_cte AS p, folder AS c 
WHERE c.parent_id = p.id
)
SELECT distinct * FROM nodes_cte

我如何创建 postgresql 函数来 return 与上述查询相同的结果？

Answer 1

创建一个 SQL 函数返回 table:

CREATE OR REPLACE FUNCTION my_function()
RETURNS TABLE (id int, parent_id int, name text, top_id int)
LANGUAGE SQL AS $$
    WITH RECURSIVE nodes_cte AS (
    SELECT tn.id, tn.parent_id, tn.name, tn.id top_id
    FROM folder AS tn 
    WHERE tn.parent_id =-1
    UNION ALL
    SELECT c.id, c.parent_id, c.name, p.top_id
    FROM nodes_cte AS p, folder AS c 
    WHERE c.parent_id = p.id
    )
    SELECT distinct * FROM nodes_cte
$$;

--use:
SELECT * 
FROM my_function();

请注意，返回的 table 中的 name 列应与 table folder.

中的类型相同

在 PostgreSQL 函数中使用递归查询

Use a recursive query in a PostgreSQL function

sql

postgresql

recursive-query