首頁  >  文章  >  後端開發  >  Go 中如何在沒有根資料夾的情況下壓縮資料夾內容?

Go 中如何在沒有根資料夾的情況下壓縮資料夾內容?

Susan Sarandon
Susan Sarandon原創
2024-11-18 05:03:02749瀏覽

How to Zip Folder Content Without the Root Folder in Go?

在沒有根資料夾的情況下壓縮資料夾中的內容

要求是建立一個包含目錄中檔案的ZIP 文件,不包括目錄本身作為提取時的根資料夾。

提供的程式碼片段嘗試透過使用以下行設定標頭名稱來實現此目的:

header.Name = filepath.Join(baseDir, strings.TrimPrefix(path, source))

但是,此程式碼新增了基本目錄(例如,「dir1」)新增至標頭名稱中,從而產生檔案嵌套在目錄結構中的存檔。

要解決此問題,應將該行替換為以下內容:

header.Name = strings.TrimPrefix(path, source)

這會從標頭名稱中刪除基本目錄,確保在沒有目錄結構的情況下提取文件。

修改後的程式碼將如下所示:

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

func Zipit(source, target string) error {
    zipfile, err := os.Create(target)
    if err != nil {
        return err
    }
    defer zipfile.Close()

    archive := zip.NewWriter(zipfile)
    defer archive.Close()

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

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

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

        if info.IsDir() {
            header.Name += "/"
        } else {
            header.Method = zip.Deflate
        }

        writer, err := archive.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
}

func main() {
    err := Zipit("path/dir1" +"/", "test"+".zip")
    if err != nil {
        fmt.Println(err)
    }
}

此程式碼有效地壓縮「dir1」目錄中的內容,而不將目錄本身包含在 ZIP 檔案中。

以上是Go 中如何在沒有根資料夾的情況下壓縮資料夾內容?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn