不能只用SUM或AVG因业务规则复杂(如出租车计费、加权停留时长),需自定义聚合;CREATE AGGREGATE必须正确定义SFUNC、STYPE和INITCOND三部分,缺一不可。

为什么不能只用 SUM 或 AVG?
因为业务规则往往不是简单加总或平均。比如出租车计费:起步价 3.5 元 + 每公里 2.2 元,还要对单次行程总费用四舍五入到整数元;又比如计算某类用户行为的“加权停留时长”,需按设备类型给不同权重。这些逻辑无法靠内置聚合函数表达,硬写在应用层会导致 SQL 查询失去分组能力、重复计算、难以复用。
CREATE AGGREGATE 的三个必填组件怎么配?
PostgreSQL 自定义聚合必须明确定义状态如何流转,核心是三个部分:
-
SFUNC(state transition function):每处理一行就调用一次,接收上一次结果和当前行值,返回新状态。例如taxi_accum(numeric, numeric, numeric)中第一个参数是累计金额,第二个是当前公里数,第三个是单价(可动态传入) -
STYPE:状态的数据类型,必须与SFUNC返回值一致。常见为numeric、text[]、jsonb等 -
INITCOND:初始状态值,如起步价'3.50',或空数组'{}'。不写则默认为NULL,但很多SFUNC不支持NULL输入,容易报错ERROR: function ... cannot be called with NULL argument
漏掉 INITCOND 或类型不匹配,是新手最常卡住的地方。
字符串拼接、百分位数、JSON 合并怎么写?
不同场景对应不同状态设计模式:
- 拼接字符串(类似 MySQL 的
GROUP_CONCAT):STYPE = text[],SFUNC = array_append,FINALFUNC用array_to_string(current_state, ',')。注意:别直接用text类型做累加器,字符串拼接在大结果集下性能差且易溢出 - 计算第 90 百分位数:
STYPE = numeric[],SFUNC把每行值塞进数组,FINALFUNC对数组排序后插值取位。要小心array_length(results, 1)为空时崩溃,得加IF array_length(results, 1) IS NULL OR array_length(results, 1) = 0 THEN RETURN NULL; END IF; - 合并 JSON 对象(如把多条日志的
metadata字段合并成一个对象):STYPE = jsonb,SFUNC用jsonb_concat($1, $2),但注意 PostgreSQL 原生不提供jsonb_concat,得自己写 PL/pgSQL 函数遍历键值,或用jsonb_object_agg配合子查询替代
调试时 RAISE NOTICE 怎么用才不翻车?
聚合函数执行过程不可见,靠 RAISE NOTICE 打印中间态是最直接的调试方式,但有两个坑:
- 不能在
FINALFUNC里打印最终结果后还返回NULL—— 这会导致整个聚合结果为NULL,且无报错,只默默消失 -
RAISE NOTICE在并行查询中会乱序输出,看不出执行流。建议先关掉并行:SET max_parallel_workers_per_gather = 0;再测试 - 生产环境务必删掉所有
RAISE NOTICE,否则日志爆炸,且影响性能 —— 每次触发都涉及服务器日志缓冲区操作
真正难的不是写出语法正确的聚合函数,而是让状态类型、初始值、过渡逻辑三者严丝合缝。哪怕一个逗号写成中文顿号,INITCOND 就会变成字符串字面量而非数值,后续所有计算全错。

















