module STMContainers.Map
(
Map,
Key,
new,
insert,
delete,
lookup,
focus,
foldM,
null,
)
where
import STMContainers.Prelude hiding (insert, delete, lookup, alter, foldM, toList, empty, null)
import qualified STMContainers.HAMT as HAMT
import qualified STMContainers.HAMT.Nodes as HAMTNodes
import qualified Focus
newtype Map k v = Map (HAMT.HAMT (Association k v))
type Key a = (Eq a, Hashable a)
type Association k v = (k, v)
instance (Eq k) => HAMTNodes.Element (Association k v) where
type ElementKey (Association k v) = k
elementKey (k, v) = k
associationValue :: Association k v -> v
associationValue (_, v) = v
lookup :: (Key k) => k -> Map k v -> STM (Maybe v)
lookup k = focus Focus.lookupM k
insert :: (Key k) => v -> k -> Map k v -> STM ()
insert !v !k (Map h) = HAMT.insert (k, v) h
delete :: (Key k) => k -> Map k v -> STM ()
delete k (Map h) = HAMT.focus Focus.deleteM k h
focus :: (Key k) => Focus.StrategyM STM v r -> k -> Map k v -> STM r
focus f k (Map h) = HAMT.focus f' k h
where
f' = (fmap . fmap . fmap) (\v -> k `seq` v `seq` (k, v)) . f . fmap associationValue
foldM :: (a -> (k, v) -> STM a) -> a -> Map k v -> STM a
foldM s a (Map h) = HAMT.foldM s a h
new :: STM (Map k v)
new = Map <$> HAMT.new
null :: Map k v -> STM Bool
null (Map h) = HAMT.null h