Início / dba / Perguntas / 299929
Accepted
Héctor
Asked: 2021-09-22 09:48:29 +0800 CST

PLPGSQL armazena colunas de uma saída de consulta 2D em variáveis ​​de matriz

  • 772

Em alguma etapa de uma função plpgsql eu preciso armazenar um resultado de consulta 2D em variáveis ​​de matriz.

O código a seguir faz o trabalho para escalares (mas falha com matrizes):

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

O seguinte faz o trabalho para UMA coluna e UMA variável de matriz, mas não mais do que isso:

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

Como eu poderia armazenar várias linhas de col_a, b, c, d... em suas respectivas variáveis ​​de matriz sem ter que fazer uma consulta separada para cada coluna? Como no primeiro exemplo de código, mas para várias linhas e matrizes.

postgresql functions

1 respostas

  1. Best Answer

    Use array_agg()para construir as matrizes. Cuidado, ele também agregará NULL, portanto, certifique-se de que é isso que você deseja.

    dbfiddle com dados e funções de exemplo.

    Estou assumindo que o predicado on col_cse destina a ser o mesmo para col_ae col_b. Você terá que usar FILTERou alguma outra construção se eles exigirem predicados separados. Eu coloquei um exemplo de FILTERno violino.

    Tabela de entrada:

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

    Função:

    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$

    Resultado:

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

    Referência: https://www.postgresql.org/docs/current/functions-aggregate.html

    • 2

relate perguntas