使用状态monad返回值

时间:2012-11-22 15:58:16

标签: haskell monads state-monad

我是Monads和Haskell的新手,并试图了解如何在使用它们时返回值。我的代码如下所示:

foo :: A -> B
foo a = do  b <- fooC a (C 0)
            -- want to return just (B "b")

fooC :: A -> C -> State MyState B
fooC a c = return (B "b")

我尝试使用snd (snd b),但显然State MyState B不是元组?如何返回所需的值(B "b")

编辑:考虑到Daniel的建议,重写如下:

data MyState = MyState String
data C = C Int
foo :: String -> String
-- want to return just "b"
foo a = evalState (fooC a) (C 0)

fooC :: String -> Int -> State MyState String
fooC a c = return "b"

仍然会导致编译错误:

Couldn't match expected type `State s0 String'
            with actual type `Int -> State MyState String'
In the return type of a call of `fooC'
Probable cause: `fooC' is applied to too few arguments
In the first argument of `evalState', namely `(fooC a)'
In the expression: evalState (fooC a) (C 0)

编辑2 :已修复!最终版本如下:

import Control.Monad.State
data MyState = MyState String
data C = C Int
foo :: String -> String
-- want to return just (B "b")
foo a = evalState (fooC a (C 0)) (MyState "whatever")

fooC :: String -> C -> State MyState String
fooC a c = return "b"

main = print(foo("test"))
-- prints "b"

1 个答案:

答案 0 :(得分:6)

您需要的是

foo a = evalState (fooC a (C 0)) (MyState "whatever")

构造State MyState B动作fooC a (C 0),打开它以获取函数,并将该函数应用于初始状态。由于此示例中未使用该状态,因此您也可以使用undefined代替MyState "whatever",但通常情况下,您需要提供有意义的初始状态。

State MyState B不是元组,它与函数同构

MyState -> (B, MyState)

但该函数包含在newtype中(详细信息因monad变换器库包和版本而异),因此要访问应用于初始状态的函数的结果,需要一个展开函数。对于State,有

runState :: State s r -> (s -> (r,s))

它为你提供了返回该对的函数,

evalState :: State s r -> (s -> r)

为您提供由fst组成的函数,因此最终状态将被丢弃,并且

execState :: State s r -> (s -> s)

snd组成函数,因此只返回最终状态。