快捷導(dǎo)航

詳解Go語言如何高效解壓ZIP文件

更新時(shí)間：2025年03月24日 10:35:34 作者：程序員愛釣魚

在日常開發(fā)中,我們經(jīng)常需要處理 ZIP 文件,本文主要為大家介紹一個(gè)使用 Go 語言編寫的高效 ZIP 文件解壓工具,希望對大家有一定的幫助

在日常開發(fā)中，我們經(jīng)常需要處理 ZIP 文件，例如從遠(yuǎn)程服務(wù)器下載壓縮包后解壓、備份數(shù)據(jù)或處理日志文件等。在本文中，我們將介紹一個(gè)使用 Go 語言編寫的高效 ZIP 文件解壓工具，并提供示例代碼幫助你快速上手。

代碼實(shí)現(xiàn)

以下是 Unzip 函數(shù)的完整實(shí)現(xiàn)，它可以將 ZIP 文件解壓到指定的目錄，并返回解壓后的文件路徑列表。

package utils

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

// Unzip 解壓 ZIP 文件到指定目錄
func Unzip(zipFile string, destDir string) ([]string, error) {
	zipReader, err := zip.OpenReader(zipFile)
	var paths []string
	if err != nil {
		return []string{}, err
	}
	defer zipReader.Close()

	for _, f := range zipReader.File {
		if strings.Index(f.Name, "..") > -1 {
			return []string{}, fmt.Errorf("%s 文件名不合法", f.Name)
		}
		fpath := filepath.Join(destDir, f.Name)
		paths = append(paths, fpath)
		if f.FileInfo().IsDir() {
			os.MkdirAll(fpath, os.ModePerm)
		} else {
			if err = os.MkdirAll(filepath.Dir(fpath), os.ModePerm); err != nil {
				return []string{}, err
			}

			inFile, err := f.Open()
			if err != nil {
				return []string{}, err
			}
			defer inFile.Close()

			outFile, err := os.OpenFile(fpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())
			if err != nil {
				return []string{}, err
			}
			defer outFile.Close()

			_, err = io.Copy(outFile, inFile)
			if err != nil {
				return []string{}, err
			}
		}
	}
	return paths, nil
}

使用示例

在 main.go 文件中，我們可以使用 Unzip 方法來解壓 ZIP 文件。

package main

import (
	"fmt"
	"log"
	"utils"
)

func main() {
	zipFile := "example.zip"  // 需要解壓的 ZIP 文件
	destDir := "output"        // 解壓目標(biāo)目錄

	files, err := utils.Unzip(zipFile, destDir)
	if err != nil {
		log.Fatalf("解壓失敗: %v", err)
	}

	fmt.Println("解壓成功，文件列表:")
	for _, file := range files {
		fmt.Println(file)
	}
}

代碼解析

打開 ZIP 文件：zip.OpenReader(zipFile) 用于打開 ZIP 文件。
遍歷 ZIP 內(nèi)部文件：使用 for _, f := range zipReader.File 遍歷 ZIP 內(nèi)的所有文件。
安全性檢查：通過 strings.Index(f.Name, "..") 防止路徑遍歷攻擊。
創(chuàng)建目錄和文件：使用 os.MkdirAll 創(chuàng)建必要的文件夾，并使用 os.OpenFile 創(chuàng)建文件。
文件復(fù)制：io.Copy(outFile, inFile) 復(fù)制文件內(nèi)容。