1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
-- |
-- Copyright: (c) 2022 Andrew Lelechenko
-- Licence: BSD3
-- Maintainer: Andrew Lelechenko <andrew.lelechenko@gmail.com>
module BenchText (benchText) where
import qualified Data.ByteString as B
import qualified Data.ByteString.Builder as B
import qualified Data.Text as T
import qualified Data.Text.Encoding as T
import Data.Text.Builder.Linear.Buffer
import Data.Text.Lazy (toStrict)
import Data.Text.Lazy.Builder (toLazyText, fromText)
import Test.Tasty.Bench
#ifdef MIN_VERSION_text_builder
import qualified Text.Builder
#endif
#ifdef MIN_VERSION_bytestring_strict_builder
import qualified ByteString.StrictBuilder
#endif
txt ∷ T.Text
txt = T.pack "Haskell + Linear Types = ♡"
benchLazyBuilder ∷ Int → T.Text
benchLazyBuilder = toStrict . toLazyText . go mempty
where
txtB = fromText txt
go !acc 0 = acc
go !acc n = go (txtB <> (acc <> txtB)) (n - 1)
benchLazyBuilderBS ∷ Int → B.ByteString
benchLazyBuilderBS = B.toStrict . B.toLazyByteString . go mempty
where
txtB = B.byteString $ T.encodeUtf8 txt
go !acc 0 = acc
go !acc n = go (txtB <> (acc <> txtB)) (n - 1)
#ifdef MIN_VERSION_text_builder
benchStrictBuilder ∷ Int → T.Text
benchStrictBuilder = Text.Builder.run . go mempty
where
txtB = Text.Builder.text txt
go !acc 0 = acc
go !acc n = go (txtB <> (acc <> txtB)) (n - 1)
#endif
#ifdef MIN_VERSION_bytestring_strict_builder
benchStrictBuilderBS ∷ Int → B.ByteString
benchStrictBuilderBS = ByteString.StrictBuilder.builderBytes . go mempty
where
txtB = ByteString.StrictBuilder.bytes $ T.encodeUtf8 txt
go !acc 0 = acc
go !acc n = go (txtB <> (acc <> txtB)) (n - 1)
#endif
benchLinearBuilder ∷ Int → T.Text
benchLinearBuilder m = runBuffer (\b → go b m)
where
go ∷ Buffer ⊸ Int → Buffer
go !acc 0 = acc
go !acc n = go (txt <| (acc |> txt)) (n - 1)
benchText ∷ Benchmark
benchText = bgroup "Text" $ map mkGroup [1e0, 1e1, 1e2, 1e3, 1e4, 1e5, 1e6]
mkGroup :: Int → Benchmark
mkGroup n = bgroup (show n)
[ bench "Data.Text.Lazy.Builder" $ nf benchLazyBuilder n
, bench "Data.ByteString.Builder" $ nf benchLazyBuilderBS n
#ifdef MIN_VERSION_text_builder
, bench "Text.Builder" $ nf benchStrictBuilder n
#endif
#ifdef MIN_VERSION_bytestring_strict_builder
, bench "ByteString.StrictBuilder" $ nf benchStrictBuilderBS n
#endif
, bench "Data.Text.Builder.Linear" $ nf benchLinearBuilder n
]
|
