主页 / dba / 问题 / 299929
Accepted
Héctor
Asked: 2021-09-22 09:48:29 +0800 CST

PLPGSQL 将 2D 查询输出的列存储到数组变量中

  • 772

在 plpgsql 函数的某个步骤中，我需要将 2D 查询结果存储到数组变量中。

以下代码为标量完成工作（但对数组失败）：

SELECT col_a, col_b
FROM my_table
WHERE col_c = condition
INTO var_a, var_b;

以下为 ONE 列和 ONE 数组变量完成了这项工作，但仅此而已：

SELECT ARRAY(
    SELECT col_a
    FROM my_table
    WHERE col_c > condition
) INTO arr_a;

如何将 col_a、b、c、d... 中的多行存储到它们各自的数组变量中，而不必对每一列进行单独的查询？与第一个代码示例一样，但用于多行和数组。

postgresql functions

1 个回答

  1. Best Answer

    用于array_agg()构建阵列。注意，它也会聚合NULL，所以请确保这是您想要的。

    带有数据和示例函数的dbfiddle 。

    我假设 on 的谓词对于和col_c都是相同的。如果它们需要单独的谓词，您将不得不使用或其他一些构造。我在小提琴中举了一个例子。col_acol_bFILTERFILTER

    输入表：

    CREATE TABLE my_table(col_a int, col_b int, col_c int);
INSERT INTO my_table VALUES
    (1, 11, 21),
    (2, 12, 22),
    (3, 13, 23),
    (4, 14, 24),
    (5, 15, 25),
    (6, 16, 26),
    (7, 17, 27),
    (8, 18, 28),
    (9, 19, 29),
    (10, 20, 30);

    功能：

    CREATE OR REPLACE FUNCTION array_load() RETURNS RECORD
 LANGUAGE plpgsql
 IMMUTABLE PARALLEL SAFE STRICT
AS $function$
DECLARE
    arr_a integer[];
    arr_b integer[];
BEGIN
SELECT array_agg(col_a), array_agg(col_b)
FROM my_table
WHERE col_c > 26
INTO arr_a, arr_b;
RETURN (arr_a, arr_b);
END;
$function$

    输出：

    select array_load();
           array_load               
--------------------------------
 ("{7,8,9,10}","{17,18,19,20}")
(1 row)

    参考：https ://www.postgresql.org/docs/current/functions-aggregate.html

    • 2

相关问题