PL / pgSQL Array of Rows - PullRequest
Новая
       16

PL / pgSQL Array of Rows

0 голосов
/ 09 марта 2012

Возможно ли следующее?
Я хочу, чтобы процедура, написанная на PL / pgSQL, принимала в качестве параметра что-то вроде "коллекции строк", я имею в виду, что мне нужно передать функцию в словарную структуру:

Псевдокод: 

function_name({key1:val1,key2:val2, key3:val3 [, ...] })

Ответы [ 3 ]

7 голосов
/ 10 марта 2012

С современным PostgreSQL вы можете упростить такую ​​функцию.
Тестовая настройка: 

CREATE TABLE tbl1 (id int, value text);

В этом случае нет необходимости явно создавать тип (если тип основан на строке таблицы), он создается неявно для каждой таблицы.
Функция: 

CREATE FUNCTION f_insert_rows_into_tbl1(tbl1[])
  RETURNS VOID AS
$BODY$
    INSERT INTO tbl1 (id,value)
    SELECT (a).*
    FROM   (SELECT unnest($1) AS a) x;
$BODY$ LANGUAGE sql;

Звоните: 

SELECT f_insert_rows_into_tbl1('{"(1,foo)","(2,bar)"}');

Обратите внимание на синтаксис ввода для массива строк!

1 голос
/ 10 марта 2012

Посетите этот сайт http://postgres.cz/wiki/PostgreSQL_SQL_Tricks и найдите Array to table. В более новых версиях могут быть и другие способы, но это было в моих закладках с недавних пор. Источник с сайта: 

CREATE OR REPLACE FUNCTION unpack(anyarray)
RETURNS SETOF anyelement AS $$ 
SELECT $1[i] 
   FROM generate_series(array_lower($1,1), 
                        array_upper($1,1)) g(i);
$$ LANGUAGE sql STRICT IMMUTABLE;

select unpack(array['a','b','c']);
0 голосов
/ 10 марта 2012

Я думаю, что вы можете использовать составной тип и аргумент массива. Что-то вроде (не проверено): 

create type keyvalue as (
  key text,
  value text
);

create function function_name(keyvalue[]) returns null as $$
declare
  keyvalue x;
begin
  foreach x in array $1 loop
    insert into tablename(key,value) values (x.key, x.value);
  end loop
end;
$$ language plpgsql;

Это не будет настоящий словарь, так как вы можете добавить дубликаты keyvalue s и keyvalue s с тем же ключом и другим значением в этот массив. Но для передачи аргументов функции этого должно быть достаточно.

Использование unpack(), упомянутое JustBob , позволит не использовать цикл, который не очень SQL-y, но обрабатывать этот массив как набор.

...