mirror of https://github.com/voidlizard/hbs2
41 lines
1.2 KiB
Haskell
41 lines
1.2 KiB
Haskell
module HBS2.Storage.NCQ3.Internal.Memtable where
|
|
|
|
import HBS2.Storage.NCQ3.Internal.Types
|
|
import HBS2.Storage.NCQ3.Internal.Prelude
|
|
|
|
import Data.ByteString qualified as BS
|
|
import Data.HashMap.Strict qualified as HM
|
|
import Data.Vector qualified as V
|
|
|
|
ncqShardIdx :: NCQStorage3 -> HashRef -> Int
|
|
ncqShardIdx NCQStorage3{..} h =
|
|
fromIntegral (BS.head (coerce h)) `mod` V.length ncqMemTable
|
|
{-# INLINE ncqShardIdx #-}
|
|
|
|
ncqGetShard :: NCQStorage3 -> HashRef -> Shard
|
|
ncqGetShard ncq@NCQStorage3{..} h = ncqMemTable ! ncqShardIdx ncq h
|
|
{-# INLINE ncqGetShard #-}
|
|
|
|
|
|
ncqLookupEntrySTM :: NCQStorage3 -> HashRef -> STM (Maybe NCQEntry)
|
|
ncqLookupEntrySTM ncq h = readTVar (ncqGetShard ncq h) <&> HM.lookup h
|
|
|
|
ncqAlterEntrySTM :: NCQStorage3
|
|
-> HashRef
|
|
-> (Maybe NCQEntry -> Maybe NCQEntry)
|
|
-> STM ()
|
|
ncqAlterEntrySTM ncq h alterFn = do
|
|
let shard = ncqGetShard ncq h
|
|
modifyTVar shard (HM.alter alterFn h)
|
|
|
|
ncqStorageSync3 :: forall m . MonadUnliftIO m => NCQStorage3 -> m ()
|
|
ncqStorageSync3 NCQStorage3{..} = atomically $ writeTVar ncqSyncReq True
|
|
|
|
ncqOperation :: MonadIO m => NCQStorage3 -> m a -> m a -> m a
|
|
ncqOperation ncq m0 m = do
|
|
alive <- readTVarIO (ncqAlive ncq)
|
|
if alive then m else m0
|
|
|
|
|
|
|