mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-18 21:12:09 +03:00
This commit applies ormolu to the whole Haskell code base by running `make format`. For in-flight branches, simply merging changes from `main` will result in merge conflicts. To avoid this, update your branch using the following instructions. Replace `<format-commit>` by the hash of *this* commit. $ git checkout my-feature-branch $ git merge <format-commit>^ # and resolve conflicts normally $ make format $ git commit -a -m "reformat with ormolu" $ git merge -s ours post-ormolu https://github.com/hasura/graphql-engine-mono/pull/2404 GitOrigin-RevId: 75049f5c12f430c615eafb4c6b8e83e371e01c8e
42 lines
1.2 KiB
Haskell
42 lines
1.2 KiB
Haskell
module Hasura.GraphQL.Execute.LiveQuery.TMap
|
|
( TMap,
|
|
new,
|
|
reset,
|
|
null,
|
|
lookup,
|
|
insert,
|
|
delete,
|
|
toList,
|
|
)
|
|
where
|
|
|
|
import Control.Concurrent.STM
|
|
import Data.HashMap.Strict qualified as Map
|
|
import Hasura.Prelude hiding (lookup, null, toList)
|
|
|
|
-- | A coarse-grained transactional map implemented by simply wrapping a 'Map.HashMap' in a 'TVar'.
|
|
-- Compared to "StmContainers.Map", this provides much faster iteration over the elements at the
|
|
-- cost of significantly increased contention on writes.
|
|
newtype TMap k v = TMap {unTMap :: TVar (Map.HashMap k v)}
|
|
|
|
new :: STM (TMap k v)
|
|
new = TMap <$> newTVar Map.empty
|
|
|
|
reset :: TMap k v -> STM ()
|
|
reset = flip writeTVar Map.empty . unTMap
|
|
|
|
null :: TMap k v -> STM Bool
|
|
null = fmap Map.null . readTVar . unTMap
|
|
|
|
lookup :: (Eq k, Hashable k) => k -> TMap k v -> STM (Maybe v)
|
|
lookup k = fmap (Map.lookup k) . readTVar . unTMap
|
|
|
|
insert :: (Eq k, Hashable k) => v -> k -> TMap k v -> STM ()
|
|
insert !v k mapTv = modifyTVar' (unTMap mapTv) $ Map.insert k v
|
|
|
|
delete :: (Eq k, Hashable k) => k -> TMap k v -> STM ()
|
|
delete k mapTv = modifyTVar' (unTMap mapTv) $ Map.delete k
|
|
|
|
toList :: TMap k v -> STM [(k, v)]
|
|
toList = fmap Map.toList . readTVar . unTMap
|