有没有办法用Haskell的管道模拟三通?

时间:2018-02-03 23:37:11

标签: haskell conduit

我正在尝试从Haskell程序中生成一个进程,我想将其标准错误流打印到屏幕上,同时将相同的流写入文件,就像tee命令实现的那样。

我可以打印标准错误流:

import Data.Conduit ((.|), runConduit)
import qualified Data.Conduit.List as CL
import Data.Conduit.Process

main :: IO ()
main = do
  (ClosedStream, ClosedStream, err, sph) <- streamingProcess (shell myCommand)

  runConduit $ err .| CL.mapM_ print

我可以将流指向文件:

import System.IO (withFile, IOMode (..))
import Data.Conduit.Process

main :: IO ()
main = do
  let logFile = "myCommand.log"
  withFile logFile WriteMode $ \h -> do
    (ClosedStream, ClosedStream, UseProvidedHandle, sph) <-
      streamingProcess (shell myCommand) {std_err = UseHandle h}

我怎样才能同时做到这两件事?

1 个答案:

答案 0 :(得分:5)

一种方法是使用conduitFile函数:

#!/usr/bin/env stack
-- stack --resolver lts-10.3 script
import Conduit
import Data.Conduit.Process
import Data.Conduit.Binary (conduitFile)

main :: IO ()
main = do
(ClosedStream, out, ClosedStream, ph) <-
    streamingProcess (shell "cat /etc/passwd")

runConduitRes $ out .| conduitFile "myCommand.log" .| printC

另一种方法是使用ZipSink一次以两种不同的方式使用流:

#!/usr/bin/env stack
-- stack --resolver lts-10.3 script
import Conduit
import Data.Conduit.Process

main :: IO ()
main = do
(ClosedStream, out, ClosedStream, ph) <-
    streamingProcess (shell "cat /etc/passwd")

runConduitRes $ out .| getZipSink (
    ZipSink (sinkFile "myCommand.log") *>
    ZipSink printC)