haskell 中的惰性 IO：如何返回由某些阻塞 IO 生成的惰性列表？

Question

getText = do
    c <- getChar
    s <- getText
    return (c : s)

main = do
    s <- getText
    putStr s

我希望看到的是每次按“Enter”后都会回显输入行。 但是没有任何回应......（我知道这是一个无限循环） 在执行完上面的所有“IO”之前，它似乎不会“return”。 ...

但是，下面的代码：

main = do
    s <- getContents
    putStr s

输入后立即显示该行。

给定函数getChar，我可以写一个行为类似于getContents的getText吗？

Answer 1

你谈论 Enter 但你没有在你的代码中检查它。

试试这个：

getText = do
    c <- getChar
    if (c == '\n')
       then return [c]
       else do
              s <- getText
              return (c : s)

main = do
    s <- getText
    putStr s

Answer 2

这是为 ... unsafeInterleaveIO 做的工作 - 使惰性 IO 成为可能的特殊操作。它使您可以将 IO 操作转换为绑定到 thunk 的操作。然后可以将其存储在一个结构中，并且该操作只会在需要其结果时进行评估。

getText = unsafeInterleaveIO $ do
    c <- getChar
    s <- getText
    return (c : s)

现在您的getText 立即返回，每个getChar 只暂停计算。如果您需要结果，则运行它。

Answer 3

这可以通过unsafeInterleaveIO 函数来完成 System.IO.Unsafe。然后你的getText 函数就变成了

getText = do
    c <- getChar
    s <- unsafeInterleaveIO $ getText
    return (c : s)

稍微抽象一下，我们可以得到一个函数来概括这种行为

lazyDoIO :: IO a -> IO [a]
lazyDoIO act = unsafeInterleaveIO $ do
    now <- act
    rest <- lazyDoIO act
    return (now : rest)

getText = lazyDoIO getChar

然而，大多数 Haskeller 会在这样做时畏缩。如果你想做 IO生成数据的增量流处理，会更安全 使用像 Pipes 或 Conduits 这样的库。