Z-IO-0.6.2.0: bench/Http.hs

{-# LANGUAGE OverloadedStrings   #-}
{-# LANGUAGE ScopedTypeVariables #-}

{-| HTTP benchmark test

This program read HTTP request(without parsing), and send some respond. You can
use HTTP benchmark tools such as ab or wrk to test IO throughput, remember to
add a proper heap size to improve performance (-Hx parammeter):

ulimit -n 10000
http-bench +RTS -H512M
wrk -c5000 http://localhost:8888

-}

module Main where

import           Control.Monad
import           Foreign.ForeignPtr
import qualified Z.Data.Vector      as V
import qualified Z.Foreign          as FFI
import           Z.IO
import           Z.IO.Network

main :: IO ()
main = startTCPServer defaultTCPServerConfig $ \ uvs ->  do
        recvbuf <- mallocForeignPtrBytes 2048
        sendcontent' <- FFI.pinPrimVector sendcontent
        catch (echo uvs recvbuf sendcontent') (\ (e::SomeException) -> return ())
  where
    echo uvs recvbuf sendcontent' = loop
      where
        loop = do
            r <- withForeignPtr recvbuf $ \ p -> do
                readInput uvs p 2048
            when (r /= 0) $ do
                FFI.withPrimVectorSafe sendcontent' $ writeOutput uvs
                loop

    sendcontent :: V.Bytes
    sendcontent =
        "HTTP/1.1 200 OK\r\n\
        \Content-Type: text/html; charset=UTF-8\r\n\
        \Content-Length: 500\r\n\
        \Connection: Keep-Alive\r\n\
        \\r\n" `V.append` (V.pack $ replicate 500 48)