Haskell中如何与IO ()类型程序的标准输入交互？

问题背景 #

需要直接与已有的Haskell IO () 程序（比如示例中的echo）进行标准输入输出交互，而非通过createProcess调用外部shell命令（如nix run .#echo）——后者在Nix环境中存在失效问题，且完全没必要绕外部进程。

解决方案：模拟进程管道的线程内实现 #

通过创建管道句柄+fork线程运行目标程序的方式，直接在当前进程内模拟外部进程的IO交互，无需启动外部命令。以下是完整实现：

1. 核心函数实现 #

这个函数替代createProcess，直接接收IO ()动作并返回交互用的句柄：

import System.IO
import Control.Concurrent
import Control.Exception (finally)

-- 接收IO ()动作，返回供交互的输入句柄、输出句柄，以及用于同步的MVar
runIOWithPipes :: IO () -> IO (Handle, Handle, MVar ())
runIOWithPipes action = do
  -- 创建双向管道：我们写输入句柄，目标程序读；目标程序写输出句柄，我们读
  (progStdinRead, ourStdinWrite) <- createPipe
  (ourStdoutRead, progStdoutWrite) <- createPipe

  -- 设置行缓冲，避免因块缓冲导致的死锁
  mapM_ (`hSetBuffering` LineBuffering) [progStdinRead, ourStdinWrite, ourStdoutRead, progStdoutWrite]

  -- 创建同步用MVar，用于等待目标程序执行完成
  doneSignal <- newEmptyMVar

  -- 启动线程运行目标程序，替换其标准IO句柄为管道
  _ <- forkIO $ do
    -- 保存原标准IO句柄，后续恢复
    origStdin <- hDuplicate stdin
    origStdout <- hDuplicate stdout

    -- 将目标程序的stdin/stdout替换为管道句柄
    hDuplicateTo progStdinRead stdin
    hDuplicateTo progStdoutWrite stdout

    -- 运行目标动作，确保无论成功失败都恢复原句柄并清理管道
    action `finally` do
      hDuplicateTo origStdin stdin
      hDuplicateTo origStdout stdout
      hClose progStdinRead
      hClose progStdoutWrite
      putMVar doneSignal () -- 发出完成信号

  return (ourStdinWrite, ourStdoutRead, doneSignal)

2. 测试代码修改 #

用上面的函数替代原来的createProcess调用，修改testEcho：

echo :: IO ()
echo = getLine >>= putStrLn

testEcho :: IO ()
testEcho = do
  (stdinHdl, stdoutHdl, doneSignal) <- runIOWithPipes echo

  -- 向目标程序发送输入
  hPutStrLn stdinHdl "Hello, echo!"
  hClose stdinHdl -- 关闭输入，让目标程序知道没有更多数据

  -- 读取目标程序的输出
  response <- hGetContents stdoutHdl

  -- 等待目标程序执行完成，避免提前清理资源
  takeMVar doneSignal

  -- 清理句柄
  hClose stdoutHdl

  -- 验证结果
  if response == "Hello, echo!\n" then
    putStrLn "Success!"
  else error $ "Failure: 预期'Hello, echo!\n'，实际得到'" ++ response ++ "'"

关键说明 #

1. 缓冲设置：必须设置LineBuffering或NoBuffering，否则默认的块缓冲会导致程序因等待缓冲区填满而死锁。
2. 句柄恢复：用finally确保目标程序运行结束后恢复原标准IO句柄，避免影响当前进程的其他操作。
3. 同步机制：用MVar等待目标程序完成，比直接killThread更安全，能确保程序正常处理完输入输出。
4. Nix环境适配：完全避免了外部进程调用，不需要依赖nix run构建或查找包，解决了Nix环境下外部命令失效的问题。

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