git.iko.soy/typeable/aeson
on 0.3.2.3 help sign in
typeable/aeson/benchmarks/AesonEncode.hs
44 lines 1.4 kB history raw
1{-# LANGUAGE BangPatterns, OverloadedStrings #-}
3import Control.Exception
4import Control.Monad
5import Data.Aeson
6import Data.Aeson.Encode
7import Data.Aeson.Parser
8import Data.Attoparsec
9import Data.Time.Clock
10import System.Environment (getArgs)
11import System.IO
12import qualified Data.ByteString as B
13import qualified Data.ByteString.Lazy as L
14import qualified Data.ByteString.Lazy.Internal as L
15import Control.DeepSeq
17instance NFData L.ByteString where
18 rnf = go
19 where go (L.Chunk _ cs) = go cs
20 go L.Empty = ()
21 {-# INLINE rnf #-}
23main = do
24 (cnt:args) <- getArgs
25 let count = read cnt :: Int
26 forM_ args $ \arg -> bracket (openFile arg ReadMode) hClose $ \h -> do
27 putStrLn $ arg ++ ":"
28 let refill = B.hGet h 16384
29 result <- parseWith refill json =<< refill
30 r <- case result of
31 Done _ r -> return r
32 _ -> fail $ "failed to read " ++ show arg
33 start <- getCurrentTime
34 let loop !n r
35 | n >= count = return ()
36 | otherwise = {-# SCC "loop" #-} do
37 case result of
38 Done _ r -> rnf (encode r) `seq` loop (n+1) r
39 _ -> error $ "failed to read " ++ show arg
40 loop 0 r
41 delta <- flip diffUTCTime start `fmap` getCurrentTime
42 let rate = fromIntegral count / (fromRational . toRational) delta :: Double
43 putStrLn $ " " ++ show delta
44 putStrLn $ " " ++ show (round rate) ++ " per second"