haskell - 使用 “staircasing” 代码中的 `Maybe` 值重构 `IO`

Question

以下函数f尝试通过使用IO(也许是Int)函数两次来读取Int两次，但是“短路” ”成功读取一个Int后执行:

readInt :: IO (Maybe Int)

f :: IO (Maybe Int)
f = do
  n1 <- readInt
  case n1 of
      Just n' -> return (Just n')
      Nothing -> do
        n2 <- readInt
        case n2 of
              Just n' -> return (Just n')
              Nothing -> return Nothing

有没有好的方法来重构这段代码？如果我将其扩展到三次尝试，这将变得非常棘手......

(我的思考过程:看到这个“楼梯”告诉我也许我应该使用 Maybe 的 Monad 实例，但是因为它已经在 >IO monad，那么我就必须使用 MaybeT(?)。但是，我只需要 一个 的 readInt 来成功，所以 Maybe monad 在第一个 Nothing 上出错的行为在这里是错误的......)

Answer 1

您可以使用 MaybeT 和 MonadPlus 实例来使用 msum:

f :: MaybeT IO Int
f = msum [readInt, readInt, readInt]