Batcher test.

gistfile1.hs

import Control.Concurrent.Lifted
import Control.Monad.Base
import Data.IORef.Lifted
import Data.Traversable
import qualified Data.HashSet as S
import qualified Data.HashMap.Strict as M
import System.IO.Unsafe
import Debug.Trace
:set prompt "ghci> "

let coordinationRef = unsafePerformIO $ newIORef True
let service = (\rs -> (putStr "Looking up: " >> print rs >> putStrLn "") >> (return $ M.fromList $ fmap (\r -> (r, r * 2)) $ S.toList rs)) :: MultiGetService IO Int Int
let temp = batchingService defaultBatchingOptions service
let (singleService, multiService) = unsafePerformIO temp
let test keys = fork (multiService keys >>= (\r -> atomicModifyIORef' coordinationRef (\_ -> (True, traceShow r ()))))
traverse test [S.fromList [1, 2, 3], S.fromList [2, 3, 4], S.fromList [3, 4, 5], S.fromList [1, 3, 5, 7, 9, 11]]

{-
The above prints out this:
Looking up: fromList [1,2,3,4,5,7,9,11]
fromList [(1,2),(3,6),(5,10),(7,14),(9,18),(11,22)]
fromList [(3,6),(4,8),(5,10)]
fromList [(2,4),(3,6),(4,8)]
fromList [(1,2),(2,4),(3,6)]
-}