Аналог chunksOf для ByteString?

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

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


person Filip van Hoft    schedule 28.09.2015    source источник


Ответы (1)


И ByteString, и Lazy.Bytestring имеют splitAt функции, которые можно использовать для 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"
person Cirdec    schedule 28.09.2015
comment
Это именно то, что я искал. Спасибо за подробное описание. - person Filip van Hoft; 28.09.2015