Haskell中hGetContents阻塞问题：读取多行数据

这个问题我之前也碰到过，核心问题出在hGetContents的行为特性上——它会一直读取输入流直到遇到EOF，但你的服务器发送完消息后并没有关闭连接，也没有传递明确的结束信号，导致客户端卡在hGetContents这里，永远执行不到后面的putStrLn "done!"。

下面给你几个不同场景下的解决方案：

方案1：发送完数据后关闭连接（适合一次性通信） #

如果你的服务器和客户端是一次性交互（发完数据就断开连接），那只需要在服务器发送完消息后关闭句柄，这样客户端的hGetContents会收到EOF，就能继续执行后续代码。

服务器端修改： #

-- 发送消息后关闭句柄
hPutStrLn handle message
hClose handle  -- 关键：发送完关闭连接

客户端不需要修改原来的代码，收到EOF后hGetContents会返回读取到的内容，然后继续执行putStrLn "done!"。

方案2：使用明确的结束标记（适合保持长连接） #

如果需要保持TCP连接继续后续通信，不能直接关闭连接，那可以约定一个特殊的结束标记（比如"__END__"），服务器发送完所有数据后发送这个标记，客户端循环读取每一行，直到遇到标记为止。

客户端修改： #

import System.IO

main = do
    -- 假设已经完成TCP连接，拿到了handle
    readUntilEnd handle
    putStrLn "done!"
  where
    readUntilEnd h = do
        line <- hGetLine h
        if line == "__END__"
            then return ()  -- 遇到结束标记，停止读取
            else do
                putStrLn line
                readUntilEnd h

服务器端对应修改： #

-- 发送所有数据后，发送结束标记
mapM_ (hPutStrLn handle) messages  -- messages是你要发送的多行数据列表
hPutStrLn handle "__END__"

方案3：先发送数据长度（适合结构化数据传输） #

如果要传输的数据是结构化的，可以先发送数据的长度（比如行数或者字节数），客户端先读取长度，再精准读取对应数量的内容，避免阻塞。

服务器端代码： #

import System.IO
import Text.Read (readMaybe)

sendMultiLine handle lines = do
    -- 先发送行数，转成字符串发送
    hPutStrLn handle (show $ length lines)
    -- 发送每一行数据
    mapM_ (hPutStrLn handle) lines

客户端代码： #

import System.IO
import Control.Monad (replicateM)

receiveMultiLine handle = do
    -- 读取行数并转换为整数
    countStr <- hGetLine handle
    case readMaybe countStr of
        Just count -> replicateM count (hGetLine handle)
        Nothing -> return []  -- 处理格式错误的情况

main = do
    -- 连接逻辑...
    responseLines <- receiveMultiLine handle
    mapM_ putStrLn responseLines
    putStrLn "done!"

这样客户端读取到指定行数后就会停止，不会一直等待EOF，自然就能执行后续的代码了。

内容的提问来源于stack exchange，提问作者Stefan