Haskell 中的递归 IO

Question

在 Haskell 中，我可以轻松定义一个递归函数，它接受一个值并返回一个字符串：

Prelude> let countdown i = if (i > 0) then (show i) ++ countdown (i-1) else ""
Prelude> countdown 5
"54321"

我想使用相同类型的设计从文件句柄读取可用数据。在这种特殊情况下，我需要以与 hGetContents 相同的方式读取数据，但不让句柄处于“半关闭”状态，以便我可以与使用 createProcess 打开的进程的 stdin/stdout 句柄进行循环交互

main = do
    -- do work to get hin / hout handles for subprocess input / output

    hPutStrLn hin "whats up?"

    -- works
    -- putStrLn =<< hGetContents hout

    putStrLn =<< hGetLines hout

    where
        hGetLines h = do
            readable <- hIsReadable h
            if readable
                then hGetLine h ++ hGetLines h
                else []

：错误：

Couldn't match expected type `IO b0' with actual type `[a0]'
In the expression: hGetLine h : hGetLines h

我知道有各种库可用于完成我想要完成的任务，但因为我正在学习我的问题实际上是如何执行递归 IO。蒂亚！

Answer 1

朴素的解决方案，严格和O(n)堆栈

你仍然必须使用do-表示法，这会导致这样的结果：

import System.IO
import System.IO.Unsafe (unsafeInterleaveIO)

-- Too strict!
hGetLines :: Handle -> IO [String]
hGetLines h = do
    readable <- hIsReadable h
    if readable
        then do
            x  <- hGetLine h
            xs <- hGetLines h
            return (x:xs)
        else return []

但是请参阅我的评论一下，这个版本的hGetLines太严格了！

惰性流式版本

在拥有所有输入之前，它不会返回您的列表。你需要一些懒一点的东西。为此，我们有 unsafeInterleaveIO，

-- Just right
hGetLines' :: Handle -> IO [String]
hGetLines' h = unsafeInterleaveIO $ do
    readable <- hIsReadable h
    if readable
        then do
            x  <- hGetLine h
            xs <- hGetLines' h
            return (x:xs)
        else return []

现在您可以开始将结果逐行流式传输到您的消费者代码：

*Main> hGetLines' stdin
123
["123"345
,"345"321
,"321"^D^CInterrupted.

Answer 2

如果您检查 ghci 中 (++) 的类型，您会得到：

Prelude> :t (++)
(++) :: [a] -> [a] -> [a]

这意味着您只能将列表附加在一起（请记住 String 是 [Char] 的别名，所以它是一个列表）。 hGetLine的类型是Handle -> IO String，而hGetLines的类型应该是IO [String]，所以你不能追加这些值。 (:) 的类型为 a -> [a] 在这里效果更好。

if readable
  then do
    -- First you need to extract them
    a <- hGetLine h
    b <- hGetLines h
    -- a and b have type String
    -- Now we can cons them and then go back into IO
    return (a : b)

这同样适用于 else []。您需要返回 IO [String] 类型的值。将其更改为 return []

另外，您将无法仅 putStrLn 自 (=<< hGetLines h) 以来的行为您提供 [String] 而不是 String ，这是 putStrLn 所期望的。
这可以通过多种方式解决。一种是首先连接值。 putStrln 。连接=<< (hGetLines h)。或者您可以使用 mapM_ putStrLn (hGetLines h) 打印每一行。

Answer 3

这就是说，部分代码期望 hGetLines h 具有类型 IO a，而另一部分则发现它具有类型 [a]。您可能希望 if 语句是：

if readable
    then return hGetLine h ++ hGetLines h
    else return []