mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-17 12:31:52 +03:00
41 lines
1.5 KiB
Haskell
41 lines
1.5 KiB
Haskell
module Hasura.Eventing.Common where
|
|
|
|
import Control.Concurrent.STM.TVar
|
|
import Control.Monad.STM
|
|
import Hasura.Prelude
|
|
import Hasura.RQL.Types.EventTrigger (EventId)
|
|
import Hasura.RQL.Types.ScheduledTrigger (CronEventId, OneOffScheduledEventId)
|
|
|
|
import qualified Data.Set as Set
|
|
|
|
data LockedEventsCtx
|
|
= LockedEventsCtx
|
|
{ leCronEvents :: TVar (Set.Set CronEventId)
|
|
, leOneOffEvents :: TVar (Set.Set OneOffScheduledEventId)
|
|
, leEvents :: TVar (Set.Set EventId)
|
|
}
|
|
|
|
initLockedEventsCtx :: STM LockedEventsCtx
|
|
initLockedEventsCtx = do
|
|
leCronEvents <- newTVar Set.empty
|
|
leOneOffEvents <- newTVar Set.empty
|
|
leEvents <- newTVar Set.empty
|
|
return $ LockedEventsCtx{..}
|
|
|
|
-- | After the events are fetched from the DB, we store the locked events
|
|
-- in a hash set(order doesn't matter and look ups are faster) in the
|
|
-- event engine context
|
|
saveLockedEvents :: (MonadIO m) => [Text] -> TVar (Set.Set Text) -> m ()
|
|
saveLockedEvents eventIds lockedEvents =
|
|
liftIO $ atomically $ do
|
|
lockedEventsVals <- readTVar lockedEvents
|
|
writeTVar lockedEvents $!
|
|
Set.union lockedEventsVals $ Set.fromList eventIds
|
|
|
|
-- | Remove an event from the 'LockedEventsCtx' after it has been processed
|
|
removeEventFromLockedEvents :: MonadIO m => Text -> TVar (Set.Set Text) -> m ()
|
|
removeEventFromLockedEvents eventId lockedEvents =
|
|
liftIO $ atomically $ do
|
|
lockedEventsVals <- readTVar lockedEvents
|
|
writeTVar lockedEvents $! Set.delete eventId lockedEventsVals
|