5

我想做这样的事情：

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');
4

2 回答 2

11

从 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 的答案更好，但大多数通过标题找到这个问题的用户会更喜欢我的答案。

于 2014-05-14T09:03:11.067 回答
8

您不需要循环，您可以使用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 进行检查。

于 2012-08-24T03:44:37.017 回答