req-conduit-1.0.2: weigh-bench/Main.hs

{-# LANGUAGE DataKinds #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE OverloadedStrings #-}
{-# OPTIONS_GHC -fno-warn-orphans #-}

module Main (main) where

import Control.Exception (throwIO)
import Control.Monad
import Data.ByteString qualified as B
import Data.Conduit (runConduitRes, (.|))
import Data.Conduit.Binary qualified as CB
import Data.Conduit.List qualified as CL
import Data.Int (Int64)
import Network.HTTP.Req
import Network.HTTP.Req.Conduit
import System.IO.Temp
import Weigh

main :: IO ()
main = mainWith $ do
  setColumns [Case, Allocated, GCs, Max]
  io "streaming 1 M request  body" bigRequest (1 * 1024 * 1024)
  io "streaming 2 M request  body" bigRequest (2 * 1024 * 1024)
  io "streaming 4 M request  body" bigRequest (4 * 1024 * 1024)
  io "streaming 8 M request  body" bigRequest (8 * 1024 * 1024)
  io "streaming 1 M response body" bigResponse (1 * 1024 * 1024)
  io "streaming 2 M response body" bigResponse (2 * 1024 * 1024)
  io "streaming 4 M response body" bigResponse (4 * 1024 * 1024)
  io "streaming 8 M response body" bigResponse (8 * 1024 * 1024)

bigRequest :: Int64 -> IO ()
bigRequest size' = do
  let size = (size' `quot` 1024) * 1024
      chunk = B.replicate 1024 0
  let src = CL.replicate (fromIntegral size `quot` 1024) chunk
  void $
    req
      POST
      (httpbin /: "post")
      (ReqBodySource size src)
      ignoreResponse
      mempty

bigResponse :: Int -> IO ()
bigResponse size = withSystemTempFile "req-conduit" $ \_ h ->
  reqBr GET (httpbin /: "stream-bytes" /~ size) NoReqBody mempty $ \r ->
    runConduitRes $
      responseBodySource r .| CB.sinkHandle h

----------------------------------------------------------------------------
-- Instances

instance MonadHttp IO where
  handleHttpException = throwIO

----------------------------------------------------------------------------
-- Helpers

-- | 'Url' representing <https://httpbin.org>.
httpbin :: Url 'Https
httpbin = https "httpbin.org"