grapesy-1.0.0: test-stress/Test/Stress/Common.hs

module Test.Stress.Common
  ( -- * Logging
    say

    -- * Miscellaneous
  , randomMsg
  ) where

import Data.ByteString.Lazy qualified as Lazy
import System.Random

-- | Generate a random 'Lazy.ByteString' between 128 and 256 bytes in length
randomMsg :: IO Lazy.ByteString
randomMsg = do
    g1 <- getStdGen
    let (l, g2) = randomR (128, 256) g1
    return . Lazy.fromStrict . fst $ genByteString l g2

-- | Log the message, if logging is enabled
say :: Bool -> String -> IO ()
say enabled msg
    | enabled
    = putStrLn $ "test-stress: " ++ msg
    | otherwise
    = return ()