Haskell进程管理与流重定向：gzip/gunzip管道实现问题

在 Bash 中我们可以通过管道轻松组合 gzip 和 gunzip 实现输入输出直通：

echo "test" | gzip | gunzip
# 输出结果: test

但你编写的 Haskell 代码在传入"test"后始终无法终止，核心问题出在进程输入流未关闭和进程生命周期管理缺失上，以下是具体分析和修正方案：

问题根源 #

1. gzip 进程未收到输入结束信号：你只通过 hPutStr 向 gzip 的输入句柄写入了数据，但没有关闭该句柄。gzip 会一直等待更多输入，不会输出任何压缩数据，导致下游的 gunzip 也阻塞在等待输入的状态，最终 hGetContents 永远无法获取到输出。
2. 未等待子进程完成：缺少对 gzip 和 gunzip 进程的等待逻辑，可能导致资源泄漏，同时无法确保进程完成所有数据处理。

修正后的代码 #

方案一：按顺序等待进程完成 #

先让 gzip 处理完所有输入并输出数据，再启动 gunzip 处理：

module GZipID where

import System.Process
import System.IO
import Control.Concurrent (waitForProcess)

compress :: String -> IO String
compress input = do
    -- 启动gzip进程，获取句柄和进程引用
    (Just hin, Just hout, _, gzipProc) <- createProcess (proc "gzip" [])
        { std_in  = CreatePipe
        , std_out = CreatePipe
        }
    
    -- 写入输入后立即关闭句柄，告诉gzip输入已结束
    hPutStr hin input
    hClose hin
    
    -- 等待gzip进程处理完成，确保所有压缩数据写入输出管道
    _ <- waitForProcess gzipProc
    
    -- 启动gunzip进程，复用gzip的输出管道作为输入
    (_, Just hfin, _, gunzipProc) <- createProcess (proc "gunzip" [])
        { std_in  = UseHandle hout
        , std_out = CreatePipe
        }
    
    -- 读取gunzip的输出
    result <- hGetContents hfin
    
    -- 等待gunzip进程完成，关闭所有句柄
    _ <- waitForProcess gunzipProc
    hClose hfin
    hClose hout
    
    return result

方案二：模拟Bash异步管道 #

让 gzip 和 gunzip 同时运行，更贴近原生管道的异步处理逻辑：

module GZipID where

import System.Process
import System.IO
import Control.Concurrent (waitForProcess)
import Control.Monad (void)

compress :: String -> IO String
compress input = do
    -- 启动gzip进程
    (Just hin, Just hout, _, gzipProc) <- createProcess (proc "gzip" [])
        { std_in  = CreatePipe
        , std_out = CreatePipe
        }
    
    -- 直接启动gunzip进程，用gzip的输出作为输入
    (_, Just hfin, _, gunzipProc) <- createProcess (proc "gunzip" [])
        { std_in  = UseHandle hout
        , std_out = CreatePipe
        }
    
    -- 写入输入并关闭gzip的输入句柄，触发压缩流程
    hPutStr hin input
    hClose hin
    
    -- 读取gunzip的输出（懒加载特性会自动触发数据流动）
    result <- hGetContents hfin
    
    -- 等待两个进程全部完成，避免僵尸进程
    void $ waitForProcess gzipProc
    void $ waitForProcess gunzipProc
    
    -- 关闭剩余句柄，释放资源
    hClose hout
    hClose hfin
    
    return result

关键注意事项 #

• 必须关闭输入句柄：hClose hin 是解决阻塞的核心操作，它会向 gzip 发送EOF信号，告知进程没有更多输入，从而触发压缩和输出流程。
• 等待进程退出：waitForProcess 确保子进程正常终止，避免系统资源泄漏，同时保证所有数据都被完整处理。
• 句柄资源管理：所有通过管道创建的句柄都需要显式关闭，防止资源泄漏。

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