Golang如何实现文件备份与恢复-Golang-PHP中文网

Golang如何实现文件备份与恢复

P粉602998670

发布： 2025-10-25 09:36:02

原创

681人浏览过

答案：Go语言通过os、io和archive/zip包实现文件备份与恢复。1. 单文件备份使用os.Open和os.Create配合io.Copy复制内容；2. 多文件或目录备份利用filepath.Walk遍历并用zip.Writer将文件写入ZIP归档，保持路径结构；3. 恢复时通过zip.OpenReader解压，逐个提取文件，检查路径防止目录穿越；4. 实际应用需校验权限、避免覆盖、分块读写大文件并添加SHA256校验确保完整性。标准库已足够支撑稳定可靠的备份程序。

golang如何实现文件备份与恢复

在Go语言中实现文件备份与恢复，核心是利用标准库中的 os、io 和 archive/zip 等包来完成文件读取、写入和归档操作。整个过程不复杂，关键是处理好路径、权限和错误。

1. 单个文件的备份（复制）

最简单的备份方式就是将源文件复制到目标位置。

使用 os.Open 读取原文件，用 os.Create 创建目标文件，再通过 io.Copy 完成内容传输。

示例代码：

立即学习“go语言免费学习笔记（深入）”；

func copyFile(src, dst string) error {
    sourceFile, err := os.Open(src)
    if err != nil {
        return err
    }
    defer sourceFile.Close()

    destFile, err := os.Create(dst)
    if err != nil {
        return err
    }
    defer destFile.Close()

    _, err = io.Copy(destFile, sourceFile)
    return err
}

登录后复制

调用 copyFile("data.txt", "backup/data.txt") 即可完成单文件备份。

2. 多文件或目录打包为ZIP备份

实际应用中，通常需要备份整个目录。使用 zip 归档更便于管理和传输。

Go 的 archive/zip 包支持创建压缩文件，遍历目录并逐个写入 ZIP。

关键步骤：

使用 filepath.Walk 遍历目录
为每个文件在 ZIP 中创建对应路径的 zip.FileHeader
读取原文件内容并写入 zip.Writer

示例简化逻辑：

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

func backupToZip(sourceDir, zipFile string) error {
    f, err := os.Create(zipFile)
    if err != nil {
        return err
    }
    defer f.Close()

    zipWriter := zip.NewWriter(f)
    defer zipWriter.Close()

    filepath.Walk(sourceDir, func(path string, info os.FileInfo, err error) error {
        if err != nil {
            return err
        }

        // 创建文件头
        relPath, _ := filepath.Rel(sourceDir, path)
        header, _ := zip.FileInfoHeader(info, "")
        header.Name = relPath
        if info.IsDir() {
            header.Name += "/"
        }
        writer, _ := zipWriter.CreateHeader(header)

        if !info.IsDir() {
            file, _ := os.Open(path)
            defer file.Close()
            io.Copy(writer, file)
        }
        return nil
    })

    return nil
}

登录后复制

这样就能把整个文件夹打包成一个 .zip 文件用于长期保存。

3. 从备份中恢复文件

恢复即解压 ZIP 文件到指定目录，注意路径安全，避免目录穿越攻击。

读取 ZIP 文件，逐个提取内容，并确保目标路径在允许范围内。

基本流程：

打开 ZIP 文件，创建 zip.Reader
遍历每个文件项，检查路径是否包含 ".." 等非法字符
创建目标目录结构，写入文件内容

部分代码示意：

func restoreFromZip(zipFile, targetDir string) error {
    r, err := zip.OpenReader(zipFile)
    if err != nil {
        return err
    }
    defer r.Close()

    for _, file := range r.File {
        filePath := filepath.Join(targetDir, file.Name)
        
        // 简单防御路径穿越
        if !strings.HasPrefix(filePath, targetDir) {
            return fmt.Errorf("invalid file path: %s", filePath)
        }

        if file.FileInfo().IsDir() {
            os.MkdirAll(filePath, 0755)
            continue
        }

        rc, _ := file.Open()
        outFile, _ := os.Create(filePath)
        io.Copy(outFile, rc)
        outFile.Close()
        rc.Close()
    }
    return nil
}

登录后复制