Question

我使用以下代码将某个文件夹压缩到给定路径，我目前面临的问题是我需要将包含内容的某个文件夹压缩到特定目标而不是在同一目录中

例如

路径中的文件夹的来源

"/Users/i03434/go/src/zdf/BUILD"

靶

"/Users/i03434/go/src/zdf/app/info.zip"

目前我尝试添加新路径[2]哪个没有帮助，任何想法怎么做？

这是所有代码

func zipit(params ...string) error {

    zipfile, err := os.Create(params[1])
    if err != nil {
        return err
    }
    defer zipfile.Close()

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

    info, err := os.Stat(params[0])
    if err != nil {
        return err
    }

    var baseDir string
    if info.IsDir(); len(params) > 2 {
        baseDir = params[2]
    } else {
        baseDir = filepath.Base(params[0])
    }

    if baseDir != "" {
        baseDir += "/"
    }

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

        if info.IsDir() {
            return nil
        }

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

        if baseDir != "" {
            header.Name = filepath.Join(strings.TrimPrefix(path, baseDir))
        }

        header.Method = zip.Deflate

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

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

    return err
}

zip的逻辑按预期工作，zip应该符合jar规范

Answer 1

用于测试：

您可以传递一个隐含了io.Writer接口的模拟变量zipFile作为函数zipit的参数，并将其与测试中的预期数据进行比较。
另外，您还需要在目标位置找到一组已知文件，以便将其与模拟中的值进行比较。

Refer this for testing io.Writer

测试代码：

使用source和newBaseDir的特定值创建的压缩文件的字符串值应事先知道并存储在want中。

func TestZipIt(t *testing.T) {

    source := ""
    newBaseDir := ""

    var zipFileMock bytes.Buffer
    if err := zipit(zipFileMock, source, newBaseDir); err != nil {
        t.Fatalf("zipit() returned an error: %s", err)
    }

    got := b.String()

    // want is the value of the zipped file as a string
    want := ...

    if got != want {
        t.Errorf("zipit() test failed")
        // t.Errorf("zipit() = %q, want %q", got, want)
    }
}

程序代码：

func main() {
    ...
    // params[0] is source
    // params[1] is destination
    // params[2] is newBaseDirectory

    zipfile, err := os.Create(destination)
    if err != nil {
        // handle error
    }
    defer zipfile.Close()

    if err = zipit(zipfile, source, newBaseDir); err != nil {
        // handle error
    }
    ...
}

func zipit(zipFile io.Writer, source, newBaseDir string) error {

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

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

    var baseDir string
    if info.IsDir() {
        baseDir = filepath.Dir(source)
    }

    if baseDir != "" {
        baseDir += "/"
    }

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

        if info.IsDir() {
            return nil
        }

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

        if baseDir != "" {
            header.Name = filepath.Join(newBaseDir, strings.TrimPrefix(path, baseDir))
        }

        header.Method = zip.Deflate

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

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

    return err
}

原始答案：

有几件事要考虑。

您需要使用filepath.Dir而不是filepath.Base。 Base给出路径的最后一个元素而不是基本目录。
if info.IsDir(); len(params) > 2 {仅检查条件len(params) > 2，对info.IsDir()进行评估，但未在任何地方使用。

引用：如果使用Go [1] [2]

中的简短语句格式

应该是

if info.IsDir() {
    if len(params) > 2 {
        ...
    } else {
        ...
    }
}

如果我对您的要求的理解正确。需要从文件标题中删除旧的基本路径，然后如果可用，则替换为param [2]的值；如果为空，则假定我们维护一个相对的文件结构。

更改：

var baseDir string
if info.IsDir(); len(params) > 2 {
    baseDir = params[2]
} else {
    baseDir = filepath.Base(params[0])
}

应该是

    var baseDir, newBaseDir string
    if info.IsDir() {
        baseDir = filepath.Dir(params[0])
        if len(params) > 2 {
            newBaseDir = params[2]
        }
    }

AND

header.Name = filepath.Join(strings.TrimPrefix(path, baseDir))
成为
header.Name = filepath.Join(newBaseDir, strings.TrimPrefix(path, baseDir))

包含源和目标的zip文件夹

1 个答案: