Bắt giữ/lấy cắp dữ liệu stdout trong haskell

Question

Làm cách nào để xác định 'catchOutput' để chỉ chạy các kết quả đầu ra chính 'thanh'?

Tức là, làm cách nào tôi có thể truy cập cả luồng đầu ra (đầu ra) và đầu ra thực tế của tác vụ io riêng biệt?

catchOutput :: IO a -> IO (a,String) 
catchOutput = undefined 

doSomethingWithOutput :: IO a -> IO() 
doSomethingWithOutput io = do 
    (_ioOutp, stdOutp) <- catchOutput io 
    if stdOutp == "foo" 
     then putStrLn "bar" 
     else putStrLn "fail!" 

main = doSomethingWithOutput (putStr "foo") 

tốt nhất giả thuyết "giải pháp" tôi đã tìm thấy cho đến nay bao gồm chuyển hướng stdout, inspired by this, để một dòng tập tin và sau đó đọc từ tập tin đó (Bên cạnh đó là siêu xấu xí tôi đã không có khả năng đọc trực tiếp sau khi viết từ một tập tin Có thể tạo ra một "dòng đệm tùy chỉnh" mà không cần phải lưu trữ trong một tập tin?). Mặc dù cảm thấy 'hơi' giống như một ca khúc phụ.

Một góc khác có vẻ như sử dụng 'hGetContents stdout' nếu điều đó được cho là làm những gì tôi nghĩ. Nhưng tôi không được phép đọc từ stdout. Mặc dù googling nó dường như cho thấy rằng nó đã được used.

Answer 1

Tại sao không chỉ sử dụng một người viết đơn thuần thay thế? Ví dụ,

import Control.Monad.Writer 

doSomethingWithOutput :: WriterT String IO a -> IO() 
doSomethingWithOutput io = do 
    (_, res) <- runWriterT io 
    if res == "foo" 
     then putStrLn "bar" 
     else putStrLn "fail!" 

main = doSomethingWithOutput (tell "foo") 

Ngoài ra, bạn có thể sửa đổi hành động bên trong của bạn để có một Handle để viết thư cho thay vì stdout. Sau đó bạn có thể sử dụng một cái gì đó như knob để tạo một bộ xử lý trong bộ nhớ mà bạn có thể chuyển đến hành động bên trong và kiểm tra nội dung của nó sau đó.

Answer 2

Như @hammar chỉ ra, bạn có thể sử dụng một núm để tạo ra một tập tin trong bộ nhớ, nhưng bạn cũng có thể sử dụng hDuplicate và hDuplicateTo thay đổi stdout đến tập tin bộ nhớ, và ngược lại. Nội dung nào đó giống như mã hoàn toàn chưa được kiểm tra sau đây:

catchOutput io = do 
    knob <- newKnob (pack []) 
    let before = do 
     h <- newFileHandle knob "<stdout>" WriteMode 
     stdout' <- hDuplicate stdout 
     hDuplicateTo h stdout 
     hClose h 
     return stdout' 
     after stdout' = do 
     hDuplicateTo stdout' stdout 
     hClose stdout' 
    a <- bracket_ before after io 
    bytes <- Data.Knob.getContents knob 
    return (a, unpack bytes) 

Answer 3

Tôi đã sử dụng chức năng sau để kiểm tra đơn vị chức năng in thành giá trị xuất chuẩn.

import GHC.IO.Handle 
import System.IO 
import System.Directory 

catchOutput :: IO() -> IO String 
catchOutput f = do 
    tmpd <- getTemporaryDirectory 
    (tmpf, tmph) <- openTempFile tmpd "haskell_stdout" 
    stdout_dup <- hDuplicate stdout 
    hDuplicateTo tmph stdout 
    hClose tmph 
    f 
    hDuplicateTo stdout_dup stdout 
    str <- readFile tmpf 
    removeFile tmpf 
    return str 

Tôi không chắc về cách tiếp cận tệp trong bộ nhớ, nhưng nó hoạt động tốt cho một lượng nhỏ đầu ra với một tệp tạm thời.

Answer 4

Có một số gói trên Hackage hứa hẹn sẽ thực hiện điều đó: io-capture và im lặng. âm thầm dường như được duy trì và hoạt động trên Windows quá (io-capture chỉ hoạt động trên Unix). Với âm thầm, bạn sử dụng chụp:

import System.IO.Silently 

main = do 
    (output, _) <- capture $ putStr "hello" 
    putStrLn $ output ++ " world" 

Lưu ý rằng nó hoạt động bằng cách chuyển hướng đầu ra sang tệp tạm thời và sau đó đọc ... Nhưng miễn là nó hoạt động!