git.iko.soy/typeable/aeson
on 0.8.1.1 help sign in
typeable/aeson/benchmarks/AesonEncode.hs
37 lines 1.2 kB history raw
1{-# LANGUAGE BangPatterns, OverloadedStrings #-}
3import Control.Exception
4import Control.Monad
5import Data.Aeson
6import Data.Attoparsec (IResult(..), parseWith)
7import Data.Char (isDigit)
8import Data.Time.Clock
9import System.Environment (getArgs)
10import System.IO
11import qualified Data.ByteString as B
12import Control.DeepSeq
14main :: IO ()
15main = do
16 args0 <- getArgs
17 let (cnt,args) = case args0 of
18 (i:c:a) | all isDigit i && all isDigit c -> (c,a)
19 (c:a) -> (c,a)
20 let count = read cnt :: Int
21 forM_ args $ \arg -> bracket (openFile arg ReadMode) hClose $ \h -> do
22 putStrLn $ arg ++ ":"
23 let refill = B.hGet h 16384
24 result0 <- parseWith refill json =<< refill
25 r0 <- case result0 of
26 Done _ r -> return r
27 _ -> fail $ "failed to read " ++ show arg
28 start <- getCurrentTime
29 let loop !n r
30 | n >= count = return ()
31 | otherwise = {-# SCC "loop" #-} do
32 rnf (encode r) `seq` loop (n+1) r
33 loop 0 r0
34 delta <- flip diffUTCTime start `fmap` getCurrentTime
35 let rate = fromIntegral count / realToFrac delta :: Double
36 putStrLn $ " " ++ cnt ++ " good, " ++ show delta
37 putStrLn $ " " ++ show (round rate :: Int) ++ " per second"