Как эффективно повторно использовать входные данные из стандартного ввода в Haskell

Question

Я понимаю, что я не должен пытаться перечитываю из stdin из-за ошибок о Haskell IO - handle closed Например, в поле ниже:

main = do 
    x <- getContents 
    putStrLn $ map id x 
    x <- getContents  --problem line 
    putStrLn x 

второй вызов x <- getContents вызовет ошибка:

test: <stdin>: hGetContents: illegal operation (handle is closed) 

конечно, я могу опустить вторую линию для чтения из getContents.

main = do 
    x <- getContents 
    putStrLn $ map id x 
    putStrLn x 

Но будет ли это проблемой производительности/памяти? Будет ли GHC хранить все содержимое с stdin в основной памяти?

Я предполагаю, что в первый раз, когда потребляется x, GHC может выбросить порты x, которые уже обработаны. Поэтому теоретически GHC может использовать только небольшую часть постоянной памяти для обработки. Но так как мы снова будем использовать x (и снова), похоже, что GHC ничего не может выбросить. (И он не может читать снова с stdin).

Мое понимание влияния памяти здесь правильно? И если да, есть ли исправление?

Answer 1

Да, ваше понимание верное: если вы повторно используете x, ghc должен хранить все это в памяти.

Я думаю, что возможное решение - использовать его лениво (один раз).

Предположим, вы хотите вывести x нескольким выходным ручкам hdls :: [Handle]. Наивный подход:

main :: IO() 
main = do 
    x <- getContents 
    forM_ hdls $ \hdl -> do 
     hPutStr hdl x 

Это будет читать stdin в x как первый hPutStr пересекает строку (по крайней мере, для небуферных ручек, hPutStr просто цикл, который вызывает hPutChar для каждого символа в строке). С этого момента он будет храниться в памяти для всех следующих hdl s.

В качестве альтернативы:

main :: IO() 
main = do 
    x <- getContents 
    forM_ x $ \c -> do 
     forM_ hdls $ \hdl -> do 
      hPutChar hdl c 

Здесь мы транспонированные петли: Вместо итерации ручки (и для каждой ручки итерации входных символов), мы перебирать входные символы, и для каждого символа , мы печатаем его каждому дескриптору.

Я не тестировал его, но эта форма должна гарантировать, что нам не нужно много памяти, потому что каждый входной символ c используется один раз, а затем отбрасывается.