theatre-dev-0.1: hspec/TheatreDev/StmBasedSpec/IO.hs

module TheatreDev.StmBasedSpec.IO where

import Control.Concurrent.Async
import TheatreDev.StmBased (Actor)
import TheatreDev.StmBased qualified as Actor
import Prelude

simulateReduction :: (Show a) => ([Actor a] -> Actor a) -> Int -> Int -> [a] -> IO [[a]]
simulateReduction reducer actorsNum generatorsNum messages =
  do
    resultsVar <- newTVarIO []

    actor <-
      fmap reducer
        $ replicateM actorsNum
        $ Actor.spawnStatefulIndividual
          []
          ( \state ->
              atomically
                $ modifyTVar' resultsVar (reverse state :)
          )
          ( \state msg ->
              return $ msg : state
          )

    mapConcurrently id
      $ replicate generatorsNum
      $ for_ messages
      $ Actor.tell actor

    Actor.kill actor
    Actor.wait actor

    readTVarIO resultsVar