9

我想从haskell 程序中调用一个进程并捕获stdout 以及stderr。

我所做的:

(_, stdout, stderr) <- readProcessWithExitCode "command" [] ""

问题:通过这种方式,stdout 和 stderr 被分别捕获,但是我希望消息出现在正确的位置(否则我会简单地stdout ++ stderr将错误消息与其对应的 stdout 分开)。

我知道如果我将输出通过管道传输到文件中,我可以实现这一点,即

tmp <- openFile "temp.file" ...
createProcess (proc "command" []) { stdout = UseHandle tmp,
                                    stderr = UseHandle tmp }

所以我目前的解决方法是将输出通过管道传输到临时文件并将其读回。但是我正在寻找一种更直接的方法。

如果我肯定在 unix 上,我会简单地调用一个 shell 命令 á la

command 2>&1

就是这样。但是,我希望它尽可能便携。

我需要这个:我已经构建了一个很小的 ​​haskell cgi 脚本(只是为了玩它),它调用某个程序并打印输出。我想对输出进行 html 转义,因此我不能简单地将其通过管道传输到标准输出。

我在想:也许可以创建一个内存中的句柄,比如 Java 中的 PipedInputStream/PipedOutputStream,或者允许在内存中处理 IO 流的 ArrayInputStream/ArrayOutputStream。我四处寻找:: Handle关于hoogle的功能,但没有找到任何东西。

也许还有另一个 Haskell 模块可以让我合并两个流?

4

2 回答 2

6

您可以使用pipes同时合并两个输入流。第一个技巧是同时从两个流中读取,你可以使用stm包来做到这一点:

import Control.Applicative
import Control.Proxy
import Control.Concurrent
import Control.Concurrent.STM
import System.Process

toTMVarC :: (Proxy p) => TMVar a -> () -> Consumer p a IO r
toTMVarC tmvar () = runIdentityP $ forever $ do
    a <- request ()
    lift $ atomically $ putTMVar tmvar a

fromTMVarS :: (Proxy p) => TMVar a -> () -> Producer p a IO r
fromTMVarS tmvar () = runIdentityP $ forever $ do
    a <- lift $ atomically $ takeTMVar tmvar
    respond a

pipes-stm我将很快在一个包中提供上述原语,但现在使用上述原语。

然后,您只需将每个Handle内容分别提供给一个MVar并同时从两者中读取:

main = do
    (_, mStdout, mStderr, _) <- createProcess (proc "ls" [])
    case (,) <$> mStdout <*> mStderr of
        Nothing               -> return ()
        Just (stdout, stderr) -> do
            out <- newEmptyTMVarIO
            err <- newEmptyTMVarIO
            forkIO $ runProxy $ hGetLineS stdout >-> toTMVarC out
            forkIO $ runProxy $ hGetLineS stderr >-> toTMVarC err
            let combine () = runIdentityP $ forever $ do
                    str <- lift $ atomically $
                        takeTMVar out `orElse` takeTMVar err
                    respond str
            runProxy $ combine >-> putStrLnD

只需更改putStrLnD您想要处理输入的任何内容。

要了解有关该pipes程序包的更多信息,请阅读Control.Proxy.Tutorial

于 2012-12-24T21:56:56.117 回答
4

For posix system you can use createPipe and fdToHandle in System.Posix.IO to create a pair of new handles (I'm not sure where to close those handles and fds though..):

  readProcessWithMergedOutput :: String -> IO (ExitCode, String)
  readProcessWithMergedOutput command = do
    (p_r, p_w) <- createPipe
    h_r <- fdToHandle p_r
    h_w <- fdToHandle p_w
    (_, _, _, h_proc) <- createProcess (proc command [])
                                       { std_out = UseHandle h_w
                                       , std_err = UseHandle h_w
                                       }
    ret_code <- waitForProcess h_proc
    content <- hGetContents h_r
    return (ret_code, content)

For windows, this post implemented a cross-platform createPipe.

于 2012-12-25T03:18:20.467 回答