tmapmvar 0.0.0 → 0.0.1

raw patch · 2 files changed

+87/−1 lines, 2 filesdep +hashabledep +unordered-containers

Dependencies added: hashable, unordered-containers

Files

• src/Control/Concurrent/STM/TMapMVar/Hash.hs +83/−0
• tmapmvar.cabal +4/−1

@@ -0,0 +1,83 @@+module Control.Concurrent.STM.TMapMVar.Hash where++import Data.Maybe (catMaybes)+import Data.HashMap.Strict (HashMap)+import Data.Hashable (Hashable)+import qualified Data.HashMap.Strict as HashMap+import Control.Monad (void)+import Control.Concurrent.STM (STM)+import Control.Concurrent.STM.TVar (TVar, newTVar, modifyTVar', readTVar)+import Control.Concurrent.STM.TMVar (TMVar, readTMVar, tryReadTMVar, newEmptyTMVar, takeTMVar, tryTakeTMVar, swapTMVar, putTMVar)++++newtype TMapMVar k a = TMapMVar+  { getTMapMVar :: TVar (HashMap k (TMVar a))+  }+++newTMapMVar :: STM (TMapMVar k a)+newTMapMVar = TMapMVar <$> newTVar HashMap.empty++keys :: TMapMVar k a -> STM [k]+keys (TMapMVar m) = HashMap.keys <$> readTVar m++peekElems :: TMapMVar k a -> STM [a]+peekElems t@(TMapMVar m) = do+  m' <- readTVar m+  let ts = HashMap.elems m'+  catMaybes <$> traverse tryReadTMVar ts+++-- | Blocks if it's full+insert :: (Eq k, Hashable k) => TMapMVar k a -> k -> a -> STM ()+insert t k a = do+  x <- getTMVar t k+  putTMVar x a++-- | Doesn't Block+insertForce :: (Eq k, Hashable k) => TMapMVar k a -> k -> a -> STM ()+insertForce t k a = do+  x <- getTMVar t k+  void $ swapTMVar x a+++-- | Blocks, and deletes upon looking it up+lookup :: (Eq k, Hashable k) => TMapMVar k a -> k -> STM a+lookup t k = do+  x <- getTMVar t k+  takeTMVar x++tryLookup :: (Eq k, Hashable k) => TMapMVar k a -> k -> STM (Maybe a)+tryLookup t k = do+  x <- getTMVar t k+  tryTakeTMVar x++-- | Blocks, but doesn't delete when looking it up+observe :: (Eq k, Hashable k) => TMapMVar k a -> k -> STM a+observe t k = do+  x <- getTMVar t k+  readTMVar x++tryObserve :: (Eq k, Hashable k) => TMapMVar k a -> k -> STM (Maybe a)+tryObserve t k = do+  x <- getTMVar t k+  tryReadTMVar x++delete :: (Eq k, Hashable k) => TMapMVar k a -> k -> STM ()+delete t k = do+  void $ tryLookup t k++++-- * Utils++getTMVar :: (Eq k, Hashable k) => TMapMVar k a -> k -> STM (TMVar a)+getTMVar (TMapMVar m) k = do+  m' <- readTVar m+  case HashMap.lookup k m' of+    Nothing -> do+      t <- newEmptyTMVar+      modifyTVar' m (HashMap.insert k t)+      pure t+    Just t -> pure t

@@ -1,5 +1,5 @@ name:                tmapmvar-version:             0.0.0+version:             0.0.1 synopsis:            A single-entity stateful Map in STM, similar to tmapchan -- description: homepage:            https://github.com/athanclark/tmapmvar#readme@@ -16,9 +16,12 @@ library   hs-source-dirs:      src   exposed-modules:     Control.Concurrent.STM.TMapMVar+                       Control.Concurrent.STM.TMapMVar.Hash   build-depends:       base >= 4.8 && < 5                      , containers+                     , hashable                      , stm+                     , unordered-containers   default-language:    Haskell2010  source-repository head