如何查找PostgreSQL聚合函数的实现？以stddev_pop()为例

PostgreSQL的内置聚合函数（比如stddev_pop()）不像用户自定义函数那样能直接通过pg_get_functiondef()获取定义，因为它们底层多是C语言实现，且依赖聚合框架的状态转移逻辑。你可以按以下步骤拆解查看：

1. 查询聚合函数的核心组件信息 #

聚合函数的逻辑由状态转移函数、最终计算函数等组件组成，通过查询系统表pg_aggregate关联pg_proc可以获取这些关键信息：

SELECT
  p.proname,
  p.oid AS agg_oid,
  a.aggtransfn::regproc AS trans_func,
  a.aggfinalfn::regproc AS final_func,
  a.aggtranstype::regtype AS state_type
FROM pg_proc p
JOIN pg_aggregate a ON p.oid = a.aggfnoid
WHERE p.proname = 'stddev_pop';

这条查询会返回stddev_pop各个参数版本对应的转移函数（比如float8_accum、numeric_accum）和最终函数（比如float8_stddev_pop、numeric_stddev_pop）。

2. 查看组件函数的定义（针对SQL实现的函数） #

如果转移/最终函数是SQL实现的，可以用pg_get_functiondef()查看，比如：

SELECT pg_get_functiondef('float8_stddev_pop'::regproc);

但对于stddev_pop这类内置函数，这些组件基本都是C语言函数，执行上述语句会提示“函数是内置的”，无法直接获取SQL定义。

3. 查看C语言源码实现 #

PostgreSQL的内置聚合函数源码存放在源码包的src/backend/utils/adt目录下：

• 针对float8类型的标准差计算，实现代码在float8.c文件中，包含float8_accum（状态转移）和float8_stddev_pop（最终计算）的逻辑。
• 数值类型（numeric）的实现对应numeric.c文件，可查找numeric_accum和numeric_stddev_pop相关函数。

以float8_stddev_pop为例，核心逻辑基于累计的总和、平方和、计数计算总体标准差：

Datum
float8_stddev_pop(PG_FUNCTION_ARGS)
{
  ArrayType  *transarray = PG_GETARG_ARRAYTYPE_P(0);
  double     *vals;
  int         n;
  double      sum, sum2, count;
  double      var;

  vals = (double *) ARR_DATA_PTR(transarray);
  n = ARR_DIMS(transarray)[0];
  if (n == 0)
    PG_RETURN_NULL();

  sum = vals[0];
  sum2 = vals[1];
  count = vals[2];

  if (count <= 0)
    PG_RETURN_NULL();

  /* 总体方差 = (sum2 - sum*sum/count) / count */
  var = (sum2 - sum * sum / count) / count;
  if (var < 0)
    var = 0;

  PG_RETURN_FLOAT8(sqrt(var));
}

4. 确认聚合函数的完整执行流程 #

stddev_pop的执行逻辑是：

• 初始化状态变量（存储总和、平方和、计数）
• 遍历每一行数据，调用转移函数更新状态
• 遍历结束后，调用最终函数基于状态计算总体标准差

内容的提问来源于stack exchange，提问作者588chm