Golang压缩解压文件 zip和tar标准库实践-Golang-PHP中文网

Golang压缩解压文件 zip和tar标准库实践

P粉602998670

发布： 2025-08-28 11:41:01

原创

837人浏览过

Golang中使用archive/zip和archive/tar库实现ZIP与TAR文件的压缩解压，通过CompressZip、DecompressZip、CompressTarGz和DecompressTarGz函数分别完成ZIP和带gzip压缩的TAR文件的读写操作，支持目录遍历、流式处理以降低内存占用，并结合defer确保资源释放；处理大文件时推荐分块读写避免内存溢出，同时通过gzip.Writer设置压缩级别可优化性能与压缩率，结合错误检查与MkdirAll创建目标路径，实现健壮的文件压缩解压功能。

golang压缩解压文件 zip和tar标准库实践

Golang提供了

archive/zip

登录后复制

和

archive/tar

登录后复制

标准库来处理ZIP和TAR文件的压缩与解压缩。本文将深入探讨这两个库的实践应用，助你轻松搞定文件压缩解压任务。

解决方案

Golang的标准库

archive/zip

登录后复制

和

archive/tar

登录后复制

分别用于处理ZIP和TAR格式的压缩文件。使用这两个库，我们可以方便地实现文件的压缩和解压缩功能。

ZIP文件的压缩与解压缩

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

压缩ZIP文件，需要创建ZIP文件，然后将需要压缩的文件写入该ZIP文件中。解压缩ZIP文件则需要打开ZIP文件，然后读取其中的每一个文件。

package main

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

// CompressZip 压缩文件到zip
func CompressZip(srcDir string, destZip string) error {
    outFile, err := os.Create(destZip)
    if err != nil {
        return err
    }
    defer outFile.Close()

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

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

        header, err := zip.FileInfoHeader(info)
        if err != nil {
            return err
        }

        header.Name, err = filepath.Rel(srcDir, path)
        if err != nil {
            return err
        }

        if info.IsDir() {
            header.Name += "/"
        }

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

        if info.IsDir() {
            return nil
        }

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

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

    return err
}

// DecompressZip 解压zip文件
func DecompressZip(srcZip string, destDir string) error {
    reader, err := zip.OpenReader(srcZip)
    if err != nil {
        return err
    }
    defer reader.Close()

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

        if file.FileInfo().IsDir() {
            err := os.MkdirAll(filePath, os.ModePerm)
            if err != nil {
                return err
            }
            continue
        }

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

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

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

        _, err = io.Copy(targetFile, fileReader)
        if err != nil {
            return err
        }
    }

    return nil
}

func main() {
    srcDir := "test_data" // 替换为你的源目录
    destZip := "test.zip"  // 替换为你的目标zip文件
    destDir := "unzipped"  // 替换为你的解压目标目录

    // 创建测试目录和文件
    os.MkdirAll(srcDir, os.ModePerm)
    os.WriteFile(filepath.Join(srcDir, "file1.txt"), []byte("This is file1."), os.ModePerm)
    os.WriteFile(filepath.Join(srcDir, "file2.txt"), []byte("This is file2."), os.ModePerm)
    os.MkdirAll(filepath.Join(srcDir, "subdir"), os.ModePerm)
    os.WriteFile(filepath.Join(srcDir, "subdir", "file3.txt"), []byte("This is file3."), os.ModePerm)

    err := CompressZip(srcDir, destZip)
    if err != nil {
        fmt.Println("Error compressing:", err)
        return
    }
    fmt.Println("Successfully compressed to", destZip)

    err = os.MkdirAll(destDir, os.ModePerm)
    if err != nil {
        fmt.Println("Error creating destination directory:", err)
        return
    }

    err = DecompressZip(destZip, destDir)
    if err != nil {
        fmt.Println("Error decompressing:", err)
        return
    }
    fmt.Println("Successfully decompressed to", destDir)

    //清理测试文件
    os.RemoveAll(srcDir)
    os.Remove(destZip)
    os.RemoveAll(destDir)
}

登录后复制

TAR文件的压缩与解压缩

TAR文件的压缩与ZIP类似，也是先创建TAR文件，然后将需要压缩的文件写入。解压缩TAR文件也是打开TAR文件，然后逐个读取文件。需要注意的是，TAR格式通常不包含压缩，所以一般会配合gzip等压缩算法一起使用。

巧文书

巧文书是一款AI写标书、AI写方案的产品。通过自研的先进AI大模型，精准解析招标文件，智能生成投标内容。

查看详情

package main

import (
    "archive/tar"
    "compress/gzip"
    "fmt"
    "io"
    "os"
    "path/filepath"
)

// CompressTarGz 压缩文件到tar.gz
func CompressTarGz(srcDir string, destTarGz string) error {
    outFile, err := os.Create(destTarGz)
    if err != nil {
        return err
    }
    defer outFile.Close()

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

    tarWriter := tar.NewWriter(gzipWriter)
    defer tarWriter.Close()

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

        header, err := tar.FileInfoHeader(info, info.Name())
        if err != nil {
            return err
        }

        header.Name, err = filepath.Rel(srcDir, path)
        if err != nil {
            return err
        }

        if info.IsDir() {
            header.Name += "/"
        }

        err = tarWriter.WriteHeader(header)
        if err != nil {
            return err
        }

        if info.IsDir() {
            return nil
        }

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

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

    return err
}

// DecompressTarGz 解压tar.gz文件
func DecompressTarGz(srcTarGz string, destDir string) error {
    file, err := os.Open(srcTarGz)
    if err != nil {
        return err
    }
    defer file.Close()

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

    tarReader := tar.NewReader(gzipReader)

    for {
        header, err := tarReader.Next()

        if err == io.EOF {
            break
        }

        if err != nil {
            return err
        }

        filePath := filepath.Join(destDir, header.Name)

        switch header.Typeflag {
        case tar.TypeDir:
            err := os.MkdirAll(filePath, os.ModePerm)
            if err != nil {
                return err
            }
        case tar.TypeReg:
            err := os.MkdirAll(filepath.Dir(filePath), os.ModePerm)
            if err != nil {
                return err
            }

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

            _, err = io.Copy(file, tarReader)
            if err != nil {
                return err
            }
        default:
            fmt.Printf("Unhandled type: %v\n", header.Typeflag)
        }
    }

    return nil
}

func main() {
    srcDir := "test_data"       // 替换为你的源目录
    destTarGz := "test.tar.gz" // 替换为你的目标tar.gz文件
    destDir := "untarred"       // 替换为你的解压目标目录

    // 创建测试目录和文件
    os.MkdirAll(srcDir, os.ModePerm)
    os.WriteFile(filepath.Join(srcDir, "file1.txt"), []byte("This is file1."), os.ModePerm)
    os.WriteFile(filepath.Join(srcDir, "file2.txt"), []byte("This is file2."), os.ModePerm)
    os.MkdirAll(filepath.Join(srcDir, "subdir"), os.ModePerm)
    os.WriteFile(filepath.Join(srcDir, "subdir", "file3.txt"), []byte("This is file3."), os.ModePerm)

    err := CompressTarGz(srcDir, destTarGz)
    if err != nil {
        fmt.Println("Error compressing:", err)
        return
    }
    fmt.Println("Successfully compressed to", destTarGz)

    err = os.MkdirAll(destDir, os.ModePerm)
    if err != nil {
        fmt.Println("Error creating destination directory:", err)
        return
    }

    err = DecompressTarGz(destTarGz, destDir)
    if err != nil {
        fmt.Println("Error decompressing:", err)
        return
    }
    fmt.Println("Successfully decompressed to", destDir)

    //清理测试文件
    os.RemoveAll(srcDir)
    os.Remove(destTarGz)
    os.RemoveAll(destDir)
}

登录后复制

如何处理大型文件压缩解压？

处理大型文件时，内存消耗是个大问题。为了避免