Bryan O'Sullivan avatar Bryan O'Sullivan committed 3b88e63

Make EncodeFile a bit more realistic - read just once

Comments (0)

Files changed (1)

benchmarks/EncodeFile.hs

 import System.IO
 import qualified Data.ByteString as B
 import qualified Data.ByteString.Lazy as L
+import qualified Data.ByteString.Lazy.Internal as L
+import Control.DeepSeq
+
+instance NFData L.ByteString where
+    rnf = go
+      where go (L.Chunk _ cs) = go cs
+            go L.Empty        = ()
+    {-# INLINE rnf #-}
 
 main = do
   (cnt:args) <- getArgs
   let count = read cnt :: Int
   forM_ args $ \arg -> bracket (openFile arg ReadMode) hClose $ \h -> do
     putStrLn $ arg ++ ":"
+    let refill = B.hGet h 16384
+    result <- parseWith refill json =<< refill
+    r <- case result of
+           Done _ r -> return r
+           _        -> fail $ "failed to read " ++ show arg
     start <- getCurrentTime
-    let loop !n
+    let loop !n r
             | n >= count = return ()
             | otherwise = {-# SCC "loop" #-} do
-          hSeek h AbsoluteSeek 0
-          let refill = B.hGet h 16384
-          result <- parseWith refill json =<< refill
           case result of
-            Done _ r -> L.length (encode r) `seq` loop (n+1)
+            Done _ r -> rnf (encode r) `seq` loop (n+1) r
             _        -> error $ "failed to read " ++ show arg
-    loop 0
+    loop 0 r
     end <- getCurrentTime
     putStrLn $ "  " ++ show (diffUTCTime end start)
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.