如何使用Haskell和海龟库从文件流式传输时删除行

Question

假设我想从一个文件流到另一个文件,但我想跳过输入文件的前n行.如果没有先使用'fold'折叠整个第一个文件,我该怎么做？

import Turtle
main = output "/tmp/b.txt" (f (input "/tmp/a.txt"))

应该做什么才能实现这一目标？

ps:我没有足够的声誉来创建'haskell-turtle'标签.

Answer 1

我认为这是正确的代码:

import Data.IORef
import Turtle

drop :: Int -> Shell a -> Shell a
drop n s = Shell (\(FoldM step begin done) -> do
    ref <- newIORef 0
    let step' x a = do
            n' <- readIORef ref
            writeIORef ref (n' + 1)
            if n' < n then return x else step x a
    foldIO s (FoldM step' begin done) )

...除了我可能称之为别的东西,drop以避免与前奏曲发生冲突.

它几乎完全相同Turtle.Prelude.limit(参见源代码进行比较).唯一的区别是我已经撤销了声明的then和else条款if.

如果这解决了你的问题,那么我会把它添加到Turtle.Prelude.