Haskell: generische IORef, MVar?

Question

Ich habe die folgende Funktion, die für die IO Monade spezifisch ist:

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 

Beispiel Nutzung:

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

Prints "hello" einmal.

Ich möchte (ein Haustier ärgern), damit es für so viele Fälle wie möglich funktioniert (nicht nur in IO).

fand ich stateref auf Hackage und mit ihm mein Code wie folgt aussieht:

{-# 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 

Gibt es eine Alternative für stateref oder einen besseren Weg, es zu benutzen, als ich es tat?

Answer 1

Ich habe eine kitschige kleine MonadRef Klasse für ein paar separate Gelegenheiten für meinen eigenen persönlichen Gebrauch umgeschrieben und jemand hat wahrscheinlich einen auf Hackage, aber ich kann keinen finden, der mit anderem Gepäck unbelastet ist.

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 
    ... 

Dann ist es Ihre memoization Routine zu abstrahieren leicht

(obwohl Sie wahrscheinlich IORef in diesem Zusammenhang mit einem MVar ersetzen möchten.) [Edit: geklärte verbage]