主页 / dba / 问题 / 344673
Accepted
machineghost
Asked: 2025-01-13 02:35:11 +0800 CST

循环遍历 PostgreSQL 结果并仅当第二个查询有结果时才返回结果？

  • 772

我正在尝试编写一个递归 PostgreSQL 函数，其基本内容是：

  1. 获取群组 ID
  2. 如果该组中有一个项目，则返回它
  3. 如果不是，则查找该组的子组
  4. 在每个子组上重复该功能

该函数接受组 ID 和要查找的项目的标题，到目前为止，我所得到的函数主体内容如下：

RETURN QUERY SELECT -- ... return children with title, if any
-- If there were none, look for grandchildren (and so on)
IF NOT FOUND THEN
   -- Loop through the sub-group IDs
   FOR sub_group_id IN
     SELECT -- ... sub-groups of the provided group
   LOOP
     -- Recursion time ... stuck here

我想要表达的是：

  1. 跑步SELECT * FROM this_function(sub_group_id, title);
  2. 如果返回 > 0 个结果，则返回这些结果
  3. 如果没有，则继续循环

但如果我只能说：

LOOP
  RETURN QUERY SELECT * FROM this_function(sub_group_id, title);

它没有达到预期的效果。有人能帮我在 PostgreSQL 中表达上述内容吗？

postgresql

1 个回答

  1. Best Answer

    您可以编写递归 SQL，不需要为此使用函数：

    with recursive
  test as ( select *, title = 'foo' is_match from t ),
  w (id,parent_id,title,is_match) as (
    select * from test where parent_id is null
    union all
    select t.*
    from test t join w on t.parent_id = w.id
    where not w.is_match
  )
select id from w where is_match;
    ID
    1
    21
    231

    小提琴

    如果你需要一个函数，你当然可以将递归 SQL 包装成一个函数：

    create function f(query text) returns setof integer as $$
  with recursive
    test as ( select *, title = query is_match from t ),
    w (id,parent_id,title,is_match) as (
      select * from test where parent_id is null
      union all
      select t.*
      from test t join w on t.parent_id = w.id
      where not w.is_match
    )
  select id from w where is_match;
$$ language sql;

    select * from f('foo');
    f
    1
    21
    231

    小提琴

    • 2

相关问题