如何将函数应用于变体?

时间:2012-04-26 16:57:24

标签: functional-programming ocaml

让这个类型=

type intC = int;;
type boolC = bool;
type stringC = string;;

type component = A of intC | B of boolC | C of stringC;;

如果我想在组件A的类型a上应用函数,我是否需要系统地解构组件?

例如,我必须这样做:

let add comp =
  match comp with 
   | A i -> Some (i + 2) (*only A interests me, I return i + 2*)
   | _ -> None           (*otherwise I return nothing*)

然后对于组件A的任何功能?有什么意思可以避免你的裁员吗?

2 个答案:

答案 0 :(得分:4)

这取决于您将对您的类型执行哪种操作。

@nlucaroni给出的解决方案非常好,但是如果你想做一些更通用(和复杂)的事情,你可以使用记录来保存部分地图功能:

type 'a component_m = {
  a : intC  -> 'a;
  b : boolC -> 'a;
  c : stringC -> 'a;
}

let map_component m = function
  | A a -> m.a a
  | B b -> m.b b
  | C c -> m.c c

let add = map_component {
  a = (fun x -> Some (x + 2)); 
  b = (fun _ -> None);
  c = (fun _ -> None);
}

如果您不想每次都必须编写(fun _ -> None)函数,您还可以使用扩展的默认值:

let none = {
  a = (fun _ -> None);
  b = (fun _ -> None);
  c = (fun _ -> None);
}

let add = map_component { none with a = fun x -> Some (x+2) }

你可以用仿函数做同样的事情,但在我看来,这变得有点过分了。

答案 1 :(得分:3)

您可以将更高阶函数传递给为您执行组件销毁的函数,

let apply_if_a f = function
   | A i          -> Some (f i)
   | (B _ | C _)  -> None

这个的类型是,

val apply_if_a : (int -> 'a) -> component -> 'a option

如您所见,应用于任何A值的函数的多态性。此外,大多数人都远离了所有人,_,而是要彻底解决。