go 语言压缩生成多一级和不多第一级的代码。

多一级

func Zip(dest string, paths ...string) error {
    zfile, err := os.Create(dest)
    defer zfile.Close()
    if err != nil {
        return err
    }
    defer zfile.Close()
    zipWriter := zip.NewWriter(zfile)
    defer zipWriter.Close()
    for _, src := range paths {
        // remove the trailing path sepeartor if it is a directory
        src := strings.TrimSuffix(src, string(os.PathSeparator))
        err = filepath.Walk(src, func(path string, info os.FileInfo, err error) error {
            if err != nil {
                return err
            }
            // create local file header
            header, err := zip.FileInfoHeader(info)
            if err != nil {
                return err
            }
            // set compression method to deflate
            header.Method = zip.Deflate
            // set relative path of file in zip archive
            header.Name, err = filepath.Rel(filepath.Dir(src), path)
            if err != nil {
                return err
            }
            if info.IsDir() {
                header.Name += string(os.PathSeparator)
            }
            // create writer for writing header
            headerWriter, err := zipWriter.CreateHeader(header)
            if err != nil {
                return err
            }
            if info.IsDir() {
                return nil
            }
            f, err := os.Open(path)
            if err != nil {
                return err
            }
            defer f.Close()
            _, err = io.Copy(headerWriter, f)
            return err
        })
        if err != nil {
            return err
        }
    }
    return nil
}

func Zip2(dest string, paths ...string) error {
    zfile, err := os.Create(dest)
    defer zfile.Close()
    if err != nil {
        return err
    }
    defer zfile.Close()
    zipWriter := zip.NewWriter(zfile)
    defer zipWriter.Close()
    for _, src := range paths {
        // remove the trailing path sepeartor if it is a directory
        src := strings.TrimSuffix(src, string(os.PathSeparator))
        err = filepath.Walk(src, func(path string, info os.FileInfo, err error) error {
            if path == src {
                return nil
            }
            if err != nil {
                return err
            }
            // create local file header
            header, err := zip.FileInfoHeader(info)
            header.Name = strings.TrimPrefix(path, src+`\`)
            if err != nil {
                return err
            }
            // set compression method to deflate
            if info.IsDir() {
                header.Name += `\`
            } else {
                header.Method = zip.Deflate
            }

            // create writer for writing header
            headerWriter, err := zipWriter.CreateHeader(header)
            if err != nil {
                return err
            }
            if !info.IsDir() {
                f, _ := os.Open(path)
                defer f.Close()
                _, err = io.Copy(headerWriter, f)
            }
            return nil

        })
    }
    return nil
}

不会多

func Zip2(dest string, paths ...string) error {
    zfile, err := os.Create(dest)
    defer zfile.Close()
    if err != nil {
        return err
    }
    defer zfile.Close()
    zipWriter := zip.NewWriter(zfile)
    defer zipWriter.Close()
    for _, src := range paths {
        // remove the trailing path sepeartor if it is a directory
        src := strings.TrimSuffix(src, string(os.PathSeparator))
        err = filepath.Walk(src, func(path string, info os.FileInfo, err error) error {
            if path == src {
                return nil
            }
            if err != nil {
                return err
            }
            // create local file header
            header, err := zip.FileInfoHeader(info)
            header.Name = strings.TrimPrefix(path, src+`\`)
            if err != nil {
                return err
            }
            // set compression method to deflate
            if info.IsDir() {
                header.Name += `\`
            } else {
                header.Method = zip.Deflate
            }

            // create writer for writing header
            headerWriter, err := zipWriter.CreateHeader(header)
            if err != nil {
                return err
            }
            if !info.IsDir() {
                f, _ := os.Open(path)
                defer f.Close()
                _, err = io.Copy(headerWriter, f)
            }
            return nil

        })
    }
    return nil
}

统一解压代码

func Unzip(src string, dest string) error {
	reader, err := zip.OpenReader(src)
	if err != nil {
		return err
	}
	defer reader.Close()
	for _, file := range reader.File {
		filePath := path.Join(dest, file.Name)
		if file.FileInfo().IsDir() {
			os.MkdirAll(filePath, os.ModePerm)
		} else {
			if err = os.MkdirAll(filepath.Dir(filePath), os.ModePerm); err != nil {
				return err
			}
			inFile, err := file.Open()
			if err != nil {
				return err
			}
			defer inFile.Close()
			outFile, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, file.Mode())
			if err != nil {
				return err
			}
			defer outFile.Close()
			_, err = io.Copy(outFile, inFile)
			if err != nil {
				return err
			}
		}
	}
	return nil
}

测试如下

dir_path := "D:\\qinyu\\go-project-test\\patternTargetZip"
Zip2("baseline_p-b20.zip", dir_path)
Unzip("baseline_p-b20.zip", "") //工作目录

dir_path := "D:\\qinyu\\go-project-test\\patternTargetZip"
Zip("baseline_p-b20.zip", dir_path)
Unzip("baseline_p-b20.zip", "") //无果

dir_path := "D:\\qinyu\\go-project-test\\patternTargetZip"
Zip2("baseline_p-b20.zip", dir_path)
Unzip("baseline_p-b20.zip", "baseline_p-b20") //指定baseline_p-b20目录下，不再多一级

dir_path := "D:\\qinyu\\go-project-test\\patternTargetZip"
Zip("baseline_p-b20.zip", dir_path)
Unzip("baseline_p-b20.zip", "baseline_p-b20") //指定baseline_p-b20目录下，多一级