Go语言高效原地编辑大型文本文件指南

package main

import (
    "bufio"
    "fmt"
    "io"
    "io/ioutil"
    "os"
    "path/filepath"
    "strings"
)

// editFileInPlace 模拟在Go语言中原地编辑大型文本文件
// 它逐行读取文件，对满足条件的行进行修改，然后将结果写入一个临时文件，
// 最后用临时文件替换原始文件。
func editFileInPlace(filePath string, oldText, newText string) error {
    // 1. 打开原始文件进行读取
    originalFile, err := os.Open(filePath)
    if err != nil {
        return fmt.Errorf("无法打开原始文件 %s: %w", filePath, err)
    }
    defer originalFile.Close() // 确保文件关闭

    // 2. 创建一个临时文件用于写入修改后的内容
    // ioutil.TempFile 会在默认临时目录中创建一个唯一命名的文件
    tempFile, err := ioutil.TempFile(filepath.Dir(filePath), "temp_edit_")
    if err != nil {
        return fmt.Errorf("无法创建临时文件: %w", err)
    }
    defer tempFile.Close()         // 确保临时文件关闭
    defer os.Remove(tempFile.Name()) // 在函数结束时尝试删除临时文件，以防替换失败

    // 3. 使用 bufio.Scanner 逐行读取原始文件
    scanner := bufio.NewScanner(originalFile)
    writer := bufio.NewWriter(tempFile) // 使用 bufio.Writer 提高写入效率

    for scanner.Scan() {
        line := scanner.Text()
        // 执行您的修改逻辑
        if strings.Contains(line, oldText) {
            line = strings.ReplaceAll(line, oldText, newText)
            fmt.Printf("修改行: %s -> %s\n", scanner.Text(), line)
        }
        // 将修改后的行或原始行写入临时文件
        _, err := writer.WriteString(line + "\n")
        if err != nil {
            return fmt.Errorf("写入临时文件失败: %w", err)
        }
    }

    // 检查扫描过程中是否有错误
    if err := scanner.Err(); err != nil {
        return fmt.Errorf("读取原始文件失败: %w", err)
    }

    // 确保所有缓存的写入都已刷新到临时文件
    if err := writer.Flush(); err != nil {
        return fmt.Errorf("刷新写入器失败: %w", err)
    }

    // 4. 关闭文件以确保所有内容都已写入磁盘
    originalFile.Close() // 提前关闭原始文件，以便后续重命名
    tempFile.Close()     // 提前关闭临时文件，以便后续重命名

    // 5. 用临时文件替换原始文件
    // os.Rename 是一个原子操作，可以确保替换的安全性
    if err := os.Rename(tempFile.Name(), filePath); err != nil {
        return fmt.Errorf("重命名临时文件失败: %w", err)
    }

    fmt.Printf("文件 %s 已成功更新。\n", filePath)
    return nil
}

func main() {
    // 示例用法：
    fileName := "large_text_file.txt"
    content := `This is line 1.
Hello world!
This is line 3.
Hello Go!
This is line 5.
Hello world!`

    // 创建一个示例文件
    err := ioutil.WriteFile(fileName, []byte(content), 0644)
    if err != nil {
        fmt.Printf("创建示例文件失败: %v\n", err)
        return
    }
    fmt.Printf("创建文件 %s 成功。\n", fileName)

    // 调用原地编辑函数
    err = editFileInPlace(fileName, "Hello world!", "Go is awesome!")
    if err != nil {
        fmt.Printf("编辑文件失败: %v\n", err)
        return
    }

    // 验证文件内容
    fmt.Println("\n--- 编辑后的文件内容 ---")
    editedContent, err := ioutil.ReadFile(fileName)
    if err != nil {
        fmt.Printf("读取编辑后文件失败: %v\n", err)
        return
    }
    fmt.Println(string(editedContent))

    // 清理示例文件
    // os.Remove(fileName)
}