使用 golang 解压缩文件的简单方法

Question

有没有一种简单的方法可以用 golang 解压缩文件？

现在我的代码是：

func Unzip(src, dest string) error {

    r, err := zip.OpenReader(src)

    if err != nil {

        return err

    }

    defer r.Close()

    for _, f := range r.File {

        rc, err := f.Open()

        if err != nil {

            return err

        }

        defer rc.Close()

        path := filepath.Join(dest, f.Name)

        if f.FileInfo().IsDir() {

            os.MkdirAll(path, f.Mode())

        } else {

            f, err := os.OpenFile(

                path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())

            if err != nil {

                return err

            }

            defer f.Close()

            _, err = io.Copy(f, rc)

            if err != nil {

                return err

            }

        }

    }

    return nil

}

Answer 1

OP 解决方案的轻微返工以创建包含目录（dest如果它不存在），并将文件提取/写入包装在一个闭包中以消除defer .Close()每个@Nick Craig-Wood评论的调用堆栈：func Unzip(src, dest string) error {    r, err := zip.OpenReader(src)    if err != nil {        return err    }    defer func() {        if err := r.Close(); err != nil {            panic(err)        }    }()    os.MkdirAll(dest, 0755)    // Closure to address file descriptors issue with all the deferred .Close() methods    extractAndWriteFile := func(f *zip.File) error {        rc, err := f.Open()        if err != nil {            return err        }        defer func() {            if err := rc.Close(); err != nil {                panic(err)            }        }()        path := filepath.Join(dest, f.Name)        // Check for ZipSlip (Directory traversal)        if !strings.HasPrefix(path, filepath.Clean(dest) + string(os.PathSeparator)) {            return fmt.Errorf("illegal file path: %s", path)        }        if f.FileInfo().IsDir() {            os.MkdirAll(path, f.Mode())        } else {            os.MkdirAll(filepath.Dir(path), f.Mode())            f, err := os.OpenFile(path, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, f.Mode())            if err != nil {                return err            }            defer func() {                if err := f.Close(); err != nil {                    panic(err)                }            }()            _, err = io.Copy(f, rc)            if err != nil {                return err            }        }        return nil    }    for _, f := range r.File {        err := extractAndWriteFile(f)        if err != nil {            return err        }    }    return nil}注意：更新为包括 Close() 错误处理（如果我们正在寻找最佳实践，不妨遵循所有这些）。