sockets - hGetContents 与 hGetLine

Question

我在 Haskell 中有一个程序从套接字获取所有输入并打印它。

     main = withSocketsDo $ do
            sock <- listenOn $ PortNumber 5002
            netLoop sock

        netLoop sock = do
           (h,_,_) <- accept sock
           hSetBuffering h NoBuffering
           forkIO $ workLoop h
           netLoop sock


        workLoop :: Handle -> IO ()
        workLoop h = do
        str <- hGetContents h   
        putStr str
        --do any work

但问题是这个解决方案正在关闭一个套接字，但我想将计算结果写出到同一个套接字。
但是如果我尝试使用 hGetLine相反 hGetContents我遇到了一些奇怪的行为。我的程序在按 Ctrl-C 之前什么都不显示，然后我看到发送的网络数据的第一行。我建议这种行为与 lasy 执行有关，但是为什么 hGetContents 可以按预期工作而 hGetLine 不行？

Answer 1

您需要使用 LineBuffering如果您想使用 hGetLine 逐行阅读.我得到它的工作

import Network
import System.IO
import Control.Concurrent

main :: IO ()
main = withSocketsDo $ do
    sock <- listenOn $ PortNumber 5002
    netLoop sock

netLoop :: Socket -> IO ()
netLoop sock = do
    putStrLn "Accepting socket"
    (h,_,_) <- accept sock
    putStrLn "Accepted socket"
    hSetBuffering h LineBuffering
    putStrLn "Starting workLoop"
    forkIO $ workLoop h
    netLoop sock


workLoop :: Handle -> IO ()
workLoop h = do
    putStrLn "workLoop started"
    str <- hGetLine h
    putStrLn $ "Read text: " ++ str
    -- do any work

并使用 python 脚本对其进行了测试

import socket

s = socket()
s.connect(('127.0.0.1', 5002))
s.send('testing\n')
s.close()

我得到了输出

Accepting socket
Accepted socket
Starting workLoop
workLoop started
Accepting socket
Read text: testing

如果我将其更改为 NoBuffering，我会得到相同的行为和 hGetContents