fix benchmark script

glguy · Feb 22, 2024 · 754d083 · 754d083
1 parent 06593ee
commit 754d083
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/benchmarker/benchmarker.hs b/benchmarker/benchmarker.hs
@@ -1,6 +1,7 @@
 
 
 import Control.Exception (evaluate)
+import qualified Data.Text.IO
 import Data.Time (diffUTCTime, getCurrentTime)
 import System.Environment (getArgs)
 import Toml (parse)
@@ -11,8 +12,7 @@ main =
     filename <- case args of
       [filename] -> pure filename
       _ -> fail "Usage: benchmarker <file.toml>"
-    txt <- readFile filename
-    evaluate (length txt) -- readFile uses lazy IO, force it to load
+    txt <- Data.Text.IO.readFile filename
     start <- getCurrentTime
     evaluate (parse txt)
     stop <- getCurrentTime

diff --git a/toml-parser.cabal b/toml-parser.cabal
@@ -137,5 +137,5 @@ executable toml-benchmarker
     buildable: False
     main-is: benchmarker.hs
     default-language: Haskell2010
-    build-depends: base, toml-parser, time
+    build-depends: base, toml-parser, time, text
     hs-source-dirs: benchmarker