git.iko.soy/typeable/aeson
on 1.1.1.0 help sign in
typeable/aeson/benchmarks/AesonParse.hs
40 lines 1.3 kB history raw
1{-# LANGUAGE BangPatterns #-}
2{-# LANGUAGE OverloadedStrings #-}
3{-# LANGUAGE PackageImports #-}
5module Main (main) where
7import Prelude ()
8import Prelude.Compat
10import "aeson-benchmarks" Data.Aeson
11import Control.Exception
12import Control.Monad
13import Data.Attoparsec.ByteString (IResult(..), parseWith)
14import Data.Time.Clock
15import System.Environment (getArgs)
16import System.IO
17import qualified Data.ByteString as B
19main :: IO ()
20main = do
21 (bs:cnt:args) <- getArgs
22 let count = read cnt :: Int
23 blkSize = read bs
24 forM_ args $ \arg -> bracket (openFile arg ReadMode) hClose $ \h -> do
25 putStrLn $ arg ++ ":"
26 start <- getCurrentTime
27 let loop !good !bad
28 | good+bad >= count = return (good, bad)
29 | otherwise = do
30 hSeek h AbsoluteSeek 0
31 let refill = B.hGet h blkSize
32 result <- parseWith refill json =<< refill
33 case result of
34 Done _ _ -> loop (good+1) bad
35 _ -> loop good (bad+1)
36 (good, _) <- loop 0 0
37 delta <- flip diffUTCTime start `fmap` getCurrentTime
38 putStrLn $ " " ++ show good ++ " good, " ++ show delta
39 let rate = fromIntegral count / realToFrac delta :: Double
40 putStrLn $ " " ++ show (round rate :: Int) ++ " per second"