diff options
Diffstat (limited to 'lib/ByteStringUtil.hs')
-rw-r--r-- | lib/ByteStringUtil.hs | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/lib/ByteStringUtil.hs b/lib/ByteStringUtil.hs new file mode 100644 index 0000000..c6b509d --- /dev/null +++ b/lib/ByteStringUtil.hs | |||
@@ -0,0 +1,23 @@ | |||
1 | module ByteStringUtil where | ||
2 | |||
3 | import Data.ByteString.Lazy.Internal | ||
4 | import Data.ByteString.Lazy | ||
5 | import qualified Data.ByteString as S | ||
6 | import System.IO | ||
7 | import System.IO.Unsafe | ||
8 | |||
9 | oneMeg :: Int | ||
10 | oneMeg = 1048576 | ||
11 | |||
12 | hGetContentsN :: Int -> Handle -> IO ByteString | ||
13 | hGetContentsN kk h = lazyRead | ||
14 | where | ||
15 | k = kk - chunkOverhead | ||
16 | lazyRead = unsafeInterleaveIO loop | ||
17 | |||
18 | loop = do | ||
19 | c <- S.hGetSome h k -- only blocks if there is no data available | ||
20 | if S.null c | ||
21 | then hClose h >> return Empty | ||
22 | else do cs <- lazyRead | ||
23 | return (Chunk c cs) | ||