Golang文件压缩解压实现方法

本文深入讲解了如何使用Golang标准库（archive/zip和compress/gzip）高效、安全地实现文件压缩与解压：zip适用于多文件打包，支持目录结构重建和DEFLATE压缩算法；gzip则专精于单文件高压缩场景，常用于日志归档或网络传输；文中不仅提供了完整可运行的示例代码，还强调了路径安全防护（如防范目录穿越攻击）、资源正确释放（defer关闭句柄）及格式选型建议，让开发者无需依赖第三方库即可构建健壮可靠的压缩解压功能。

在Golang中实现文件的压缩与解压，主要依赖标准库中的 archive/zip 和 compress/gzip。根据压缩格式的不同（如 zip 或 gzip），使用方式略有差异。下面分别介绍常见场景下的实现方法。

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

zip 是最常用的归档压缩格式之一，适合打包多个文件。Go 的 archive/zip 包提供了完整的支持。

压缩多个文件为 zip：

示例代码：

package main

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

func compressToZip(files []string, dest string) error {
    outFile, err := os.Create(dest)
    if err != nil {
        return err
    }
    defer outFile.Close()

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

    for _, file := range files {
        if err := addFileToZip(zipWriter, file); err != nil {
            return err
        }
    }
    return nil
}

func addFileToZip(zipWriter *zip.Writer, filename string) error {
    file, err := os.Open(filename)
    if err != nil {
        return err
    }
    defer file.Close()

    info, err := file.Stat()
    if err != nil {
        return err
    }

    header, err := zip.FileInfoHeader(info, "")
    if err != nil {
        return err
    }
    header.Method = zip.Deflate // 使用 DEFLATE 压缩算法

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

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

解压 zip 文件：

示例代码：

func decompressZip(zipPath, destDir string) error {
    reader, err := zip.OpenReader(zipPath)
    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
        }

        inFile, err := file.Open()
        if err != nil {
            return err
        }

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

        _, err = io.Copy(outFile, inFile)
        inFile.Close()
        outFile.Close()

        if err != nil {
            return err
        }
    }
    return nil
}

使用 gzip 实现单文件压缩与解压

gzip 通常用于单个文件的压缩，常配合网络传输或日志归档使用。它不支持多文件打包，但压缩率高。

压缩文件为 .gz：

func compressWithGzip(inputFile, outputFile string) error {
    in, err := os.Open(inputFile)
    if err != nil {
        return err
    }
    defer in.Close()

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

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

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

解压 .gz 文件：

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

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

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

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

实际使用建议

选择压缩方式时考虑以下几点：

基本上就这些。Go 标准库对压缩支持良好，无需引入第三方包即可完成大多数任务。关键在于正确管理资源（文件句柄、writer 关闭）和处理路径问题。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Golang文件压缩解压实现方法》文章吧，也可关注golang学习网公众号了解相关技术文章。