OCaml中如何实现支持类型变换的链式map调用优化？

OCaml的类型系统完全支持该需求，不需要复杂的第三方依赖，以下是两种常用的低成本实现方案：

方案1：函数组合（最常用，零额外成本） #

链式调用多次List.map的本质，就是对列表的每个元素依次应用所有转换函数，我们可以直接把所有转换函数组合成一个函数，仅调用一次List.map即可，完全避免生成中间列表。
首先可以根据你的书写习惯定义函数组合运算符：

(* 前向组合运算符：f |>> g 等价于 fun x -> g (f x)，和链式调用顺序一致 *)
let (|>>) f g x = g (f x)

调用示例：

let result = 
  [1; 2; 3] 
  |> List.map (
    (fun x -> x + 1)
    |>> (fun x -> x * 2)
    |>> (fun x -> float_of_int x /. 3.)
  )

运行结果为[1.3333333333333333; 2.0; 2.6666666666666665]，和预期完全一致，仅遍历一次列表，无任何中间数据开销。

方案2：GADT异质列表（支持动态传入函数序列） #

如果你的转换步骤需要在运行时动态调整，无法提前写死，可以通过OCaml原生支持的GADT实现异质函数链来满足需求：

(* 定义异质函数链的GADT类型，支持存储不同类型的转换函数 *)
type (_, _) func_chain =
  | Nil : ('a, 'a) func_chain
  | Cons : ('a -> 'b) * ('b, 'c) func_chain -> ('a, 'c) func_chain

(* 链式map实现 *)
let rec chain_map : type a c. (a, c) func_chain -> a list -> c list =
  fun chain xs ->
    let rec apply : type a c. a -> (a, c) func_chain -> c =
      fun x -> function
        | Nil -> x
        | Cons (f, rest) -> apply (f x) rest
    in
    List.map (fun x -> apply x chain) xs

调用示例：

let result =
  [1; 2; 3]
  |> chain_map (
    Cons (
      (fun x -> x + 1),
      Cons (
        (fun x -> x * 2),
        Cons (
          (fun x -> float_of_int x /. 3.),
          Nil
        )
      )
    )
  )

方案选择建议 #

• 若转换步骤固定，优先选择函数组合方案：代码简洁，无需开启任何语言扩展，编译器优化效率最高
• 若转换步骤需要动态拼接，选择GADT异质列表方案：仅需开启GADT扩展（在代码文件头部添加[@@@ocaml.extension_enable gadt]即可），实现成本极低

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