如何用Haskell的conduit-extra模拟tee命令功能？

要让进程的标准错误同时输出到屏幕和写入文件，核心思路是捕获进程的stderr流，然后将其内容复制到两个目标（stdout和文件）。下面我给你两种可行的实现方式，分别适合不同场景：

方法一：用基础的Process库 + 线程（简单易上手） #

这种方式依赖Haskell标准库中的process包，不需要额外的复杂依赖，适合小体量的输出场景。

步骤与代码 #

首先导入必要的模块：

import System.Process (createProcess, shell, StdStream(CreatePipe), waitForProcess, ProcessHandle)
import System.IO (Handle, hGetContents, hPutStr, hClose, stdout, openFile, IOMode(WriteMode), hFlush)
import Control.Concurrent (forkIO)
import Control.Monad (void)
import Control.Exception (catch, IOException)
import System.Exit (exitFailure)

然后定义一个实现tee逻辑的函数——读取输入流，同时写入两个输出流：

teeStream :: Handle -> Handle -> Handle -> IO ()
teeStream input output1 output2 = do
    content <- hGetContents input
    -- 同时写入两个目标
    hPutStr output1 content
    hPutStr output2 content
    -- 强制刷新缓冲区，避免内容延迟显示/写入
    hFlush output1
    hFlush output2

最后是主逻辑，创建进程、分流stderr：

main :: IO ()
main = do
    -- 创建目标进程，捕获它的stderr为管道
    procResult <- createProcess (shell "your-actual-command-here")
        { std_err = CreatePipe }
    case procResult of
        (_, _, Just stderrHandle, procHandle) -> do
            -- 打开要写入的文件
            fileHandle <- openFile "stderr-log.txt" WriteMode `catch` \(e :: IOException) -> do
                putStrLn $ "无法打开日志文件: " ++ show e
                hClose stderrHandle
                exitFailure
            
            -- 启动线程处理分流（避免阻塞主进程等待输出）
            void $ forkIO $ teeStream stderrHandle stdout fileHandle
            
            -- 等待目标进程执行完毕
            exitCode <- waitForProcess procHandle
            
            -- 关闭所有句柄
            hClose stderrHandle
            hClose fileHandle
            
            putStrLn $ "进程退出码: " ++ show exitCode
        _ -> putStrLn "创建进程失败，无法捕获stderr"

关键说明 #

• createProcess的std_err = CreatePipe参数让我们能拿到进程stderr的句柄，而不是直接输出到终端。
• 用forkIO启动线程处理分流，这样主进程可以同时等待目标进程结束，不会被输出处理阻塞。
• hFlush是必须的：默认IO操作有缓冲，不加的话可能要等到缓冲区满才会输出到屏幕/文件，体验很差。

方法二：用Conduit库（流式处理，适合大输出） #

如果你的进程会产生大量stderr输出，上面的惰性读取可能会占用过多内存。这时候可以用conduit库做流式处理，逐块读取并分流，内存效率更高。

步骤与代码 #

首先需要在你的cabal文件中添加依赖：

build-depends:       base >= 4.14 && < 5
                   , process
                   , conduit >= 1.3
                   , conduit-combinators

然后导入模块并实现逻辑：

import Conduit (runConduit, sourceHandle, sinkHandle, ($$), void)
import Data.Conduit.Combinators (tee)
import System.Process (createProcess, shell, StdStream(CreatePipe), waitForProcess)
import System.IO (Handle, openFile, IOMode(WriteMode), stdout, hClose)
import Control.Exception (catch, IOException)
import System.Exit (exitFailure)

main :: IO ()
main = do
    procResult <- createProcess (shell "your-actual-command-here")
        { std_err = CreatePipe }
    case procResult of
        (_, _, Just stderrHandle, procHandle) -> do
            fileHandle <- openFile "stderr-log.txt" WriteMode `catch` \(e :: IOException) -> do
                putStrLn $ "无法打开日志文件: " ++ show e
                hClose stderrHandle
                exitFailure
            
            -- 用conduit的tee函数流式分流
            void $ runConduit $ sourceHandle stderrHandle $$ tee (sinkHandle stdout) (sinkHandle fileHandle)
            
            waitForProcess procHandle
            hClose stderrHandle
            hClose fileHandle
        _ -> putStrLn "创建进程失败，无法捕获stderr"

关键说明 #

• Conduit的sourceHandle会逐块读取输入流，tee函数直接将每一块内容同时送到两个sinkHandle（分别对应stdout和文件），全程不会把所有内容加载到内存。
• 这种方式更适合处理日志、大输出场景，性能和内存占用更优。

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