6

我需要通过键/索引遍历类型RECORD项,就像我可以在其他编程语言中使用数组结构一样。

例如:

DECLARE
    data1    record;
    data2    text;
...
BEGIN
...
FOR data1 IN
    SELECT
        *
    FROM
        sometable
LOOP

    FOR data2 IN
        SELECT
            unnest( data1 )   -- THIS IS DOESN'T WORK!
    LOOP
        RETURN NEXT data1[data2];   -- SMTH LIKE THIS
    END LOOP;

END LOOP;
4

5 回答 5

13

正如@Pavel 解释的那样,不可能像遍历数组那样简单地遍历记录。但是有几种方法可以解决它 - 取决于您的确切要求。最终,由于您想要返回同一列中的所有值,您需要将它们转换为相同的类型 -text这是明显的共同点,因为每种类型都有一个文本表示。

又快又脏

比如说,你有一个带有integer、 atext和一date列的表。

CREATE TEMP TABLE tbl(a int, b text, c date);
INSERT INTO tbl VALUES
 (1, '1text',     '2012-10-01')
,(2, '2text',     '2012-10-02')
,(3, ',3,ex,',    '2012-10-03')  -- text with commas
,(4, '",4,"ex,"', '2012-10-04')  -- text with commas and double quotes

那么解决方案可以很简单:

SELECT unnest(string_to_array(trim(t::text, '()'), ','))
FROM   tbl t;

适用于前两行,但不适用于第 3 行和第 4 行的特殊情况。
您可以使用文本表示中的逗号轻松解决问题:

SELECT unnest(('{' || trim(t::text, '()') || '}')::text[])
FROM   tbl t
WHERE  a < 4;

这可以正常工作 - 除了第 4 行在文本表示中有双引号。通过将它们加倍来逃脱。但是数组构造函数需要将它们转义为\. 不知道为什么会出现这种不兼容...

SELECT ('{' || trim(t::text, '()') || '}') FROM tbl t WHERE a = 4

产量:

{4,""",4,""ex,""",2012-10-04}

但是你需要:

SELECT '{4,"\",4,\"ex,\"",2012-10-04}'::text[];  -- works

适当的解决方案

如果您事先知道列名,那么干净的解决方案将很简单:

SELECT unnest(ARRAY[a::text,b::text,c::text])
FROM tbl

由于您对熟知类型的记录进行操作,因此您只需查询系统目录:

SELECT string_agg(a.attname || '::text', ',' ORDER  BY a.attnum)
FROM   pg_catalog.pg_attribute a 
WHERE  a.attrelid = 'tbl'::regclass
AND    a.attnum > 0
AND    a.attisdropped = FALSE

将其放入具有动态 SQL 的函数中:

CREATE OR REPLACE FUNCTION unnest_table(_tbl text)
  RETURNS SETOF text LANGUAGE plpgsql AS
$func$
BEGIN

RETURN QUERY EXECUTE '
SELECT unnest(ARRAY[' || (
    SELECT string_agg(a.attname || '::text', ',' ORDER  BY a.attnum)
    FROM   pg_catalog.pg_attribute a 
    WHERE  a.attrelid = _tbl::regclass
    AND    a.attnum > 0
    AND    a.attisdropped = false
    ) || '])
FROM   ' || _tbl::regclass;

END
$func$;

称呼:

SELECT unnest_table('tbl') AS val

回报:

val
-----
1
1text
2012-10-01
2
2text
2012-10-02
3
,3,ex,
2012-10-03
4
",4,"ex,"
2012-10-04

这无需安装额外的模块即可工作。另一种选择是安装hstore扩展并像@Craig 演示的那样使用它。

于 2012-10-26T00:05:14.487 回答
5

PL/pgSQL 并不是真正为您想要做的事情而设计的。它不认为记录是可迭代的,它是可能不同且不兼容的数据类型的元组。

PL/pgSQL 具有EXECUTE动态 SQL,但EXECUTE查询不能直接引用 PL/pgSQL 变量NEW或其他记录。

可以做的是将记录转换为hstore键/值结构,然后遍历hstore. 使用each(hstore(the_record)),它产生一key,value组元组。所有值都转换为它们的text表示。

这个玩具函数通过创建一个匿名来演示对记录的迭代ROW(..)- 它将具有列名f1, f2, f3- 然后将其转换为hstore,迭代其列/值对,并返回每一对。

CREATE EXTENSION hstore;

CREATE OR REPLACE FUNCTION hs_demo()
RETURNS TABLE ("key" text, "value" text)
LANGUAGE plpgsql AS
$$
DECLARE
  data1 record;
  hs_row record;
BEGIN
  data1 = ROW(1, 2, 'test');
  FOR hs_row IN SELECT kv."key", kv."value" FROM each(hstore(data1)) kv
  LOOP
    "key" = hs_row."key";
    "value" = hs_row."value";
    RETURN NEXT;
  END LOOP;
END;
$$;

实际上你永远不会这样写,因为整个循环可以用一个简单的RETURN QUERY语句代替,而且它做同样的事情each(hstore)——所以这只是为了展示它是如何each(hstore(record))工作的,上面的函数永远不应该被实际使用。

于 2012-10-25T23:50:52.930 回答
2

plpgsql 不支持此功能 - 记录不是像其他脚本语言那样的哈希数组 - 它类似于 C 或 ADA,在这些语言中此功能是不可能的。您可以使用其他 PL 语言,如 PLPerl 或 PLPython 或一些技巧 - 您可以使用 HSTORE 数据类型(扩展)或通过动态 SQL 进行迭代

请参阅如何使用动态 SQL 设置复合变量字段的值

但是请求这个功能通常意味着,所以你做错了。当您使用 PL/pgSQL 时,您的想法与使用 Javascript 或 Python 不同

于 2012-10-25T16:13:33.090 回答
0
FOR data2 IN
    SELECT d
    from  unnest( data1 ) s(d)
LOOP
    RETURN NEXT data2;
END LOOP;
于 2012-10-25T10:15:44.683 回答
0

如果您在循环之前对结果进行排序,您会完成您想要的。

for rc in select * from t1 order by t1.key asc loop
 return next rc;
end loop;

会做你需要的。这也是执行此类任务的最快方式。

于 2012-10-25T12:45:18.097 回答