在Haskell

时间:2015-07-10 13:27:25

标签: file haskell io

我正在尝试读取文件的内容,将文本转换为大写,然后将其写回。

这是我写的代码:

import System.IO
import Data.Char

main = do
    handle <- openFile "file.txt" ReadWriteMode
    contents <- hGetContents handle
    hClose handle
    writeFile "file.txt" (map toUpper contents)
    return ()

但是,这对文件没有任何写入,事实上,它甚至会清除它。

我做了一些改变:

main = do
    handle <- openFile "file.txt" ReadWriteMode
    contents <- hGetContents handle
    writeFile "file.txt" (map toUpper contents)
    hClose handle
    return ()

但是,我收到错误resource busy (file is locked)。我怎样才能使这个工作和为什么它在两种情况下都不起作用?

4 个答案:

答案 0 :(得分:7)

懒惰的IO很糟糕,这通常被认为是Haskell的痛点。基本上,contents在您将其写回磁盘之前不会被评估,此时无法对其进行评估,因为该文件已经关闭。您可以通过多种方式解决此问题,无需使用额外的库,您可以使用readFile函数,然后在写回之前检查长度:

import Control.Monad (when)

main = do
    contents <- readFile "file.txt"
    let newContents = map toUpper contents
    when (length newContents > 0) $
        writeFile "file.txt" newContents

我会说这段代码实际上更好,因为你没有写回一个已经空的文件,这是一个毫无意义的操作。

另一种方法是使用流媒体库,pipes是一个受欢迎的选择,有一些很好的教程和坚实的数学基础,这也是我的选择。

答案 1 :(得分:4)

我认为你的问题是hGetContents是懒惰的。使用hGetContents时,不会立即读入文件的内容。他们在需要时会被读入。

在第一个示例中,您打开文件并说出您想要的内容,然后在对文件执行任何操作之前关闭该文件。然后你写入文件。当您写入现有文件时,内容将被清除,但由于您关闭了文件,因此您无法再访问文件内容。

在第二个示例中,您打开文件,然后尝试写入文件,但因为在需要它们之前(当它们被转换并写回时)内容才真正被读取,您最终会尝试写入和读取来自同一个文件。

您可以写入名为file2.txt的文件,然后在完成后删除file.txt并将file2.txt重命名为file.txt

答案 2 :(得分:4)

这里发生了一些事情

您在ReadWriteMode中打开了文件,但只阅读了内容。为什么不为两者使用相同的句柄?

main = do
    handle <- openFile "file.txt" ReadWriteMode
    contents <- hGetContents' handle
    hSeek handle AbsoluteSeek 0
    hPutStr handle (map toUpper contents)
    hClose handle
    return ()

hGetContents会将句柄置于半封闭状态,因此您需要其他内容来读取文件内容:

hGetContents' :: Handle -> IO String
hGetContents' h = do
  eof <- hIsEOF h
  if eof
    then
      return []
    else do
      c <- hGetChar h
      fmap (c:) $ hGetContents' h

答案 3 :(得分:4)

@ bwroga的答案是完全正确的。以下是建议方法的实现(写入临时文件和重命名):

import Data.Char (toUpper)
import System.Directory (renameFile, getTemporaryDirectory)
import System.Environment (getArgs)

main = do
    [file] <- getArgs
    tmpDir <- getTemporaryDirectory
    let tmpFile = tmpDir ++ "/" ++ file
    readFile file >>= writeFile tmpFile . map toUpper
    renameFile tmpFile file