benchmarks/AesonEncode.hs · typeable/aeson

git.iko.soy/typeable/aeson

on wip/yuras/scanner help sign in

43 lines 1.3 kB history raw

1{-# LANGUAGE BangPatterns #-}

2{-# LANGUAGE NoImplicitPrelude #-}

3{-# LANGUAGE OverloadedStrings #-}

5module Main (main) where

7import Prelude.Compat

9import Control.DeepSeq

10import Control.Monad (forM_)

11import Data.Aeson

12import Scanner (Result(..), scanWith)

13import Data.Char (isDigit)

14import Data.Time.Clock

15import System.Environment (getArgs)

16import System.IO

17import qualified Data.ByteString as B

19main :: IO ()

20main = do

21 args0 <- getArgs

22 let (cnt,args) = case args0 of

23 (i:c:a) | all isDigit i && all isDigit c -> (c,a)

24 (c:a) -> (c,a)

25 [] -> error "Unexpected empty list"

26 let count = read cnt :: Int

27 forM_ args $ \arg -> withFile arg ReadMode $ \h -> do

28 putStrLn $ arg ++ ":"

29 let refill = B.hGet h 16384

30 result0 <- scanWith refill json =<< refill

31 r0 <- case result0 of

32 Done _ r -> return r

33 _ -> fail $ "failed to read " ++ show arg

34 start <- getCurrentTime

35 let loop !n r

36 | n >= count = return ()

37 | otherwise = {-# SCC "loop" #-}

38 rnf (encode r) `seq` loop (n+1) r

39 loop 0 r0

40 delta <- flip diffUTCTime start `fmap` getCurrentTime

41 let rate = fromIntegral count / realToFrac delta :: Double

42 putStrLn $ " " ++ cnt ++ " good, " ++ show delta

43 putStrLn $ " " ++ show (round rate :: Int) ++ " per second"