chunksOf analog для ByteString?

Question

Мне нужно разделить bytestring в список байтов, погружающихся сначала на 100 символов. Для списков я могу использовать chunksOf, но не для ByteString.

Есть ли правильный способ сделать это?

Answer 1

И ByteString, и Lazy.Bytestring имеют splitAtfunctions которые вы можете использовать для unfoldr списка.

import Data.List (unfoldr) 
import Data.Int (Int64) 

import qualified Data.ByteString as BS 
import qualified Data.ByteString.Lazy as Lazy 

justWhen :: (a -> Bool) -> (a -> b) -> (a -> Maybe b) 
justWhen f g a = if f a then Just (g a) else Nothing 

nothingWhen :: (a -> Bool) -> (a -> b) -> (a -> Maybe b) 
nothingWhen f = justWhen (not . f) 

chunksOf :: Int -> BS.ByteString -> [BS.ByteString] 
chunksOf x = unfoldr (nothingWhen BS.null (BS.splitAt x)) 

chunksOf' :: Int64 -> Lazy.ByteString -> [Lazy.ByteString] 
chunksOf' x = unfoldr (nothingWhen Lazy.null (Lazy.splitAt x)) 

Строительные байтовых строк для примера проще с OverloadedStrings

{-# LANGUAGE OverloadedStrings #-} 

main = do 
    print $ chunksOf 3 "" 
    print $ chunksOf 3 "Hello World!" 
    print $ chunksOf' 3 "" 
    print $ chunksOf' 3 "Hello, World"