Como posso retornar várias linhas de registros no PL / pgSQL

14

Estou tentando retornar vários registros usando o tipo de dados RECORD, existe uma maneira de acrescentar ao RECORD e adicionar / acrescentar um novo valor a cada iteração para este RECORD.

ou seja, desejo anexar para recque recse torne um conjunto de linhas quando o loop terminar, que eu posso VOLTAR no final da minha função. Atualmente, estou fazendo isso -

SELECT temp_table.col1, temp_table.col2, temp_table.col3
      INTO rec
      FROM temp_table
      WHERE temp_table.col3 = false;

meu código completo está aqui:

CREATE OR REPLACE FUNCTION validation()
  RETURNS RECORD AS $$
DECLARE
        rec RECORD;
        temp_row RECORD;
BEGIN

  CREATE TEMPORARY TABLE temp_table (col1 TEXT, col2 INTEGER, col3 BOOLEAN) ON COMMIT DROP;

  FOR temp_row IN SELECT * FROM staging.validation
  LOOP

    RAISE NOTICE 'sql: %', temp_row.sql;

    EXECUTE format('INSERT INTO temp_table %s', temp_row.sql);

    IF (SELECT DISTINCT temp_table.col3 FROM temp_table WHERE temp_table.col3 = false)=false THEN
      RAISE NOTICE 'there is a false value';

      SELECT temp_table.col1, temp_table.col2, temp_table.col3
      INTO rec
      FROM temp_table
      WHERE temp_table.col3 = false;
    END IF;


  END LOOP;
  RETURN rec;
END; $$
LANGUAGE plpgsql;

Saída atual após SELECT validation();

validation
(crea_ddf,8095,f)

Saída desejada

validation
(crea_ddf,8095,f)
(some_source_system,some_count,f)
(some_other_source_system,some_count,f)
(.....)
hky404
fonte
@EvanCarroll Oi Evan, essa é a minha pergunta, que eu publiquei por lá também ... apenas no caso de alguém sentir falta dela por aqui.
precisa saber é o seguinte
Não tenho certeza do que você está tentando fazer, poderia explicar um pouco mais?
Evan Carroll
1
@ hky404: por favor, não faça postagens cruzadas; isso apenas causa duplicação de esforços.
Martijn Pieters

Respostas:

14

A função precisa retornar um em SETOF RECORDvez de RECORDe ter um RETURN NEXTpor linha em vez de um único RETURN, como em:

CREATE FUNCTION test() RETURNS SETOF RECORD AS $$
DECLARE
 rec record;
BEGIN
  select 1,2 into rec;
  return next rec;

  select 3,4 into rec;
  return next rec;
END $$ language plpgsql;

Chamador:

=> selecione * do teste () como x (a int, b int);
 a | b
--- + ---
 1 | 2
 3 4
(2 linhas)

Observe que o SQL é forte e estático, RECORDe é difícil trabalhar com o pseudo-tipo.
Muitas vezes, é menos complicado usar desde o início um tipo composto com uma definição completa de nomes e tipos para cada coluna, seja com a TABLE(...)sintaxe de um tipo anônimo ou com CREATE TYPEum tipo de nome persistente.

Daniel Vérité
fonte
8

Use setof recorde return next recse desejar retornar vários registros de uma função, exemplo:

create or replace function test_function()
    returns setof record 
    language plpgsql as $$
declare
    rec record;
begin
    for rec in
        select i, format('str%s', i), i/2*2 = i
        from generate_series(1, 3) i
    loop
        return next rec;
    end loop;
end $$;

Essa função precisa ser chamada na cláusula FROM com uma lista de definições de coluna:

select test_function(); -- NO

ERROR:  set-valued function called in context that cannot accept a set  

select * from test_function();  -- NO

ERROR:  a column definition list is required for functions returning "record"

select * from test_function() as (id int, str text, is_even boolean);

 id | str  | is_even 
----+------+---------
  1 | str1 | f
  2 | str2 | t
  3 | str3 | f
(3 rows)

Uma opção melhor é usar returns table(...)e return query:

drop function if exists test_function();
create or replace function test_function()
    returns table (id int, str text, is_even boolean)
    language plpgsql as $$
begin
    return query
        select i, format('str%s', i), i/2*2 = i
        from generate_series(1, 3) i;
    -- you can use return query multiple times
    -- or assign values to columns
    -- and return the row:
    id = 100;
    str = 'extra';
    is_even = true;
    return next; -- without a parameter
end $$;

Uso:

select test_function(); -- possible but rather impractical

 test_function 
---------------
 (1,str1,f)
 (2,str2,t)
 (3,str3,f)
 (100,extra,t)
(4 rows)

select * from test_function();

 id  |  str  | is_even 
-----+-------+---------
   1 | str1  | f
   2 | str2  | t
   3 | str3  | f
 100 | extra | t
(4 rows)
klin
fonte
1

Esta é uma bandeira vermelha ..

  1. Você tem uma mesa validation.
  2. Você move as linhas para uma tabela temporária staging.
  3. Todas as linhas com temp_table.col3IS FALSE retornam ao usuário
  4. Juntamente com outras linhas de uma lista especificada de tabelas em que essa coluna é falsa.
  5. Então você solta a tabela temporária (no commit)

Apenas faça isso ..

WITH t AS ( SELECT true AS runthis FROM staging.validation WHERE col3 IS FALSE )
SELECT *
FROM staging.validation
WHERE t.runthis && col3 = 3
UNION ALL 
  SELECT *
  FROM some_source_system
  WHERE t.runthis && some_source_system.col3 = 3
UNION ALL 
  SELECT *
  FROM some_other_source_system
  WHERE t.runthis && some_other_source_system.col3 = 3;

Você pode até colocar isso em um VIEWse quiser

Como uma nota rodapé

SELECT DISTINCT temp_table.col3
FROM temp_table
WHERE temp_table.col3 = false

O que DISTINCTaqui faz? Apenas limite um. Na verdade, eu argumentaria que isso é ainda mais limpo.

SELECT true
FROM temp_table
WHERE temp_table.col3 = false
LIMIT 1;

Então você não precisa do estranho = false ) = FALSE

Evan Carroll
fonte