在复合 StateT/Maybe monad 中,如何取任何一种成功的可能性?

Ada*_*gle 5 monads haskell state-monad monad-transformers

在 Haskell 中,这是一个结合了 State 和 Maybe monad 的 monad:

type StatefulMaybe a = StateT Int Maybe a
Run Code Online (Sandbox Code Playgroud)

这是一个可以成功(返回一个值)或失败的计算。如果成功,它会携带一个状态以及返回值。

我想写一个函数

choice :: StatefulMaybe a -> StatefulMaybe a -> StatefulMaybe a
Run Code Online (Sandbox Code Playgroud)

这需要两次这样的计算并返回第一个成功的(如果有的话)。只有成功计算的状态变化才会被推进。

事实上,经过一些实验,我想出了如何写这个。这里是:

orMaybe :: Maybe a -> Maybe a -> Maybe a
orMaybe (Just x) _ = Just x
orMaybe Nothing x = x

choice :: StatefulMaybe a -> StatefulMaybe a -> StatefulMaybe a
choice mx my = StateT (\s ->
    (runStateT mx s) `orMaybe` (runStateT my s)
  )
Run Code Online (Sandbox Code Playgroud)

有用:

foo :: StatefulMaybe String
foo = do
    modify (+ 20)
    fail "didn't succeed"

baz :: StatefulMaybe String
baz = do
    modify (+ 30)
    return "two"

bar :: StatefulMaybe String
bar = do
    s <- choice foo baz
    return (s ++ " done")

> runStateT bar 0
Just ("two done",30)
Run Code Online (Sandbox Code Playgroud)

我的问题是:有没有比我上面的实现更容易或更自然的方法来编写这个选择函数?特别是,有什么方法可以将orMaybe功能提升到我的 monad 中?

Mar*_*ann 5

如果我正确理解这个问题,我认为这个函数已经存在:<|>

bar :: StatefulMaybe String
bar = do
    s <- foo <|> baz
    return (s ++ " done")
Run Code Online (Sandbox Code Playgroud)

<|>函数是Applicative类型类的一部分,StateT当内部 monad 既是 aFunctor又是MonadPlus实例时,类型类是一个实例,这对于 来说是正确的Maybe

*Q66992923> runStateT bar 0
Just ("two done",30)
Run Code Online (Sandbox Code Playgroud)

正如 luqui 在评论中建议的那样,我认为这mplus也应该有效,因为 的默认实现mplus<|>.