Golang中向现有Tar归档文件追加内容的实用技巧

package main

import (
    "archive/tar"
    "log"
    "os"
)

func main() {
    archivePath := "/tmp/test.tar" // 定义Tar文件路径

    // --- 阶段一：创建初始Tar归档文件 ---
    f, err := os.Create(archivePath)
    if err != nil {
        log.Fatalf("创建文件失败: %v", err)
    }
    defer f.Close() // 确保文件句柄在函数结束时关闭

    tw := tar.NewWriter(f)

    initialFiles := []struct {
        Name, Body string
    }{
        {"readme.txt", "This archive contains some text files."},
        {"gopher.txt", "Gopher names:\nGeorge\nGeoffrey\nGonzo"},
        {"todo.txt", "Get animal handling licence."},
    }

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

    if err := tw.Close(); err != nil {
        log.Fatalf("关闭tar writer失败 (初始): %v", err)
    }
    // 注意：这里f.Close()会由defer执行，但为了演示清晰，我们假设它在此处关闭
    // 实际上，为了进行后续的OpenFile，f必须先关闭。
    // 如果f被defer了，那么在第二次OpenFile之前，需要确保f已经关闭。
    // 在本例中，我们将f.Close()放在defer中是安全的，因为后续的OpenFile会重新获取文件句柄。
    log.Printf("初始Tar文件 '%s' 创建成功，包含 %d 个文件。", archivePath, len(initialFiles))


    // --- 阶段二：打开文件并追加内容 ---
    // 重新打开文件，注意使用 os.O_RDWR 模式
    f, err = os.OpenFile(archivePath, os.O_RDWR, os.ModePerm)
    if err != nil {
        log.Fatalf("重新打开文件失败: %v", err)
    }
    defer f.Close() // 确保文件句柄在函数结束时关闭

    // 将文件指针定位到文件末尾前1024字节，即覆盖原有的Tar结束标记
    if _, err = f.Seek(-1024, os.SEEK_END); err != nil {
        log.Fatalf("文件Seek操作失败: %v", err)
    }

    // 创建新的tar.Writer，它将从当前文件指针位置开始写入
    tw = tar.NewWriter(f)

    // 要追加的新文件
    newFileContent := "This is a new file appended to the archive."
    newFileName := "foo.bar"

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

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

    if _, err := tw.Write([]byte(newFileContent)); err != nil {
        log.Fatalf("写入新文件内容失败 (%s): %v", newFileName, err)
    }

    // 关闭tar.Writer，这将写入新的归档结束标记
    if err := tw.Close(); err != nil {
        log.Fatalf("关闭tar writer失败 (追加): %v", err)
    }
    log.Printf("文件 '%s' 成功追加到 Tar 归档。", newFileName)

    // 验证追加结果（可选）
    log.Println("验证Tar归档内容...")
    readAndVerifyTar(archivePath)
}

// readAndVerifyTar 用于读取并验证Tar归档内容
func readAndVerifyTar(archivePath string) {
    f, err := os.Open(archivePath)
    if err != nil {
        log.Fatalf("打开Tar文件进行验证失败: %v", err)
    }
    defer f.Close()

    tr := tar.NewReader(f)
    for {
        hdr, err := tr.Next()
        if err == tar.ErrHeader { // 遇到结束标记
            break
        }
        if err != nil {
            log.Fatalf("读取Tar头部失败: %v", err)
        }
        log.Printf("  发现文件: %s (大小: %d)", hdr.Name, hdr.Size)
    }
    log.Println("Tar归档内容验证完成。")
}