Внутри Haskell Monads

Question

У меня есть следующий код, который компилируется и работает нормально.Я попытался сделать его более компактным, заменив case newEmployee of на case scanEmployee p of, но это не сработало.Вероятно, есть простой способ удалить newEmployee (и newTeam) из кода, верно?

module Main( main ) where
import Control.Monad.State

data Employee  = EmployeeSW  Int Int | EmployeeHW Int String deriving ( Show )
data Employee' = EmployeeSW'     Int | EmployeeHW'    String deriving ( Show )

scanTeam :: [Employee] -> State (Int,Int) (Either String [Employee'])
scanTeam [    ] = return (Right [])
scanTeam (p:ps) = do
    newEmployee <- scanEmployee p
    case newEmployee of
        Left errorMsg -> return (Left errorMsg)
        Right e -> do
            newTeam <- scanTeam ps
            case newTeam of
                Right n -> return (Right (e:n))
                Left errorMsg -> return (Left errorMsg)

scanEmployee :: Employee -> State (Int,Int) (Either String Employee')
-- actual code for scanEmployee omitted ...

Answer 1

Вы можете немного упростить свой код с помощью mapM и sequence:

mapM scanEmployee :: [Employee] -> State (Int, Int) [Either String Employee')

sequence :: [ Either String a ] -> Either String [ a ]

(Обратите внимание, что эти сигнатуры типов являются упрощениями, а фактические типы являются более общими. В частности, mapM и sequence работать для любой монады (не только Either String) и любой проходимой (не просто ([])))

И написать простое решение:

scanTeam = fmap sequence . mapM scanEmployee

Answer 2

Вы можете использовать LambdaCase и быть явным с >>= вместо использования do блоков.Результат не намного короче:

scanEmployee p >>= \case
    Left errorMsg -> return (Left errorMsg)
    Right e       -> do ...