Go语言中向已关闭的Tar归档文件追加内容的实现方法

package main

import (
    "archive/tar"
    "log"
    "os"
    "path/filepath" // 引入 filepath 用于获取当前目录
)

func main() {
    // 获取当前执行目录，确保文件路径正确
    execDir, err := os.Getwd()
    if err != nil {
        log.Fatalln("获取当前目录失败:", err)
    }
    tarFilePath := filepath.Join(execDir, "test.tar")

    // --- 阶段一：创建初始tar归档 ---
    log.Println("--- 阶段一：创建初始tar归档 ---")
    f, err := os.Create(tarFilePath)
    if err != nil {
        log.Fatalln("创建文件失败:", err)
    }

    tw := tar.NewWriter(f)

    var initialFiles = []struct {
        Name, Body string
    }{
        {"readme.txt", "这是一个包含一些文本文件的归档。"},
        {"gopher.txt", "Gopher 的名字：\n乔治\n杰弗里\n冈萨洛"},
        {"todo.txt", "获取动物处理许可证。"},
    }

    for _, file := range initialFiles {
        hdr := &tar.Header{
            Name: file.Name,
            Size: int64(len(file.Body)),
        }
        if err := tw.WriteHeader(hdr); err != nil {
            log.Fatalln("写入文件头失败:", err)
        }
        if _, err := tw.Write([]byte(file.Body)); err != nil {
            log.Fatalln("写入文件内容失败:", err)
        }
        log.Printf("已写入初始文件: %s\n", file.Name)
    }

    // 关闭tar写入器，这将写入归档结束标记
    if err := tw.Close(); err != nil {
        log.Fatalln("关闭tar写入器失败:", err)
    }
    // 关闭文件句柄
    if err := f.Close(); err != nil {
        log.Fatalln("关闭文件句柄失败:", err)
    }
    log.Printf("初始归档 '%s' 创建完成。\n", tarFilePath)

    // --- 阶段二：打开文件并追加更多内容 ---
    log.Println("\n--- 阶段二：打开文件并追加更多内容 ---")

    // 以读写模式打开文件
    // os.O_RDWR 允许读写
    // os.ModePerm 使用默认文件权限
    f, err = os.OpenFile(tarFilePath, os.O_RDWR, os.ModePerm)
    if err != nil {
        log.Fatalln("重新打开文件失败:", err)
    }

    // 将文件指针回溯1024字节，覆盖原有的归档结束标记
    // os.SEEK_END 表示从文件末尾开始计算偏移量
    if _, err = f.Seek(-1024, os.SEEK_END); err != nil {
        log.Fatalln("回溯文件指针失败:", err)
    }
    log.Println("文件指针已回溯1024字节，准备覆盖结束标记。")

    // 基于回溯后的文件句柄创建新的tar写入器
    tw = tar.NewWriter(f)

    // 要追加的新文件
    newFileContent := "这是新追加的文件内容。"
    newFileName := "foo.bar"

    hdr := &tar.Header{
        Name: newFileName,
        Size: int64(len(newFileContent)),
    }

    if err := tw.WriteHeader(hdr); err != nil {
        log.Fatalln("写入新文件头失败:", err)
    }

    if _, err := tw.Write([]byte(newFileContent)); err != nil {
        log.Fatalln("写入新文件内容失败:", err)
    }
    log.Printf("已追加新文件: %s\n", newFileName)

    // 再次关闭tar写入器，写入新的归档结束标记
    if err := tw.Close(); err != nil {
        log.Fatalln("关闭追加操作的tar写入器失败:", err)
    }
    // 关闭文件句柄
    if err := f.Close(); err != nil {
        log.Fatalln("关闭追加操作的文件句柄失败:", err)
    }
    log.Printf("文件 '%s' 追加操作完成。现在可以验证归档内容。\n", tarFilePath)

    // 验证归档内容 (可选)
    log.Println("\n--- 验证归档内容 ---")
    file, err := os.Open(tarFilePath)
    if err != nil {
        log.Fatalln("打开归档文件进行验证失败:", err)
    }
    defer file.Close()

    tr := tar.NewReader(file)
    for {
        hdr, err := tr.Next()
        if err == tar.EOF {
            break // End of archive
        }
        if err != nil {
            log.Fatalln("读取归档头失败:", err)
        }
        log.Printf("发现文件: %s (大小: %d)\n", hdr.Name, hdr.Size)
    }
    log.Println("归档内容验证完成。")
}