Merge pull request #833 from phadej/escape-benchmark

Add escape function benchmarks into bench suite

Author: phadej

benchmarks/aeson-benchmarks.cabal

@@ -141,21 +141,6 @@ executable aeson-benchmark-auto-compare
     , deepseq
     , template-haskell
 
-executable aeson-benchmark-escape
-  default-language: Haskell2010
-  main-is:          Escape.hs
-  hs-source-dirs:   bench examples/src
-  ghc-options:      -Wall -O2 -rtsopts
-  build-depends:
-      aeson-benchmarks
-    , base
-    , base-compat-batteries
-    , bytestring
-    , criterion              >=1.0
-    , deepseq
-    , ghc-prim
-    , text
-
 executable aeson-benchmark-compare
   default-language: Haskell2010
   main-is:          Compare.hs

benchmarks/bench/Escape.hs

@@ -1,25 +1,32 @@
 {-# LANGUAGE NoImplicitPrelude   #-}
+{-# LANGUAGE OverloadedStrings   #-}
 {-# LANGUAGE ScopedTypeVariables #-}
 
 module Main (main) where
 
 import           Data.Aeson
 import           Prelude.Compat
 
-import           Control.DeepSeq      (NFData)
-import           Criterion.Main       (Benchmark, bench, bgroup, defaultMain,
-                                       env, nf)
-import           Data.Maybe           (fromMaybe)
-import           Data.Proxy           (Proxy (..))
-import           Data.Vector          (Vector)
-import           System.Environment   (lookupEnv)
-import           System.FilePath      ((</>))
+import           Control.DeepSeq                (NFData)
+import           Criterion.Main                 (Benchmark, bench, bgroup,
+                                                 defaultMain, env, nf, whnf)
+import           Data.Maybe                     (fromMaybe)
+import           Data.Proxy                     (Proxy (..))
+import           Data.Vector                    (Vector)
+import           System.Environment             (lookupEnv)
+import           System.FilePath                ((</>))
 
-import qualified Data.ByteString      as BS
-import qualified Data.ByteString.Lazy as LBS
+import qualified Data.Aeson.Encoding.Builder    as Aeson.EB
+import qualified Data.Aeson.Parser.UnescapeFFI  as FFI
+import qualified Data.Aeson.Parser.UnescapePure as Pure
+import qualified Data.ByteString                as BS
+import qualified Data.ByteString.Char8          as BS8
+import qualified Data.ByteString.Lazy           as LBS
+import qualified Data.Text                      as T
+import qualified Data.ByteString.Builder as B
 
 import qualified Twitter
-import qualified Twitter.Manual       ()
+import qualified Twitter.Manual                 ()
 
 import qualified GitHub
 
@@ -44,14 +51,39 @@ decodeBench name fp _ = bgroup name
     decS :: BS.ByteString -> Maybe a
     decS = decodeStrict
 
+-------------------------------------------------------------------------------
+-- Escape bench
+-------------------------------------------------------------------------------
+
+escapeBench :: Benchmark
+escapeBench = bgroup "Escape"
+    [ example "ascii" $ BS8.pack $ take 500 $ cycle ['a'..'z']
+    , example "cyrillic" $ LBS.toStrict $ B.toLazyByteString $ Aeson.EB.unquoted $ T.unwords
+      [ "Стандарт состоит из двух основных частей: универсального набора "
+      , "символов (англ. Universal character set, UCS) и семейства кодировок"
+      , "(англ. Unicode transformation format, UTF). Универсальный набор"
+      , "символов перечисляет допустимые по стандарту Юникод символы и"
+      , "присваивает каждому символу код в виде неотрицательного целого"
+      , "числа, записываемого обычно в шестнадцатеричной форме с префиксом"
+      , "U+, например, U+040F. Семейство кодировок определяет способы"
+      , "преобразования кодов символов для передачи в потоке или в файле."
+      ]
+    ]
+  where
+    example :: String -> BS.ByteString -> Benchmark
+    example name input = bgroup name
+      [ bench "ffi"  $ whnf FFI.unescapeText input
+      , bench "pure" $ whnf Pure.unescapeText input
+      ]
+
 -------------------------------------------------------------------------------
 -- Helpers
 -------------------------------------------------------------------------------
 
 readS :: FilePath -> IO BS.ByteString
 readS fp = do
     dataDir <- lookupEnv "AESON_BENCH_DATADIR"
-    BS.readFile $ maybe id (</>) dataDir fp
+    BS.readFile $ fromMaybe "json-data" dataDir </> fp
 
 readL :: FilePath -> IO LBS.ByteString
 readL fp = do
@@ -71,4 +103,5 @@ main = defaultMain
       , decodeBench "github-issues" "github-issues.json" (Proxy :: Proxy (Vector GitHub.Issue))
       ]
     ]
+  , escapeBench
   ]