怎样在PostgreSQL中使用SQL自定义聚合函数解决业务需求

浅敏小哥_4750

浅敏小哥_4750

2026-10-03

767人浏览

原创

postgresql自定义聚合函数是解决特定汇总逻辑的刚需工具,因其执行模型支持“按组逐行累积+终值变形”两阶段计算,而普通函数、子查询或窗口函数无法替代;创建需严格定义状态函数(sfunc)、终值函数(finalfunc)和聚合本身,并确保类型、初始值与执行语义一致。

怎样在postgresql中使用sql自定义聚合函数解决业务需求

PostgreSQL 的自定义聚合函数不是“高级彩蛋”,而是解决特定汇总逻辑的刚需工具——当你发现 COUNT、SUM、STRING_AGG 都没法直接表达业务规则(比如“每单起步价+里程×单价,再四舍五入到十位”),就必须自己造一个聚合。

为什么不能只用普通函数或子查询

普通标量函数(如 taxi_accum)只能处理单行;子查询或窗口函数无法天然支持“按组累积 + 最终变形”的两阶段计算。而聚合函数的执行模型是刚性的:每组内逐行调用状态函数(SFUNC),最后统一调用终值函数(FINALFUNC)。跳过这个模型,就等于绕开 PostgreSQL 的执行器优化路径,容易踩坑:

  • 手动在应用层分组累加 → 网络往返多、并发下易错、无法被 GROUP BY 推送下推
  • 用 ARRAY_AGG + UNNEST 模拟 → 内存暴涨、无索引支持、无法流式处理大组数据
  • 把逻辑写进 SELECT 子句(如 (3.5 + SUM(km)*2.2)::numeric)→ 无法封装定价策略变更,所有 SQL 都得同步改

创建聚合函数的三步实操链

以“出租车计价聚合 taxi(numeric, numeric)”为例,必须严格按顺序定义三个对象:

1. 状态累积函数(SFUNC):taxi_accum(numeric, numeric, numeric)

  • 第一个参数是上一轮结果(初始为 INITCOND),后两个是当前行字段和外部传参
  • 必须返回与第一个参数同类型的值(这里是 numeric)
  • 别漏写 LANGUAGE 'plpgsql' 和 VOLATILE(因涉及浮点运算,不可标记为 IMMUTABLE)

2. 终值处理函数(FINALFUNC):taxi_final(numeric)

Jamboss
Jamboss

一款面向大众用户的AI音乐生成应用,通过人工智能帮助用户快速创作歌曲内容。

下载
  • 只接收一个参数:SFUNC 最终累积出的状态值
  • 这里做四舍五入到十位:round($1 + 5, -1)(加5再取整是经典技巧)
  • 注意:返回类型必须与聚合声明的 STYPE 兼容

3. 聚合本身:CREATE AGGREGATE taxi(numeric, numeric)

  • STYPE = numeric:状态类型,必须和 SFUNC 第一个参数、FINALFUNC 唯一参数一致
  • INITCOND = 3.50:起步价,不能写成字符串 '3.50',否则类型不匹配报错 ERROR: invalid input syntax for type numeric
  • 不指定 COMBINEFUNC 时,该聚合无法并行执行(对大数据量影响明显)

常见错误现象与定位方法

执行 SELECT trip_id, taxi(km, 2.2) FROM t_taxi GROUP BY trip_id 报错?先查这几点:

  • ERROR: function taxi(numeric, numeric) does not exist → 聚合未创建,或当前 schema 不在 search_path 中(用 SELECT current_schemas(true) 检查)
  • ERROR: column "km" must appear in the GROUP BY clause or be used in an aggregate function → 忘了 GROUP BY trip_id,聚合函数不能脱离分组上下文单独使用
  • 结果全为 NULL → FINALFUNC 返回了 NULL(比如数组越界访问 ret[sss+1] 且 sss+1 超出长度),加 RAISE NOTICE 打印中间值最直接
  • 性能极差 → 检查 SFUNC 是否做了不必要的 I/O(如查表、调外部 API),聚合函数内严禁此类操作

字符串拼接类聚合的兼容性陷阱

想实现类似 MySQL 的 GROUP_CONCAT?别直接抄 array_to_comma_string 示例:

  • array_append 在大数据量下内存占用线性增长,比原生 STRING_AGG 慢 3–5 倍
  • PostgreSQL 9.6+ 已内置 STRING_AGG(expr, delimiter),优先用它;自定义仅用于需要特殊分隔逻辑(如“最后一项前加 and”)
  • 若坚持自定义,STYPE 设为 text 比 text[] 更省内存,但 FINALFUNC 得手动处理空值和分隔符边界
  • 注意排序:原生 STRING_AGG 支持 ORDER BY,自定义聚合需在 SFUNC 里维护有序数组,成本陡增

真正难的从来不是写完三个函数,而是让 FINALFUNC 的输出类型、SFUNC 的状态迁移、以及 INITCOND 的默认值,在所有边缘输入(空组、全 NULL 列、超长字符串)下保持行为一致——这点没测试覆盖,上线就等于埋雷。

相关文章

PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

聚合函数

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

2023.10.12

3943

8

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.27

851

4

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

2024.02.23

1029

5

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

5781

10

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

2723

4

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

2024.04.07

5760

11

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

2024.04.29

7621

6

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

1030

5

sql中删除一列的命令是什么
sql中删除一列的命令是什么

在sql中,使用alter table语句可以删除一列,语法为:alter table table_name drop column column_name。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

912

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.6万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.4万人学习