Haskell: IORef chung, MVar?

Question

tôi đã thực hiện các chức năng sau đó là cụ thể cho các đơn nguyên 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 

Ví dụ sử dụng:

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

In "hello" một lần.

Tôi muốn (một chú chó cưng) để làm cho nó hoạt động trong nhiều trường hợp nhất có thể (không chỉ trong IO).

tôi thấy stateref trên hackage và cùng với nó mã của tôi trông như thế này:

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

Có một sự thay thế cho stateref hoặc một cách tốt hơn để sử dụng nó hơn tôi đã làm?

Answer 1

Tôi đã viết lại một ít cheesy ít MonadRef lớp vào một vài dịp riêng biệt cho mục đích cá nhân của riêng tôi và ai đó có thể có trên Hackage, nhưng tôi không thể tìm thấy một trong đó là không bị cản trở với hành lý khác.

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

Sau đó, nó rất dễ dàng để tóm tắt đi thói quen memoization của bạn (mặc dù bạn có thể muốn thay thế IORef trong bối cảnh này với một MVar.)

[Edit: làm rõ verbage]