0

我正在编写一个执行以下操作的函数：

创建具有单个字段的临时表。此字段是来自特定表的最多 5 个变量的总和的结果。

假设我有下表：

create table src (x1 numeric, x2 numeric);
insert into src values (2,1),(5,2),(10,4);

我的代码是：

create or replace function qwert(cod numeric, v1 numeric default 0
    , v2 numeric default 0, v3 numeric default 0, v4 numeric default 0,
    v5 numeric default 0)
returns numeric as
$func$
declare vv numeric;
begin
vv = v1+v2+v3+v4+v5;
execute '
    drop table if exists t' || cod || '; 
    create temporary table t' || cod || ' as 
    select ' || vv || ' ;'
    ;
return vv;
end
$func$ language plpgsql;

如果我运行： select qwert(1, x1,x2) from src;

预期结果是一个表 t1：

 column1 
---------
       3
       7
       14
(3 rows)

结果是：

db1=# select * from t1;
 ?column? 
----------
       14
(1 row)

在我的代码中，行：return vv; 只是在那里检查是否正确创建了 vv。

有人可以帮助你吗？

4

1 回答 1

1

会这样工作：

CREATE OR REPLACE FUNCTION qwert(_tbl text, cols text[])
  RETURNS numeric AS
$func$
BEGIN

EXECUTE format('
     DROP TABLE IF EXISTS %1$I;
     CREATE TEMPORARY TABLE %1$I AS 
     SELECT %2$s AS col_sum FROM src;'
   ,_tbl
   ,(SELECT string_agg(quote_ident(i), ' + ') FROM unnest(cols) i)
    );

RETURN 1;  -- still unclear? Add yourself ...
END
$func$ LANGUAGE PLPGSQL;

称呼：

SELECT qwert('t1', ARRAY['x1','x2']);

或者：

SELECT qwert('t1', '{x1,x2}');

format()需要 Postgres 9.1 或更高版本。

text我对临时表名使用参数，对列名使用参数，然后array of text使用 和 的组合构建表达式。不要忘记为列命名（在我的前任中）。unnest()quote_ident()string_agg()col_sum

有关清理值以用作dba.SE上此相关答案中的标识符的详细信息。您可以通过这种方式传递任意数量的列。

于 2013-10-21T16:40:19.853 回答