如何在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;

对你而言,mu的答案更好,但是大多数通过标题找到这个问题的用户都更喜欢我的回答.

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进行检查.