在 PL/pgSQL 中迭代 integer[]

Question

我试图在 plpgsql 函数中循环一个整数数组 (integer[])。像这样的：

declare
    a integer[] = array[1,2,3];
    i bigint;
begin
    for i in a
loop 
    raise notice "% ",i;
end loop;
return true;
end

在我的实际用例中，整数数组a 作为参数传递给函数。我收到此错误：

ERROR:  syntax error at or near "$1"
LINE 1:   $1

如何正确循环遍历数组？

Answer 1

DO
$do$
DECLARE
   a integer[] := array[1,2,3];
   i integer;                      -- int, not bigint
BEGIN
   FOR i IN 1 .. array_upper(a, 1)
   LOOP
      RAISE NOTICE '%', a[i];      -- single quotes
   END LOOP;
END
$do$;

或者在 PostgreSQL 9.1 或更高版本中使用FOREACH 更简单：

   FOREACH i IN ARRAY a
   LOOP 
      RAISE NOTICE '%', i;
   END LOOP;

多维数组见：

• Loop over array dimension in plpgsql

但是，使用 generate_series() 或 unnest() 的基于集合的解决方案通常比在大集合上循环更快。基本示例：

• PostgreSQL: Frequency Table Expansion
• Select each month between a start and end date

搜索标签generate-series 或unnest 了解更多信息。