mirror of
https://github.com/simplex-chat/simplexmq.git
synced 2026-09-01 18:08:36 +00:00
Evgeny PoberezkinandJRoberts
14cb88e725
* remove completed async action handles from memory * name Co-authored-by: JRoberts <8711996+jr-simplex@users.noreply.github.com> Co-authored-by: JRoberts <8711996+jr-simplex@users.noreply.github.com>
26 lines
774 B
Haskell
26 lines
774 B
Haskell
module Simplex.Messaging.Agent.TAsyncs where
|
|
|
|
import Control.Concurrent.STM (stateTVar)
|
|
import Control.Monad.IO.Unlift (MonadUnliftIO)
|
|
import Simplex.Messaging.TMap (TMap)
|
|
import qualified Simplex.Messaging.TMap as TM
|
|
import UnliftIO.Async (Async, async)
|
|
import UnliftIO.STM
|
|
|
|
data TAsyncs = TAsyncs
|
|
{ actionId :: TVar Int,
|
|
actions :: TMap Int (Async ())
|
|
}
|
|
|
|
newTAsyncs :: STM TAsyncs
|
|
newTAsyncs = TAsyncs <$> newTVar 0 <*> TM.empty
|
|
|
|
newAsyncAction :: MonadUnliftIO m => (Int -> m ()) -> TAsyncs -> m ()
|
|
newAsyncAction action as = do
|
|
aId <- atomically $ stateTVar (actionId as) $ \i -> (i + 1, i + 1)
|
|
a <- async $ action aId
|
|
atomically $ TM.insert aId a $ actions as
|
|
|
|
removeAsyncAction :: Int -> TAsyncs -> STM ()
|
|
removeAsyncAction aId = TM.delete aId . actions
|