我是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"
答案 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
组成函数,因此只返回最终状态。