Go 泛型约束如何设计可复用的分组聚合函数

报表代码里最容易失控的一段，往往不是 SQL，而是把不同业务对象分组后再累计金额、数量或状态。把这段逻辑复制三遍，字段一改就要同时修三处。Go 泛型可以把“怎么取分组键”和“怎么合并一行”变成约束明确的参数，让调用方保留类型检查，又不牺牲可读性。

一个实用的分组聚合函数，核心不是把所有类型都包进去，而是只抽象键和值：键负责定位 map，值负责通过 append 或自定义合并规则形成聚合结果。

先把分组键与聚合值拆成两个类型

假设输入是订单行。每行有客户、月份和金额，今天要按客户分组，后面还可能按月份分组。最先要固定的是边界：K 只表示键，V 表示输入元素，A 表示聚合后的值。

type Order struct {
    Customer string
    Month    string
    Amount   int
}

type KeyFunc[V any, K comparable] func(V) K
type MergeFunc[A any, V any] func(A, V) A

这里的 comparable 直接保证 K 能出现在 map[K]A 中。切片、函数和包含它们的结构体不能作为 map 键，编译器会在调用前指出问题。

GroupBy 如何把 map 累积交给调用方

GroupBy 不应该猜测聚合规则。它只负责遍历输入、调用 key 提取函数、找到当前聚合值，再把新值交给 MergeFunc。收集明细时，合并函数内部用 append；求和时，则显式返回新的整数。

func GroupBy[V any, K comparable, A any](items []V, key KeyFunc[V, K], merge MergeFunc[A, V], zero A) map[K]A {
    result := make(map[K]A)
    for _, item := range items {
        groupKey := key(item)
        current, ok := result[groupKey]
        if !ok { current = zero }
        result[groupKey] = merge(current, item)
    }
    return result
}

zero 只是新分组的起点：切片可以是 nil，数字可以是 0。真正的业务语义仍在 merge 里。

两个调用方式分别验证收集与求和

用 append 收集同组订单

byCustomer := GroupBy(orders,
    func(o Order) string { return o.Customer },
    func(current []Order, item Order) []Order { return append(current, item) },
    []Order(nil),
)

重复客户只会命中同一个 map 槽位，append 将新订单放入已有切片。验收时检查每个客户的订单数和金额明细是否仍与输入一致。

用明确的合并规则计算金额

byMonth := GroupBy(orders,
    func(o Order) string { return o.Month },
    func(current int, item Order) int { return current + item.Amount },
    0,
)

这次聚合值是 int，不必创建中间订单切片。若金额可能超出业务范围，应换成明确的金额类型，并在边界测试中验证溢出策略。

结果顺序与空输入要单独验收

Go 的 map 不承诺遍历顺序。需要展示报表时，先取出 key 排序，再按排序后的 key 读取聚合结果。空输入则应返回长度为零的 map，避免调用方增加 nil 分支。

至少覆盖三组测试：空切片、全部相同键、多个键且输入顺序交错。明细收集检查 append 后的切片长度，金额求和逐组核对金额，而不是只断言分组数量。

常见问题

为什么键类型必须满足 comparable？

因为分组结果使用 map[K]A，Go 要求 map 的键可比较。把约束写在泛型参数上，比运行时反射更早暴露错误。

能不能让 GroupBy 自动决定合并方式？

不建议。收集、求和、取最大值和去重的语义不同，显式传入 MergeFunc 才能让调用点看清业务规则。

为什么结果不能直接按 map 遍历输出？

map 遍历顺序不稳定。需要稳定展示时先排序 key，再读取对应的聚合结果。

把抽象控制在调用边界内

泛型适合消除重复的遍历骨架，不适合把业务字段、排序规则和错误策略全部隐藏起来。保留简短的 GroupBy，让 key 和 merge 在调用点可见，再用空输入、重复键和顺序测试守住结果边界。