У меня есть таблица PostgreSQL такой формы:
base_id int | mods smallint[]
3 | {7,15,48}
Мне нужно заполнить таблицу этой формы:
combo_id int | base_id int | mods smallint[]
1 | 3 |
2 | 3 | {7}
3 | 3 | {7,15}
4 | 3 | {7,48}
5 | 3 | {7,15,48}
6 | 3 | {15}
7 | 3 | {15,48}
8 | 3 | {48}
Я думаю, что мог бы сделать это, используя функцию, которая делает почти то же самое, перебирая первую таблицу и записывая комбинации во вторую таблицу : . Сгенерировать все комбинации в SQL
Но я новичок в Postgres и не могу понять, как это сделать с помощью plpgsql. Это не должно быть особенно быстрым; он будет запускаться только периодически на бэкэнде. Первая таблица содержит примерно 80 записей, и грубые подсчеты показывают, что мы можем ожидать около 2600 записей для второй таблицы.
Может ли кто-нибудь хотя бы указать мне правильное направление?
Изменить:Крейг :У меня PostgreSQL 9.0. Мне удалось успешно использовать UNNEST():
FOR messvar IN SELECT * FROM UNNEST(mods) AS mod WHERE mod BETWEEN 0 AND POWER(2, @n) - 1
LOOP
RAISE NOTICE '%', messvar;
END LOOP;
но тогда не знал, куда идти дальше.
Изменить:Для справки: в итоге я использовал решение Эрвина с добавлением одной строки для добавления нулевого результата ('{}' )к каждому набору, а особый случай, на который ссылается Эрвин, удален:
CREATE OR REPLACE FUNCTION f_combos(_arr integer[], _a integer[] DEFAULT '{}'::integer[], _z integer[] DEFAULT '{}'::integer[])
RETURNS SETOF integer[] LANGUAGE plpgsql AS
$BODY$
DECLARE
i int;
j int;
_up int;
BEGIN
IF array_length(_arr,1) > 0 THEN
_up := array_upper(_arr, 1);
IF _a = '{}' AND _z = '{}' THEN RETURN QUERY SELECT '{}'::int[]; END IF;
FOR i IN array_lower(_arr, 1).. _up LOOP
FOR j IN i.. _up LOOP
CASE j-i
WHEN 0,1 THEN
RETURN NEXT _a || _arr[i:j] || _z;
ELSE
RETURN NEXT _a || _arr[i:i] || _arr[j:j] || _z;
RETURN QUERY SELECT *
FROM f_combos(_arr[i+1:j-1], _a || _arr[i], _arr[j] || _z);
END CASE;
END LOOP;
END LOOP;
ELSE
RETURN NEXT _arr;
END IF;
END;
$BODY$
Затем я использовал эту функцию для заполнения моей таблицы:
INSERT INTO e_ecosystem_modified (ide_ecosystem, modifiers)
(SELECT ide_ecosystem, f_combos(modifiers) AS modifiers FROM e_ecosystem WHERE ecosystemgroup <> 'modifier' ORDER BY ide_ecosystem, modifiers);
Из 79 строк в моей исходной таблице с максимум 7 элементами в массиве модификаторов запрос занял 250 мс, чтобы заполнить 2630 строк в моей выходной таблице. Фантастический.