如何在pgSQL中使用数组循环

Question

我想做这样的事情：

CREATE OR REPLACE FUNCTION ff(int, text) RETRUNS integer AS $$
DECLARE
    r text;
BEGIN
    FOR r IN SELECT string_to_array($2, ',')
    LOOP
        INSERT INTO test(i, t) VALUES($1, r);
    END LOOP;
    RETRUN 0;
END
$$LANGUAGE pgsql;

我希望函数SELECT ff(3, 'a,b');能完成

INSERT INTO test(i, t) VALUES(3, 'a'); 
INSERT INTO test(i, t) VALUES(3, 'b');

Answer 1

自PostgreSQL 9.1以来，您可以使用FOREACH LOOP对数组进行迭代。文档中的一个示例：

CREATE FUNCTION sum(int[]) RETURNS int8 AS $$
DECLARE
  s int8 := 0;
  x int;
BEGIN
  FOREACH x IN ARRAY $1
  LOOP
    s := s + x;
  END LOOP;
  RETURN s;
END;
$$ LANGUAGE plpgsql;

对你来说，穆的答案更好，但大多数通过标题找到这个问题的用户更喜欢我的答案。

Answer 2

不需要循环，可以使用unnest将数组从string_to_array转换为一组行，然后使用简单的insert ... select构造：

create or replace function ff(int, text) returns integer as $$
begin
    insert into test(i, t)
    select $1, s 
    from unnest(string_to_array($2, ',')) as dt(s);
    return 0;
end
$$ language plpgsql;

一路上我还纠正了一些拼写错误（RETRUNS、RETRUN和pgsql）。

您也可以使用regexp_split_to_table:

create or replace function ff(int, text) returns integer as $$
begin
    insert into test(i, t)
    select $1, s
    from regexp_split_to_table($2, ',') as dt(s);
    return 0;
end
$$ language plpgsql;

如果你被困在8.1石器时代，对此无能为力，那么也许这样的方法会奏效：

create or replace function ff(int, text) returns integer as $$
declare
    a text[]; 
    i int;
begin
    select string_to_array($2, ',') into a; 
    i := 1;
    loop  
        if i > array_upper(a, 1) then
            exit;
        else
            insert into test(i, t) values($1, a[i]);
            i := i + 1;
        end if;
    end loop;                                 
    return 0;       
end
$$ language plpgsql;

我认为这应该在8.1中起作用，但我无法访问8.1进行检查。