Source

text / tests / benchmarks / FoldLines.hs

The default branch has multiple heads

{-# LANGUAGE BangPatterns #-}

import System.Environment
import System.IO
import qualified Data.Text as T
import qualified Data.Text.IO as T
import qualified Data.ByteString as S

-- Text
foldLinesT :: (a -> T.Text -> a) -> a -> Handle -> IO a
foldLinesT f z0 h = go z0
  where
    go !z = do
        eof <- hIsEOF h
        if eof
            then return z
            else do
                l <- T.hGetLine h
                let z' = f z l in go z'
{-# INLINE foldLinesT #-}

testT :: Handle -> IO Int
testT = foldLinesT (\n _ -> n + 1) 0

--ByteString
foldLinesB :: (a -> S.ByteString -> a) -> a -> Handle -> IO a
foldLinesB f z0 h = go z0
  where
    go !z = do
        eof <- hIsEOF h
        if eof
            then return z
            else do
                l <- S.hGetLine h
                let z' = f z l in go z'
{-# INLINE foldLinesB #-}

testB :: Handle -> IO Int
testB = foldLinesB (\n _ -> n + 1) 0

main = do
  (name : file : _) <- getArgs
  h <- openFile file ReadMode
  hSetBuffering h (BlockBuffering (Just 16384))
  case name of
    "bs" -> testB h
    "text" -> testT h
Tip: Filter by directory path e.g. /media app.js to search for public/media/app.js.
Tip: Use camelCasing e.g. ProjME to search for ProjectModifiedEvent.java.
Tip: Filter by extension type e.g. /repo .js to search for all .js files in the /repo directory.
Tip: Separate your search with spaces e.g. /ssh pom.xml to search for src/ssh/pom.xml.
Tip: Use ↑ and ↓ arrow keys to navigate and return to view the file.
Tip: You can also navigate files with Ctrl+j (next) and Ctrl+k (previous) and view the file with Ctrl+o.
Tip: You can also navigate files with Alt+j (next) and Alt+k (previous) and view the file with Alt+o.