git.iko.soy/text
on 2b13ada help sign in
text/benchmarks/haskell/Benchmarks/Programs/Sort.hs
71 lines 2.4 kB history raw
1-- | This benchmark sorts the lines of a file, like the @sort@ unix utility.
2--
3-- Tested in this benchmark:
4--
5-- * Reading the file
6--
7-- * Splitting into lines
8--
9-- * Sorting the lines
10--
11-- * Joining the lines
12--
13-- * Writing back to a handle
14--
15{-# LANGUAGE OverloadedStrings #-}
16module Benchmarks.Programs.Sort
17 ( benchmark
18 ) where
20import Criterion (Benchmark, bgroup, bench, whnfIO)
21import Data.Monoid (mconcat)
22import System.IO (Handle, hPutStr)
23import qualified Data.ByteString as B
24import qualified Data.ByteString.Char8 as BC
25import qualified Data.ByteString.Lazy as BL
26import qualified Data.ByteString.Lazy.Char8 as BLC
27import qualified Data.List as L
28import qualified Data.Text as T
29import qualified Data.Text.Encoding as T
30import qualified Data.Text.IO as T
31import qualified Data.Text.Lazy as TL
32import qualified Data.Text.Lazy.Builder as TLB
33import qualified Data.Text.Lazy.Encoding as TL
34import qualified Data.Text.Lazy.IO as TL
36benchmark :: FilePath -> Handle -> IO Benchmark
37benchmark i o = return $ bgroup "Sort"
38 [ bench "String" $ whnfIO $ readFile i >>= hPutStr o . string
39 , bench "ByteString" $ whnfIO $ B.readFile i >>= B.hPutStr o . byteString
40 , bench "LazyByteString" $ whnfIO $
41 BL.readFile i >>= BL.hPutStr o . lazyByteString
42 , bench "Text" $ whnfIO $ T.readFile i >>= T.hPutStr o . text
43 , bench "LazyText" $ whnfIO $ TL.readFile i >>= TL.hPutStr o . lazyText
44 , bench "TextByteString" $ whnfIO $ B.readFile i >>=
45 B.hPutStr o . T.encodeUtf8 . text . T.decodeUtf8
46 , bench "LazyTextByteString" $ whnfIO $ BL.readFile i >>=
47 BL.hPutStr o . TL.encodeUtf8 . lazyText . TL.decodeUtf8
48 , bench "TextBuilder" $ whnfIO $ B.readFile i >>=
49 BL.hPutStr o . TL.encodeUtf8 . textBuilder . T.decodeUtf8
50 ]
52string :: String -> String
53string = unlines . L.sort . lines
55byteString :: B.ByteString -> B.ByteString
56byteString = BC.unlines . L.sort . BC.lines
58lazyByteString :: BL.ByteString -> BL.ByteString
59lazyByteString = BLC.unlines . L.sort . BLC.lines
61text :: T.Text -> T.Text
62text = T.unlines . L.sort . T.lines
64lazyText :: TL.Text -> TL.Text
65lazyText = TL.unlines . L.sort . TL.lines
67-- | Text variant using a builder monoid for the final concatenation
68--
69textBuilder :: T.Text -> TL.Text
70textBuilder = TLB.toLazyText . mconcat . L.intersperse (TLB.singleton '\n') .
71 map TLB.fromText . L.sort . T.lines