能否在SQL中实现自定义reduce函数？可用存储过程或函数吗？

完全可以通过存储过程或者自定义聚合函数（UDAF）来实现这类接收多行数据、返回单个标量的reduce逻辑，我自己就经常用这种方法替代复杂连表查询，既能简化代码逻辑还能提升查询效率。

下面分两种主流方案给你举例说明：

一、自定义聚合函数（UDAF）：最贴合内置函数的用法 #

大部分主流数据库（PostgreSQL、MySQL 8.0+、SQL Server等）都支持创建自定义聚合函数，用法和count/sum完全一致，可以直接在SELECT语句中配合GROUP BY使用，性能也更接近原生函数。

举个PostgreSQL的例子，实现一个计算字段乘积的聚合函数（内置没有这个函数）：

-- 第一步：定义累积逻辑的辅助函数
CREATE OR REPLACE FUNCTION product_accum(accum NUMERIC, val NUMERIC)
RETURNS NUMERIC AS $$
BEGIN
  RETURN accum * val;
END;
$$ LANGUAGE plpgsql;

-- 第二步：创建聚合函数
CREATE AGGREGATE product(NUMERIC) (
  SFUNC = product_accum,  -- 每行调用的累积函数
  STYPE = NUMERIC,        -- 累积值的类型
  INITCOND = '1'          -- 乘积的初始值设为1
);

-- 使用方式和内置函数一模一样
SELECT product(price) FROM order_items WHERE order_id = 123;

如果是MySQL 8.0+，官方支持通过C++编写UDAF插件；如果是简单逻辑，也可以用窗口函数结合存储函数变通，但原生UDAF是最优解。

二、存储过程：处理复杂逻辑的灵活选择 #

如果你的数据库对UDAF支持有限，或者需要包含分支判断、多表关联等复杂逻辑，存储过程是更灵活的选择。它可以通过游标遍历目标数据，逐行处理并累积结果，最后输出标量。

比如MySQL里的例子，实现一个计算指定用户所有订单总消费的存储过程：

DELIMITER //

CREATE PROCEDURE calculate_user_total_spend(
  IN target_user_id INT,
  OUT total_spend DECIMAL(10,2)
)
BEGIN
  DECLARE done INT DEFAULT FALSE;
  DECLARE single_order_amount DECIMAL(10,2);
  -- 定义游标，获取目标用户的所有订单金额
  DECLARE order_cursor CURSOR FOR 
    SELECT amount FROM orders WHERE user_id = target_user_id;
  -- 游标遍历结束的处理逻辑
  DECLARE CONTINUE HANDLER FOR NOT FOUND SET done = TRUE;

  -- 初始化累积值
  SET total_spend = 0.0;
  OPEN order_cursor;
  
  -- 遍历游标，累积订单金额
  read_loop: LOOP
    FETCH order_cursor INTO single_order_amount;
    IF done THEN
      LEAVE read_loop;
    END IF;
    SET total_spend = total_spend + single_order_amount;
  END LOOP;
  
  CLOSE order_cursor;
END //

DELIMITER ;

-- 调用存储过程并查看结果
CALL calculate_user_total_spend(456, @total);
SELECT @total AS user_total_spend;

一些实用建议 #

• 优先选择自定义聚合函数：它的性能更优，数据库优化器能更好地处理，写法也更简洁。
• 存储过程适合复杂场景：如果你的reduce逻辑需要多步骤判断、调用其他函数或者关联多个表，存储过程的灵活性会更有优势，但要注意游标在大数据量下的性能问题。
• 注意数据库差异：不同数据库的语法细节差异很大（比如SQL Server的游标写法、UDAF创建语法），需要根据你使用的数据库调整代码。

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