首页 > 后端开发 > Golang > 正文

Go 中如何在没有根文件夹的情况下压缩文件夹内容?

Susan Sarandon
发布: 2024-11-18 05:03:02
原创
750 人浏览过

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中文网其他相关文章!

来源:php.cn
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
作者最新文章
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板