如何在F#中对可辨识联合值关联的数据进行分组？

现有方案的问题 #

• 多次List.map+List.choose的方式需要遍历列表N次（N为联合类型的case数量），大数据量下性能开销明显；
• 手动递归的方案虽然是单次遍历，但代码冗余，后续如果Events新增case，必须修改递归函数的模式匹配逻辑和返回值结构，维护成本高。

优化方案 #

方案1：用List.fold实现单次遍历（推荐）

用F#内置的List.fold替代手动递归，代码更简洁，性能和手动递归完全一致，同时符合F#惯用写法：

let splitEvents events =
    events
    // 用fold累积三个类型的列表，初始值为三个空列表
    |> List.fold (fun (aList, bList, cList) event ->
        match event with
        | A a -> (a :: aList, bList, cList)
        | B b -> (aList, b :: bList, cList)
        | C c -> (aList, bList, c :: cList)
    ) ([], [], [])
    // 因为用头插法累积，最后反转得到正确顺序的列表
    |> fun (a, b, c) -> (List.rev a, List.rev b, List.rev c)

这个方案只遍历一次列表，性能拉满，代码也清晰易读，后续新增case只需要在match分支里加对应的逻辑即可。

方案2：封装通用拆分函数（适合多联合类型场景）

如果需要处理多个类似的可辨识联合，可以封装一个通用函数，通过传入判断和收集逻辑来复用代码：

let splitUnion initialState cases =
    List.fold (fun acc item ->
        cases |> List.fold (fun acc' (isMatch, collect) ->
            if isMatch item then collect item acc' else acc'
        ) acc
    ) initialState

// 针对Events的具体使用
let splitEvents =
    splitUnion ([], [], []) [
        (function A _ -> true | _ -> false), (function A a -> fun (aL, bL, cL) -> (a::aL, bL, cL) | _ -> id)
        (function B _ -> true | _ -> false), (function B b -> fun (aL, bL, cL) -> (aL, b::bL, cL) | _ -> id)
        (function C _ -> true | _ -> false), (function C c -> fun (aL, bL, cL) -> (aL, bL, c::cL) | _ -> id)
    ] >> fun (a,b,c) -> (List.rev a, List.rev b, List.rev c)

后续新增case只需要在cases列表里加一行配对逻辑即可，不需要修改核心的拆分函数，扩展性更好。不过这个方案的性能比直接的fold略低一点，适合对扩展性要求更高的场景。

方案3：编译时生成拆分代码（极致性能）

如果追求绝对性能且联合类型不常变动，可以用F# Source Generators自动生成拆分函数。编译时生成的代码和手动写的递归/fold性能完全一致，同时不需要手动维护模式匹配逻辑，但实现Source Generator有一定开发成本，适合大型项目。

总结 #

• 仅针对当前Events类型：优先选方案1，兼顾性能和简洁性；
• 需处理多个联合类型或高扩展性需求：选方案2；
• 极致性能且能接受额外开发成本：选方案3。

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