试图在 PostgreSQL 中创建聚合函数

Question

我正在尝试在 PostgreSQL 中创建新的聚合函数来代替 sum() 函数

我从手册here开始我的旅程。

因为我想创建一个函数，它接受一个双精度值数组，对它们求和，然后进行一些额外的计算，所以我首先创建了最终函数：

takes double precision as input and gives double precision as output    
DECLARE
      v double precision;
BEGIN
      IF tax > 256 THEN
            v := 256;
      ELSE
            v := tax;
      END IF;
      RETURN v*0.21/0.79;
END;

然后我想创建一个聚合函数，它接受一个双精度值数组并输出一个双精度值供我之前的函数处理。

CREATE AGGREGATE aggregate_ee_income_tax (float8[]) (
  sfunc = array_agg
 ,stype = float8
 ,initcond = '{}'
 ,finalfunc = eeincometax);

当我运行该命令时，我得到的是：

错误：函数array_agg（双精度，双精度[]）确实 不存在

我有点卡在这里，因为手册将array_agg() 列为现有函数。我做错了什么？

另外，当我跑步时：

\da
                     List of aggregate functions
 Schema | Name | Result data type | Argument data types | Description 
--------+------+------------------+---------------------+-------------
(0 rows)

我的安装根本没有聚合函数？还是只列出用户定义的函数？

基本上我想了解的内容：

1) 我可以使用现有函数来汇总我的数组值吗？

2) 我如何知道函数的输入和输出数据类型？文档声称array_agg() 接受任何类型的输入。

3) 我自己的聚合函数有什么问题？

编辑 1

为了提供更多信息和更清晰地了解我正在努力实现的目标：

我对几个表有一个巨大的查询，如下所示：

SELECT sum(tax) ... from (SUBQUERY) as foo group by id

我想用我自己的聚合函数替换那个 sum 函数，这样我就不必在后端进行额外的计算 - 因为它们都可以在数据库级别完成。

编辑 2

接受蚂蚁的回答。由于最终解决方案来自 cmets，因此我将其发布在这里以供参考：

CREATE AGGREGATE aggregate_ee_income_tax (float8)
(
 sfunc = float8pl
,stype = float8
,initcond = '0.0'
,finalfunc = eeincometax
);

Answer 1

Array agg 是一个聚合函数而不是常规函数，因此不能用作新聚合的状态转换函数。您要做的是创建一个聚合函数，该函数具有与 array_agg 相同的状态转换函数和自定义最终函数。

不幸的是，array_agg 的状态转换函数是根据内部数据类型定义的，因此无法重用。幸运的是，核心中有一个现有功能已经可以满足您的需求。

CREATE AGGREGATE aggregate_ee_income_tax (float8)(
    sfunc = array_append,
    stype = float8[],
    initcond = '{}',
    finalfunc = eeincometax);

还请注意，您的类型混淆了，您可能希望将一组浮点数聚合到一个数组，而不是将一组数组聚合到一个浮点数。

Answer 2

除了@Ants 的优秀建议：

1.) 您的最终功能可以简化为：

CREATE FUNCTION eeincometax(float8)
  RETURNS float8 LANGUAGE SQL AS
$func$
SELECT (least($1, 256) * 21) / 79
$func$;

2.) 你好像在处理金钱问题？在这种情况下，我强烈建议使用 numeric（首选）或 money 类型。浮点运算通常不够精确。

3.) 聚合的初始条件可以只是0：

CREATE AGGREGATE aggregate_ee_income_tax(float8)
(
  sfunc     = float8pl
 ,stype     = float8
 ,initcond  = 0
 ,finalfunc = eeincometax
);

4.) 在您的情况下，(least(sum(tax), 256) * 21) / 79 可能比您的自定义聚合更快。 PostgreSQL 提供的聚合函数是用 C 语言编写的，并针对性能进行了优化。我会改用它。