haskell - ByteString 的 chunksOf 模拟？

Question

我需要将字节串拆分为第一个由 100 个字符组成的字节串列表。对于列表，我可以使用 chunksOf但不适用于 ByteString .

有没有一些正确的方法来做到这一点？

Answer 1

两者 ByteString和 Lazy.Bytestring有 splitAt functions您可以将其用于 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"