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

Question

假设我想从一个文件流式传输到另一个文件，但我想跳过输入文件的前 n 行。如何在不首先使用“fold”折叠整个第一个文件的情况下执行此操作？

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

“f”应该是什么来完成这个任务？

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 以外的名称，以避免与 Prelude 发生冲突。

它与 Turtle.Prelude.limit 几乎相同(请参阅 source code 进行比较)。唯一的区别是我颠倒了 if 语句的 then 和 else 子句。

如果这解决了您的问题，那么我会将其添加到 Turtle.Prelude 中。