在go中缓冲读取写入同一文件

Question

我正在尝试逐行读取一个非常大的文件，然后替换行中的某些内容并写回同一个文件。我找不到办法做到这一点。目前我已经创建了一个临时文件并将替换的行保存在临时文件中，最后将临时文件内容复制到原始文件中。但我必须进行两次读取和两次写入。我想避免这种情况。有任何想法吗。以下是我尝试过的：file, _ := os.Open("somelargesay1GbFile.txt")defer file.Close()outFile, _ := os.OpenFile("temp.txt", os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)defer outFile.Close()reader := bufio.NewReaderSize(file, 10*1024)for {    bytesread, isprefix, err := reader.ReadLine()    if err != nil {        if err == io.EOF {            data := changeTheData(string(bytesread))            outFile.WriteString(data)            if !isprefix {                outFile.WriteString("\r\n")            }        } else {            log.Fatal(err)        }        break    }    data := changeTheData(string(bytesread))    outFile.WriteString(data)    if !isprefix {        outFile.WriteString("\r\n")    }}最后我删除该文件somelargesay1GbFile.txt并将其重命名为temp.txt

Answer 1

解决方案是打开文件两次，第一次用于读取，第二次用于覆盖内容。

请参阅下面的代码。

请注意，我使用reader.ReadString('\n')而不是ReadLine. 另请注意，我在测试错误之前写入输出。这是因为最后一行不以“\n”结尾的行与错误一起返回io.EOF。

func main() {

    inFile, _ := os.Open("file.txt")

    defer inFile.Close()

    outFile, _ := os.OpenFile("file.txt", os.O_RDWR, 0777)

    defer outFile.Close()

    reader := bufio.NewReaderSize(inFile, 10*1024)

    for {

        line, err := reader.ReadString('\n')

        outFile.WriteString(changeData(line))

        if err != nil {

            if err != io.EOF {

                fmt.Println("error:", err)

            }

            break

        }

    }

}

我用这个函数测试了代码changeData，将小写字母更改为大写字母，将大写字母更改为小写字母。然后我们可以通过重复调用函数来测试更改。

func changeData(data string) string {

    o := make([]rune, 0, len(data))

    for _, r := range data {

        if unicode.IsLetter(r) {

            if unicode.IsUpper(r) {

                o = append(o, unicode.ToLower(r))

            } else {

                o = append(o, unicode.ToUpper(r))

            }

        } else {

            o = append(o, r)

        }

    }

    return string(o)

}

此代码将覆盖文件的每一行。该行是否被修改取决于changeData修改的内容。

如果您想避免覆盖每一行，因为大多数行不需要更改，那么您必须寻找要覆盖数据的位置。