tmapmvar (empty) → 0.0.0

raw patch · 5 files changed

+141/−0 lines, 5 filesdep +basedep +containersdep +stmsetup-changed

Dependencies added: base, containers, stm

Files

• LICENSE +30/−0
• README.md +1/−0
• Setup.hs +2/−0
• src/Control/Concurrent/STM/TMapMVar.hs +82/−0
• tmapmvar.cabal +26/−0

@@ -0,0 +1,30 @@+Copyright Author name here (c) 2017++All rights reserved.++Redistribution and use in source and binary forms, with or without+modification, are permitted provided that the following conditions are met:++    * Redistributions of source code must retain the above copyright+      notice, this list of conditions and the following disclaimer.++    * Redistributions in binary form must reproduce the above+      copyright notice, this list of conditions and the following+      disclaimer in the documentation and/or other materials provided+      with the distribution.++    * Neither the name of Author name here nor the names of other+      contributors may be used to endorse or promote products derived+      from this software without specific prior written permission.++THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS+"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT+LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR+A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT+OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,+SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT+LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,+DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY+THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

@@ -0,0 +1,1 @@+# tmapmvar

@@ -0,0 +1,2 @@+import Distribution.Simple+main = defaultMain

@@ -0,0 +1,82 @@+module Control.Concurrent.STM.TMapMVar where++import Data.Maybe (catMaybes)+import Data.Map.Strict (Map)+import qualified Data.Map.Strict as Map+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 (Map k (TMVar a))+  }+++newTMapMVar :: STM (TMapMVar k a)+newTMapMVar = TMapMVar <$> newTVar Map.empty++keys :: TMapMVar k a -> STM [k]+keys (TMapMVar m) = Map.keys <$> readTVar m++peekElems :: TMapMVar k a -> STM [a]+peekElems t@(TMapMVar m) = do+  m' <- readTVar m+  let ts = Map.elems m'+  catMaybes <$> traverse tryReadTMVar ts+++-- | Blocks if it's full+insert :: (Ord k) => TMapMVar k a -> k -> a -> STM ()+insert t k a = do+  x <- getTMVar t k+  putTMVar x a++-- | Doesn't Block+insertForce :: (Ord 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 :: (Ord k) => TMapMVar k a -> k -> STM a+lookup t k = do+  x <- getTMVar t k+  takeTMVar x++tryLookup :: (Ord 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 :: (Ord k) => TMapMVar k a -> k -> STM a+observe t k = do+  x <- getTMVar t k+  readTMVar x++tryObserve :: (Ord k) => TMapMVar k a -> k -> STM (Maybe a)+tryObserve t k = do+  x <- getTMVar t k+  tryReadTMVar x++delete :: (Ord k) => TMapMVar k a -> k -> STM ()+delete t k = do+  void $ tryLookup t k++++-- * Utils++getTMVar :: (Ord k) => TMapMVar k a -> k -> STM (TMVar a)+getTMVar (TMapMVar m) k = do+  m' <- readTVar m+  case Map.lookup k m' of+    Nothing -> do+      t <- newEmptyTMVar+      modifyTVar' m (Map.insert k t)+      pure t+    Just t -> pure t

@@ -0,0 +1,26 @@+name:                tmapmvar+version:             0.0.0+synopsis:            A single-entity stateful Map in STM, similar to tmapchan+-- description:+homepage:            https://github.com/athanclark/tmapmvar#readme+license:             BSD3+license-file:        LICENSE+author:              Athan Clark+maintainer:          athan.clark@gmail.com+copyright:           2017 Athan Clark+category:            Data+build-type:          Simple+extra-source-files:  README.md+cabal-version:       >=1.10++library+  hs-source-dirs:      src+  exposed-modules:     Control.Concurrent.STM.TMapMVar+  build-depends:       base >= 4.8 && < 5+                     , containers+                     , stm+  default-language:    Haskell2010++source-repository head+  type:     git+  location: https://github.com/athanclark/tmapmvar