Golang实现文件压缩解压方法

Go语言通过archive/zip和compress/gzip包支持文件压缩解压，ZIP适用于多文件打包保留结构，GZIP适合单个文件高压缩比场景，结合实际需求选择格式并正确管理资源释放即可高效实现压缩功能。

在Go语言开发中，处理文件压缩与解压是常见的需求，比如日志归档、配置打包或网络传输优化。Golang标准库提供了强大的支持，尤其是archive/zip和compress/gzip包，能够满足大多数场景下的压缩与解压操作。

Golang 中常用的压缩方式

常见的文件压缩格式有 ZIP 和 GZIP。它们用途不同：

• ZIP：适合多个文件打包压缩，保留文件结构，常用于归档
• GZIP：通常用于单个文件压缩，不支持多文件，但压缩率高，常配合HTTP传输使用

使用 zip 实现多文件压缩与解压

Go 的 archive/zip 包可以方便地创建和读取 ZIP 压缩包。

以下代码将指定目录下的所有文件打包成一个 ZIP 文件：

package main

import (
    "archive/zip"
    "io"
    "os"
    "path/filepath"
)

func compressToZip(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
        }

        // 跳过目录本身
        if path == sourceDir {
            return nil
        }

        // 创建文件头
        header, _ := zip.FileInfoHeader(info)
        header.Name = path[len(sourceDir)+1:] // 相对路径

        if info.IsDir() {
            header.Name += "/"
            zipWriter.CreateHeader(header)
            return nil
        }

        writer, err := zipWriter.CreateHeader(header)
        if err != nil {
            return err
        }

        file, err := os.Open(path)
        if err != nil {
            return err
        }
        defer file.Close()

        _, err = io.Copy(writer, file)
        return err
    })

    return nil
}

下面是从 ZIP 包中解压所有文件到目标目录的实现：

func decompressFromZip(zipFile, destDir string) error {
    reader, err := zip.OpenReader(zipFile)
    if err != nil {
        return err
    }
    defer reader.Close()

    for _, file := range reader.File {
        filePath := filepath.Join(destDir, file.Name)

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

        if err := os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err != nil {
            return err
        }

        f, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, file.Mode())
        if err != nil {
            return err
        }

        rc, err := file.Open()
        if err != nil {
            f.Close()
            return err
        }

        _, err = io.Copy(f, rc)
        rc.Close()
        f.Close()

        if err != nil {
            return err
        }
    }

    return nil
}

使用 gzip 压缩单个文件

GZIP 更适合单个大文件的压缩，例如日志文件。

将一个普通文件压缩为 .gz 格式：

import (
    "compress/gzip"
    "os"
)

func compressWithGzip(source, gzFile string) error {
    inFile, err := os.Open(source)
    if err != nil {
        return err
    }
    defer inFile.Close()

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

    gzWriter := gzip.NewWriter(outFile)
    defer gzWriter.Close()

    _, err = io.Copy(gzWriter, inFile)
    return err
}

将 .gz 文件还原为原始内容：

func decompressGzip(gzFile, target string) error {
    gzFileReader, err := os.Open(gzFile)
    if err != nil {
        return err
    }
    defer gzFileReader.Close()

    gzReader, err := gzip.NewReader(gzFileReader)
    if err != nil {
        return err
    }
    defer gzReader.Close()

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

    _, err = io.Copy(outFile, gzReader)
    return err
}

实际使用建议

• 对于需要保留目录结构的批量文件，优先选择 ZIP
• 传输或存储单个大文本文件时，GZIP 效率更高
• 注意设置合理的文件权限（如 os.ModePerm）
• 压缩过程中及时关闭资源，避免句柄泄露
• 大文件处理时可考虑分块读写，降低内存占用

基本上就这些。Golang 的标准库已经足够应对大多数压缩需求，无需引入第三方依赖。掌握 zip 和 gzip 的基本用法后，可以轻松集成到日志系统、API 响应、配置导出等模块中。关键在于理解每种格式的适用场景，并正确管理文件流和资源释放。