# Difference between revisions of "New monads/MaybeT"

From HaskellWiki

(added MonadIO and MonadState instances) |
(Add Reader and Writer instances) |
||

Line 22: | Line 22: | ||

instance Monad m => Monad (MaybeT m) where | instance Monad m => Monad (MaybeT m) where | ||

− | return = MaybeT . return . | + | return = MaybeT . return . Just |

x >>= f = MaybeT $ runMaybeT x >>= maybe (return Nothing) (runMaybeT . f) | x >>= f = MaybeT $ runMaybeT x >>= maybe (return Nothing) (runMaybeT . f) | ||

fail _ = MaybeT $ return Nothing | fail _ = MaybeT $ return Nothing | ||

instance Monad m => MonadPlus (MaybeT m) where | instance Monad m => MonadPlus (MaybeT m) where | ||

− | mzero = MaybeT $ return | + | mzero = MaybeT $ return Nothing |

mplus x y = MaybeT $ liftM2 mplus (runMaybeT x) (runMaybeT y) | mplus x y = MaybeT $ liftM2 mplus (runMaybeT x) (runMaybeT y) | ||

Line 33: | Line 33: | ||

instance MonadTrans MaybeT where | instance MonadTrans MaybeT where | ||

− | lift = MaybeT . liftM | + | lift = MaybeT . liftM Just |

-- (Add other MTL instances, and a MonadFix instance) | -- (Add other MTL instances, and a MonadFix instance) | ||

Line 40: | Line 40: | ||

liftIO = lift . liftIO | liftIO = lift . liftIO | ||

− | instance | + | instance MonadState s m => MonadState s (MaybeT m) where |

get = lift get | get = lift get | ||

put = lift . put | put = lift . put | ||

+ | |||

+ | instance MonadReader r m => MonadReader r (MaybeT m) where | ||

+ | ask = lift ask | ||

+ | local f = MaybeT . local f . runMaybeT | ||

+ | |||

+ | instance (Monoid w, MonadWriter w m) => MonadWriter w (MaybeT m) where | ||

+ | tell = lift . tell | ||

+ | listen m = MaybeT (do (mv,w) <- listen (runMaybeT m) | ||

+ | case mv of | ||

+ | Nothing -> return Nothing | ||

+ | Just v -> return (Just (v,w))) | ||

+ | pass m = MaybeT (do mvf <- runMaybeT m | ||

+ | case mvf of | ||

+ | Nothing -> return Nothing | ||

+ | Just (v,f) -> pass (return (Just v,f))) | ||

+ | |||

</haskell> | </haskell> | ||

[[Category:Code]] | [[Category:Code]] |

## Revision as of 18:54, 28 July 2007

The Maybe monad deserves a transformer, just like the other classic monads.

## The code

```
{-# OPTIONS_GHC -fglasgow-exts -fallow-undecidable-instances #-}
module Control.Monad.Maybe
(MaybeT(runMaybeT),
module Control.Monad,
module Control.Monad.Trans)
where
import Control.Monad
import Control.Monad.Trans
import Control.Monad.State
newtype MaybeT m a = MaybeT {runMaybeT :: m (Maybe a)}
instance Functor m => Functor (MaybeT m) where
fmap f x = MaybeT $ fmap (fmap f) $ runMaybeT x
instance Monad m => Monad (MaybeT m) where
return = MaybeT . return . Just
x >>= f = MaybeT $ runMaybeT x >>= maybe (return Nothing) (runMaybeT . f)
fail _ = MaybeT $ return Nothing
instance Monad m => MonadPlus (MaybeT m) where
mzero = MaybeT $ return Nothing
mplus x y = MaybeT $ liftM2 mplus (runMaybeT x) (runMaybeT y)
-- Provide other MTL instances, for convenience
instance MonadTrans MaybeT where
lift = MaybeT . liftM Just
-- (Add other MTL instances, and a MonadFix instance)
instance MonadIO m => MonadIO (maybeT m) where
liftIO = lift . liftIO
instance MonadState s m => MonadState s (MaybeT m) where
get = lift get
put = lift . put
instance MonadReader r m => MonadReader r (MaybeT m) where
ask = lift ask
local f = MaybeT . local f . runMaybeT
instance (Monoid w, MonadWriter w m) => MonadWriter w (MaybeT m) where
tell = lift . tell
listen m = MaybeT (do (mv,w) <- listen (runMaybeT m)
case mv of
Nothing -> return Nothing
Just v -> return (Just (v,w)))
pass m = MaybeT (do mvf <- runMaybeT m
case mvf of
Nothing -> return Nothing
Just (v,f) -> pass (return (Just v,f)))
```