Início / dba / Perguntas / 344673
Accepted
machineghost
Asked: 2025-01-13 02:35:11 +0800 CST

Fazer um loop pelos resultados do PostgreSQL e retornar um resultado somente se uma segunda consulta tiver resultados?

  • 772

Estou tentando escrever uma função recursiva do PostgreSQL que basicamente diz:

  1. Pegue um ID de grupo
  2. Se houver um item naquele grupo, devolva-o
  3. Caso contrário, encontre os subgrupos do grupo
  4. Repita a função em cada um dos subgrupos

A função recebe um ID de grupo e o título de um item a ser encontrado, e o que tenho até agora para o corpo é:

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

O que eu quero expressar é:

  1. Correr SELECT * FROM this_function(sub_group_id, title);
  2. Se retornar > 0 resultados, retorne esses resultados
  3. Se isso não acontecer, continue em loop

Mas se eu apenas disser:

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

não tem o efeito desejado. Alguém pode me ajudar a expressar o acima no PostgreSQL?

postgresql

1 respostas

  1. Best Answer

    Você pode escrever SQL recursivo, não precisa de uma função para isso:

    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;
    eu ia
    1
    21
    231

    violino

    Se você precisar de uma função, é claro que pode encapsular o SQL recursivo em uma:

    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');
    e
    1
    21
    231

    violino

    • 2

relate perguntas