Function packZipFile

common/persistence/problems/pack_zip.go:14–70

(rootPath string, targetPath string)

)

func packZipFile(rootPath string, targetPath string) error {
	fout, err := os.Create(targetPath)
	if err != nil {
		return errors.Errorf("create problem package file error: %s", err.Error())
	}
	defer fout.Close()

	zipWriter := zip.NewWriter(fout)
	defer zipWriter.Close()

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

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

		header.Name = strings.TrimPrefix(path, rootPath+"/")

		// 排除根目录
		if header.Name == rootPath {
			return nil
		}
		if info.IsDir() {
			header.Name += "/"
		} else {
			header.Method = zip.Deflate
		}
		// 排除bin目录
		if strings.HasPrefix(header.Name, "bin/") {
			return nil
		}

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

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

	return err
}