对OOP编程人员来说,泛函状态State是一种全新的数据类型。我们在上节做了些介绍,在这节我们讨论一下State类型的应用:用一个具体的例子来示范如何使用State类型。以下是这个例子的具体描述:
模拟一个自动糖果贩售机逻辑:贩售机有两种操作方法:投入硬币和扭动出糖旋钮。贩售机可以处于锁定和放开两种状态。模拟运作跟踪贩售机内当前的糖果和硬币数量。贩售机的操作逻辑要求如下:
1、如果机内有糖的话,投入硬币贩售机从锁定状态进入放开状态
2、在放开状态下扭动旋钮贩售机放出一块糖果后自动进入锁定状态
3、在锁定状态下扭动旋钮贩售机不做反应
4、在放开状态下投入硬币贩售机不做反应
5、没有糖果的贩售机对任何操作都不做反应
我们先把涉及到的数据类型设计出来:
type candy = Int //方便表达
type coin = Int //方便表达
sealed trait Input
case object Coin extends Input //投币
case object Turn extends Input //旋钮
case class Machine(locked: Boolean, candies: candy, coins: coin) //状态类
Machine类型就是需要变迁的状态,状态内容包括锁定状态locked,当前糖果数candies,当前硬币数coins。
我们的模拟函数款式如下:
def simulateMachine(inputs: List[Input]): State[Machine,coin]
我们先做一个比较直接容易明白的版本。首先是贩售机互动逻辑部分:这部分模拟了操作互动流程并在运行过程中对状态进行变迁,最终以输出形式返回最新状态:
def transition(input: Input, machine: Machine): Machine = {
(input, machine) match {
case (_, Machine(_,0,_)) => machine
case (Turn, Machine(true,_,_)) => machine
case (Coin, Machine(false,_,_)) => machine
case (Coin, Machine(true, _, nCoin)) => machine.copy(locked = false, coins = nCoin + 1)
case (Turn, Machine(false, nCandy, _)) => machine.copy(locked = true, candies = nCandy - 1)
}
}
这个transition函数采用了泛函状态维护风格:传入一个状态;返回新状态。流程逻辑部分是通过分析操作动作及贩售机当前状态来决定如何更新状态;一切按照设计方案要求进行。状态machine是个case class实例,这样我们可以使用machine.copy来复制一个新的状态,内容包括locked,candies,coins、又或者在某些情况下,原封不动地返回传入的状态。很明显,transition就是我们需要的状态行为函数,只要嵌入一个State实例就可以随时实现状态变迁了。
transition只能处理单个操作动作,那么如果我们输入一个List的操作动作该如何连续处理呢?既然涉及List,自然想到用递归算法应该能行:
def execute(inputs: List[Input], machine: Machine): Machine = {
inputs match {
case Nil => machine
case h::t => execute(t, transition(h,machine))
}
}
在execute函数里我们对List里每个操作元素进行transition运算,状态machine也在一连串的transition运算中自动更新了。有了这两个函数我们就很容易推断出整体流程了:获取初始状态 >>> 以此初始状态输入操作处理流程并把最终结果设定为当前状态 >>> 读取当前状态:
for {
s0 <- getState //读取起始状态
_ <- setState(execute(inputs,s0)) //人工设定新状态
s1 <- getState //读取当前状态
} yield s1.coins
整个大流程还是比较容易理解的。我们注意到状态变迁采用了临时手工设定方式 setState。整个程序代码和运行结果示范如下:
type candy = Int //方便表达
type coin = Int //方便表达
sealed trait Input
case object Coin extends Input
case object Turn extends Input
case class Machine(locked: Boolean, candies: candy, coins: coin)
def simulateMachine(inputs: List[Input]): State[Machine,coin] = {
def transition(input: Input, machine: Machine): Machine = {
(input, machine) match {
case (_, Machine(_,0,_)) => machine
case (Turn, Machine(true,_,_)) => machine
case (Coin, Machine(false,_,_)) => machine
case (Coin, Machine(true, _, nCoin)) => machine.copy(locked = false, coins = nCoin + 1)
case (Turn, Machine(false, nCandy, _)) => machine.copy(locked = true, candies = nCandy - 1)
}
}
def execute(inputs: List[Input], machine: Machine): Machine = {
inputs match {
case Nil => machine
case h::t => execute(t, transition(h,machine))
}
}
for {
s0 <- getState //读取起始状态
_ <- setState(execute(inputs,s0)) //人工设定新状态
s1 <- getState //读取当前状态
} yield s1.coins
} //> simulateMachine: (inputs: List[ch6.state.Input])ch6.state.State[ch6.state.M
//| achine,ch6.state.coin]
val inputs = List(Coin, Turn, Coin, Turn, Turn, Coin, Coin, Coin, Turn)
//> inputs : List[Product with Serializable with ch6.state.Input] = List(Coin,
//| Turn, Coin, Turn, Turn, Coin, Coin, Coin, Turn)
simulateMachine(inputs).run(Machine(true,3,0)) //> res0: (ch6.state.coin, ch6.state.Machine) = (3,Machine(true,0,3))
以上这段代码考虑到了OOP编程人员的思维模式,采用了分段表达方式使整个程序变得更容易理解。对比起来,下面的例子就可以说是真正的泛函编程风格了。同样针对以上的贩售机模拟逻辑要求,我们将用典型的泛函风格来编程。我们先看看下面两个函数:
def modify[S](f: S => S): State[S,Unit] = {
for {
s0 <- getState
_ <- setState(f(s0))
} yield ()
}
def sequence[S,A](xs: List[State[S,A]]): State[S,List[A]] = {
def go(s: S, actList: List[State[S,A]], acc: List[A]): (List[A],S) = {
actList match {
case Nil => (acc.reverse, s) //纠正排序
case h::t => h.run(s) match {case (a2,s2) => go(s2,t,a2 :: acc) }
}
}
State(s => go(s,xs,List()))
}
modify比较直白:取出当前状态,对这个状态进行变迁后再设成当前状态。sequence稍微复杂一点。我们先从它的类型匹配开始分析:接收一个List[State]、输出State[List],换句话说就是把一连串的状态变成一个状态内的一连串值。这不刚好和我们模拟函数要求匹配吗?我们要求一个函数对一连串的操作动作进行处理后产生一个最终的状态。sequence函数内部已经包含了处理循环,我们不需要execute函数了。但是这个版本的sequence函数比较低级:我是指它使用了递归算法,必须在函数内部实现状态行为run(s)。如果我们用高价一点的函数实现sequence,有可能不需要理会run(s)了:
//用右折叠:输入与输出同排序,但不是tail recursive
def sequenceByRight[S,A](xs: List[State[S,A]]): State[S,List[A]] = {
xs.foldRight(unit[S,List[A]](List())){ (f,acc) => f.map2(acc)(_ :: _) }
}
//用左折叠:输入与输出反排序,是tail recursive
def sequenceByLeft[S,A](l: List[State[S, A]]): State[S, List[A]] = {
l.reverse.foldLeft(unit[S, List[A]](List()))((acc, f) => f.map2(acc)( _ :: _ ))
}
无论用左右折叠算法都可以实现sequence功能。注意:我们没有使用run(s),因为这个东西是在flatMap里,而map2是用flatMap实现的。用这种高阶函数使程序更加简洁。
def simulateMachineFP(inputs: List[Input]): State[Machine,coin] = {
for {
_ <- sequence{
<span style="white-space:pre"> </span>inputs.map {
input => modify {
machine: Machine => {
(input, machine) match {
<span style="white-space:pre"> </span>case (_, Machine(_,0,_)) => machine
case (Turn, Machine(true,_,_)) => machine
case (Coin, Machine(false,_,_)) => machine
case (Coin, Machine(true, nCandy, nCoin)) => Machine(false,nCandy, nCoin+1)
case (Turn, Machine(false, nCandy, nCoin)) => Machine(true, nCandy - 1, nCoin)
}
}
}
}
}
s <- getState
} yield s.coins
} //> simulateMachineFP: (inputs: List[ch6.state.Input])ch6.state.State[ch6.state
//| .Machine,ch6.state.coin]
simulateMachineFP(inputs).run(Machine(true,3,0)) //> res1: (ch6.state.coin, ch6.state.Machine) = (3,Machine(true,0,3))
哇,有点过了!不过这里的确没有分段编码,一口气用sequence完成了编程。那我们还是来分析一下:sequence需要接收一个参数类型是:List[State[S,A]] 我们有一个List[Input],需要把这个List[Input] 变成 List[State[S,A]],很明显,我们需要用map来做这个转换 List[Input].map{ Input => State[S,A]}。modify返回类型State[S,Unit],所以我们用了input => modify,这在类型上是匹配的。modify,顾名思义就是更新状态,我们把状态变迁逻辑都放到了modify函数里,它的返回结果就是最终的状态。
_ <- sequence ...modify 起到了 _ <- setState 的作用,所以我们可以用 s <- getState 把最新的状态读出来。
在以上这个例子里我们采用了泛函编程风格:用类型匹配方式进行了函数组合,虽然说代码可能简单了,但清洁可能就说不上了。需要用类型匹配(type line-up)来分析理解,也就是要再熟悉多点泛函编程思考模式。
后面补充一下:如果我来选择,我会稍退一步;把逻辑部分提出来:
def simulateMachineConcise(inputs: List[Input]): State[Machine,coin] = {
def transition(input: Input, machine: Machine): Machine = {
(input, machine) match {
case (_, Machine(_,0,_)) => machine
case (Turn, Machine(true,_,_)) => machine
case (Coin, Machine(false,_,_)) => machine
case (Coin, Machine(true, nCandy, nCoin)) => Machine(false,nCandy, nCoin+1)
case (Turn, Machine(false, nCandy, nCoin)) => Machine(true, nCandy - 1, nCoin)
}
}
for {
_ <- sequence{inputs.map {input => modify {machine: Machine => transition(input, machine)}}} s <- getState
} yield s.coins
} //> simulateMachineConcise: (inputs: List[ch6.state.Input])ch6.state.State[ch6.
//| state.Machine,ch6.state.coin]
simulateMachineConcise(inputs).run(Machine(true,3,0))
//> res2: (ch6.state.coin, ch6.state.Machine) = (3,Machine(true,0,3))
这段核心代码是不是简洁多了,也比较容易理解:
for {
_ <- sequence{inputs.map {input => modify {machine: Machine => transition(input, machine)}}} s <- getState
} yield s.coins