文件压缩的一种实现

记录一下对文件进行压缩的实现，这里只列出了核心代码

/*
函数名：	makeTarGzip
		制作压缩包，提高在线传输速率
参 数：
		encFile , 		密文文件
		signFile ,		签名文件
返回值：
		压缩后文件名
		错误信息
创建时间及创建者：
		2021-06-15	Yuan_sr
*/
func makeTarGzip(encFile, signFile string) (string, error){
	tarGzipFileName =   "example.tar.gz"
    
	// file write
	fw, err := os.Create(tarGzipFileName)
	if err != nil {
		return "", err
	}
	defer fw.Close()

	// gzip write
	gw := gzip.NewWriter(fw)
	defer gw.Close()

	// tar write
	tw := tar.NewWriter(gw)
	defer tw.Close()

	// 打开文件夹
	dir, err := os.Open("./")
	if err != nil {
		return "", err
	}
	defer dir.Close()

	// 读取文件列表
	fis, err := dir.Readdir(0)
	if err != nil {
		return "", err
	}
	for _, fi := range fis{
		if fi.IsDir(){
			continue
		}
		if fi.Name() == encFile || fi.Name() == signFile{ //去掉这个条件可以实现该目录下所有文件的压缩
			ef, err := os.Open(dir.Name() + "/" + fi.Name())
			if err != nil {
				return "", err
			}
			defer ef.Close()

			//信息头
			h := new(tar.Header)
			h.Name = fi.Name()
			h.Size = fi.Size()
			h.Mode = int64(fi.Mode())
			h.ModTime = fi.ModTime()
			//写信息头
			err = tw.WriteHeader(h)
			if err != nil {
				return "", err
			}
			_, err = io.Copy(tw, ef)
			if err != nil {
				return "", err
			}
		}
	}
	return tarGzipFileName, err
}

/*
函数名：	unTarGzip
		文件解压缩
参 数：
		tarGzipFile , 	待解压缩文件
返回值：
		错误信息
创建时间及创建者：
		2021-06-15	Yuan_sr
*/
func unTarGzip (tarGzipFile string) error {
	fr, err := os.Open(tarGzipFile)
	if err != nil {
		return err
	}
	defer fr.Close()
	gr, err := gzip.NewReader(fr)
	if err != nil {
		return err
	}
	defer gr.Close()
	tr := tar.NewReader(gr)

	for {
		h, err := tr.Next()
		if err == io.EOF {
			break
		}
		if err != nil {
			return  err
		}
		fw, err := os.OpenFile(h.Name, os.O_CREATE | os.O_WRONLY, 0755)
		if err != nil {
			return err
		}
		defer fw.Close()

		_, err = io.Copy(fw, tr)
		if err != nil {
			return err
		}
	}
	return nil
}