streaming-commons-0.1.0.0: bench/decode-memory-usage.hs

{-# LANGUAGE OverloadedStrings #-}
import Data.ByteString (ByteString)
import Data.Text.StreamDecoding
import System.Environment (getArgs)

input :: [ByteString]
input = replicate 1000000 "Hello World!\n"

main :: IO ()
main = do
    args <- getArgs
    let dec =
            case args of
                ["16le"] -> streamUtf16LE
                ["16be"] -> streamUtf16BE
                ["32le"] -> streamUtf32LE
                ["32be"] -> streamUtf32BE
                ["8pure"] -> streamUtf8Pure
                _ -> streamUtf8

    loop dec input

loop :: (ByteString -> DecodeResult) -> [ByteString] -> IO ()
loop dec [] =
    case dec "" of
        DecodeResultSuccess _ _ -> return ()
        DecodeResultFailure _ _ -> error "failure1"
loop dec (bs:bss) =
    case dec bs of
        DecodeResultSuccess _ dec' -> loop dec' bss
        DecodeResultFailure _ _ -> error "failure2"