haskell - 没有 lambda 表达式如何编写这些？

Question

我正在转向 this code 的第一个版本使用StateT L8.ByteString Maybe a 。到目前为止我已经把大部分功能变成了这个

matchHeader :: L8.ByteString -> StateT L8.ByteString Maybe ()
matchHeader prefix = StateT $ \str ->
      if prefix `L8.isPrefixOf` str
         then Just ((), L8.drop (L8.length prefix) str)
         else Nothing

getNat :: Num a => StateT L8.ByteString Maybe a
getNat = StateT $ \str ->
  case L8.readInt str of
    Nothing -> Nothing
    Just (num, rest)
      | num <= 0        -> Nothing
      | otherwise       -> Just (fromIntegral num, rest)

getBytes :: Integer -> StateT L8.ByteString Maybe L8.ByteString
getBytes n = StateT $ \str ->
  let
    count = fromIntegral n
    both@(prefix, _) = L8.splitAt count str
   in if L8.length prefix < count
        then Nothing
        else Just both

但是如何在不使用 lambda 表达式的情况下编写这些内容呢？我尝试过一些变化。到目前为止还没有运气。

Answer 1

您可以更多地使用 do 表示法和状态/monad 操作。例如:

getNat :: Num a => StateT L8.ByteString Maybe a
getNat = do
  num <- StateT L8.readInt
  guard (num > 0)
  return (fromIntegral num)

编辑:根据请求，尝试getBytes:

getBytes :: Integer -> StateT L8.ByteString Maybe L8.ByteString
getBytes n = do
  let count = fromIntegral n
  prefix <- state $ L8.splitAt count
  guard (L8.length prefix >= count)
  return prefix

在 :browse-ing Control.Monad 时，我还发现了(有点新？)函数 mfilter，它可以进一步缩短这个时间，但需要一些无点性或另一个 lambda:

getBytes n = do
  let count = fromIntegral n
  mfilter ((>= count) . L8.length) . state $ L8.splitAt count

想想看，它也可以与 getNat 一起使用:

getNat = fromIntegral <$> mfilter (> 0) (StateT L8.readInt)