如何在PL / pgSQL中返回多行记录


14

我正在尝试使用RECORD数据类型返回多个记录,有没有一种方法可以追加到RECORD并在每次迭代中向此RECORD添加/附加新值。

也就是说,我想附加到,rec以便rec在循环结束时成为一组行,我可以在函数末尾返回。目前,我正在这样做-

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

我的完整代码在这里:

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;

电流输出后 SELECT validation();

validation
(crea_ddf,8095,f)

期望的输出

validation
(crea_ddf,8095,f)
(some_source_system,some_count,f)
(some_other_source_system,some_count,f)
(.....)

@EvanCarroll嗨,Evan,这是我的问题,我也张贴在那儿……以防万一有人在这里错过。
hky404

我不确定您要做什么,您能再解释一下吗?
埃文·卡罗尔

1
@ hky404:请不要交叉发布;只会造成重复的努力。
马丁·彼得斯

Answers:


14

该函数需要返回a SETOF RECORD而不是,RECORD并且RETURN NEXT每行有一个而不是single RETURN,如:

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;

呼叫者:

=>从test()中选择*作为x(a int,b int);
 一个| b
--- +-
 1 | 2
 3 | 4
(2列)

注意,由于SQL是强类型和静态类型的,因此RECORD伪类型很难使用。
通常,从一开始就使用具有完整名称和类型定义的复合类型(TABLE(...)对于匿名类型的语法或CREATE TYPE用于永久命名类型的语法),一开始就比较麻烦。


8

如果要从一个函数返回多个记录,请使用setof recordreturn next rec,例如:

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 $$;

需要在FROM子句中使用列定义列表调用此类函数:

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)

更好的选择是使用returns table(...)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 $$;

用法:

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)

1

这是一个红旗。

  1. 你有一张桌子validation
  2. 您将这些行移到临时表中staging
  3. temp_table.col3您返回给用户的任何带有IS FALSE的行
  4. 以及表的指定列表中的任何其他行,其中该列为false。
  5. 然后删除临时表(提交时)

就是这样

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;

你甚至可以把在一VIEW,如果你想

作为旁注

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

这是DISTINCT做什么的?只要做一个LIMIT。实际上,我认为这甚至更清洁。

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

那你就不需要怪异的了 = false ) = FALSE

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.