Haskell: genérico IORef, MVar?

Question

I hizo la siguiente función que es específico para la mónada IO:

memoIO :: MonadIO m => m a -> IO (m a) 
memoIO action = do 
    ref <- newMVar Nothing 
    return $ do 
    x <- maybe action return =<< liftIO (takeMVar ref) 
    liftIO . putMVar ref $ Just x 
    return x 

Ejemplo de uso:

main :: IO() 
main = do 
    p <- memoIO $ putStrLn "hello" 
    p 
    p 

Prints "hello" una vez.

Me gustaría (una locura) hacer que funcione en tantos casos como sea posible (no solo en IO).

He encontrado en stateref hackage y con ella mi código es el siguiente:

{-# LANGUAGE FlexibleContexts, FlexibleInstances, MultiParamTypeClasses, Rank2Types, UndecidableInstances #-} 

import Data.MRef 

class (NewMRef r m a, DefaultMRef r m a, PutMRef r m a, TakeMRef r m a) => MRef r m a 
instance (NewMRef r m a, DefaultMRef r m a, PutMRef r m a, TakeMRef r m a) => MRef r m a 

memo :: (MRef r m (Maybe a), Monad s) => (forall x. m x -> s x) -> s a -> m (s a) 
memo liftFunc action = do 
    ref <- newDefaultMRef Nothing 
    return $ do 
    x <- maybe action return =<< liftFunc (takeDefaultMRef ref) 
    liftFunc . putDefaultMRef ref $ Just x 
    return x 

¿Hay una alternativa para stateref o una mejor manera de utilizarlo de lo que hice?

Answer 1

He reescrito una pequeña clase cursi MonadRef en algunas ocasiones para mi uso personal y alguien probablemente tiene una en Hackage, pero no puedo encontrar una que no esté cargada con otro equipaje.

class Monad m => MonadRef m where 
    type Ref m :: * -> * 
    newRef :: a -> Ref m a 
    writeRef :: Ref m a -> -> m() 
    readRef :: Ref m a -> m a 

instance MonadRef IO where 
    type Ref IO = IORef 
    newRef = newIORef 
    writeRef = writeIORef 
    readRef = writeIORef 

instance MonadRef STM where 
    type Ref STM = TVar 
    ... 


instance MonadRef (ST s) where 
    type Ref (ST s) = STRef s 
    ... 

entonces es fácil abstraer su rutina memoization (aunque es probable que desee para reemplazar IORef en este contexto con un MVar.)

[Editar: aclarado verbage]