git.iko.soy/typeable/aeson
on 1.2.2.0 help sign in
typeable/aeson/benchmarks/AesonParse.hs
39 lines 1.2 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.Monad
12import Data.Attoparsec.ByteString (IResult(..), parseWith)
13import Data.Time.Clock
14import System.Environment (getArgs)
15import System.IO
16import qualified Data.ByteString as B
18main :: IO ()
19main = do
20 (bs:cnt:args) <- getArgs
21 let count = read cnt :: Int
22 blkSize = read bs
23 forM_ args $ \arg -> withFile arg ReadMode $ \h -> do
24 putStrLn $ arg ++ ":"
25 start <- getCurrentTime
26 let loop !good !bad
27 | good+bad >= count = return (good, bad)
28 | otherwise = do
29 hSeek h AbsoluteSeek 0
30 let refill = B.hGet h blkSize
31 result <- parseWith refill json =<< refill
32 case result of
33 Done _ _ -> loop (good+1) bad
34 _ -> loop good (bad+1)
35 (good, _) <- loop 0 0
36 delta <- flip diffUTCTime start `fmap` getCurrentTime
37 putStrLn $ " " ++ show good ++ " good, " ++ show delta
38 let rate = fromIntegral count / realToFrac delta :: Double
39 putStrLn $ " " ++ show (round rate :: Int) ++ " per second"