mirror of
https://github.com/lexi-lambda/freer-simple.git
synced 2024-12-24 22:54:27 +03:00
193 lines
6.4 KiB
Haskell
193 lines
6.4 KiB
Haskell
{-# LANGUAGE CPP #-}
|
|
{-# LANGUAGE DataKinds #-}
|
|
{-# LANGUAGE DeriveFunctor #-}
|
|
{-# LANGUAGE FlexibleContexts #-}
|
|
{-# LANGUAGE GADTs #-}
|
|
{-# LANGUAGE NoImplicitPrelude #-}
|
|
{-# LANGUAGE NoMonomorphismRestriction #-}
|
|
{-# LANGUAGE ScopedTypeVariables #-}
|
|
{-# LANGUAGE TypeOperators #-}
|
|
module Main where
|
|
|
|
import Prelude ((-))
|
|
|
|
import Control.Applicative (pure)
|
|
import Control.Monad ((>>=), (>>), replicateM_)
|
|
import Data.Either (Either(Left, Right))
|
|
import Data.Function ((.), ($), id)
|
|
import Data.Functor (Functor)
|
|
import Data.Int (Int)
|
|
import Data.Maybe (Maybe)
|
|
import Data.Ord ((<=))
|
|
import Data.String (String)
|
|
import System.IO (IO)
|
|
|
|
#if MIN_VERSION_mtl(2,2,1)
|
|
import qualified Control.Monad.Except as MTL
|
|
#else
|
|
import qualified Control.Monad.Error as MTL
|
|
#endif
|
|
import qualified Control.Monad.State as MTL
|
|
import qualified Control.Monad.Free as Free
|
|
|
|
import Criterion (bench, bgroup, whnf)
|
|
import Criterion.Main (defaultMain)
|
|
|
|
import Control.Monad.Freer (Member, Eff, run, send)
|
|
import Control.Monad.Freer.Internal (Eff(E, Val), decomp, qApp, tsingleton)
|
|
import Control.Monad.Freer.Exception (runError, throwError)
|
|
import Control.Monad.Freer.State (get, put, runState)
|
|
import Control.Monad.Freer.StateRW (ask, tell, runStateR)
|
|
|
|
import qualified Control.Eff as EE (run)
|
|
import qualified Control.Eff.Exception as EE (runExc, throwExc)
|
|
import qualified Control.Eff.State.Lazy as EE (runState, get, put)
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- State Benchmarks --
|
|
--------------------------------------------------------------------------------
|
|
|
|
oneGet :: Int -> (Int, Int)
|
|
oneGet = run . runState get
|
|
|
|
oneGetMTL :: Int -> (Int, Int)
|
|
oneGetMTL = MTL.runState MTL.get
|
|
|
|
oneGetEE :: Int -> (Int, Int)
|
|
oneGetEE n = EE.run $ EE.runState n EE.get
|
|
|
|
countDown :: Int -> (Int, Int)
|
|
countDown start = run (runState go start)
|
|
where go = get >>= (\n -> if n <= 0 then pure n else put (n-1) >> go)
|
|
|
|
countDownRW :: Int -> (Int, Int)
|
|
countDownRW start = run (runStateR go start)
|
|
where go = ask >>= (\n -> if n <= 0 then pure n else tell (n-1) >> go)
|
|
|
|
countDownMTL :: Int -> (Int, Int)
|
|
countDownMTL = MTL.runState go
|
|
where go = MTL.get >>= (\n -> if n <= 0 then pure n else MTL.put (n-1) >> go)
|
|
|
|
countDownEE :: Int -> (Int, Int)
|
|
countDownEE start = EE.run $ EE.runState start go
|
|
where go = EE.get >>= (\n -> if n <= 0 then pure n else EE.put (n-1) >> go)
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- Exception + State --
|
|
--------------------------------------------------------------------------------
|
|
countDownExc :: Int -> Either String (Int,Int)
|
|
countDownExc start = run $ runError (runState go start)
|
|
where go = get >>= (\n -> if n <= (0 :: Int) then throwError "wat" else put (n-1) >> go)
|
|
|
|
countDownExcMTL :: Int -> Either String (Int,Int)
|
|
countDownExcMTL = MTL.runStateT go
|
|
where go = MTL.get >>= (\n -> if n <= (0 :: Int) then MTL.throwError "wat" else MTL.put (n-1) >> go)
|
|
|
|
countDownExcEE :: Int -> Either String (Int,Int)
|
|
countDownExcEE start = EE.run $ EE.runExc (EE.runState start go)
|
|
where go = EE.get >>= (\n -> if n <= (0 :: Int) then EE.throwExc "wat" else EE.put (n-1) >> go)
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- Freer: Interpreter --
|
|
--------------------------------------------------------------------------------
|
|
data Http out where
|
|
Open :: String -> Http ()
|
|
Close :: Http ()
|
|
Post :: String -> Http String
|
|
Get :: Http String
|
|
|
|
open' :: Member Http r => String -> Eff r ()
|
|
open' = send . Open
|
|
|
|
close' :: Member Http r => Eff r ()
|
|
close' = send Close
|
|
|
|
post' :: Member Http r => String -> Eff r String
|
|
post' = send . Post
|
|
|
|
get' :: Member Http r => Eff r String
|
|
get' = send Get
|
|
|
|
runHttp :: Eff (Http ': r) w -> Eff r w
|
|
runHttp (Val x) = pure x
|
|
runHttp (E u q) = case decomp u of
|
|
Right (Open _) -> runHttp (qApp q ())
|
|
Right Close -> runHttp (qApp q ())
|
|
Right (Post d) -> runHttp (qApp q d)
|
|
Right Get -> runHttp (qApp q "")
|
|
Left u' -> E u' (tsingleton (runHttp . qApp q ))
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- Free: Interpreter --
|
|
--------------------------------------------------------------------------------
|
|
data FHttpT x
|
|
= FOpen String x
|
|
| FClose x
|
|
| FPost String (String -> x)
|
|
| FGet (String -> x)
|
|
deriving Functor
|
|
|
|
type FHttp = Free.Free FHttpT
|
|
|
|
fopen' :: String -> FHttp ()
|
|
fopen' s = Free.liftF $ FOpen s ()
|
|
|
|
fclose' :: FHttp ()
|
|
fclose' = Free.liftF $ FClose ()
|
|
|
|
fpost' :: String -> FHttp String
|
|
fpost' s = Free.liftF $ FPost s id
|
|
|
|
fget' :: FHttp String
|
|
fget' = Free.liftF $ FGet id
|
|
|
|
runFHttp :: FHttp a -> Maybe a
|
|
runFHttp (Free.Pure x) = pure x
|
|
runFHttp (Free.Free (FOpen _ n)) = runFHttp n
|
|
runFHttp (Free.Free (FClose n)) = runFHttp n
|
|
runFHttp (Free.Free (FPost s n)) = pure s >>= runFHttp . n
|
|
runFHttp (Free.Free (FGet n)) = pure "" >>= runFHttp . n
|
|
|
|
--------------------------------------------------------------------------------
|
|
-- Benchmark Suite --
|
|
--------------------------------------------------------------------------------
|
|
prog :: Member Http r => Eff r ()
|
|
prog = open' "cats" >> get' >> post' "cats" >> close'
|
|
|
|
prog' :: FHttp ()
|
|
prog' = fopen' "cats" >> fget' >> fpost' "cats" >> fclose'
|
|
|
|
p :: Member Http r => Int -> Eff r ()
|
|
p count = open' "cats" >> replicateM_ count (get' >> post' "cats") >> close'
|
|
|
|
p' :: Int -> FHttp ()
|
|
p' count = fopen' "cats" >> replicateM_ count (fget' >> fpost' "cats") >> fclose'
|
|
|
|
main :: IO ()
|
|
main =
|
|
defaultMain [
|
|
bgroup "State" [
|
|
bench "freer.get" $ whnf oneGet 0
|
|
, bench "mtl.get" $ whnf oneGetMTL 0
|
|
, bench "ee.get" $ whnf oneGetEE 0
|
|
],
|
|
bgroup "Countdown Bench" [
|
|
bench "freer.State" $ whnf countDown 10000
|
|
, bench "freer.StateRW" $ whnf countDownRW 10000
|
|
, bench "mtl.State" $ whnf countDownMTL 10000
|
|
, bench "ee.State" $ whnf countDownEE 10000
|
|
],
|
|
bgroup "Countdown+Except Bench" [
|
|
bench "freer.ExcState" $ whnf countDownExc 10000
|
|
, bench "mtl.ExceptState" $ whnf countDownExcMTL 10000
|
|
, bench "ee.ExcState" $ whnf countDownExcEE 10000
|
|
],
|
|
bgroup "HTTP Simple DSL" [
|
|
bench "freer" $ whnf (run . runHttp) prog
|
|
, bench "free" $ whnf runFHttp prog'
|
|
|
|
, bench "freerN" $ whnf (run . runHttp . p) 1000
|
|
, bench "freeN" $ whnf (runFHttp . p') 1000
|
|
]
|
|
]
|