go 文件压缩归档（tar+gzip）

文件压缩归档

文件归档（tar）

tar归档示例

文件压缩（gzip）：

gzip文件压缩

文件压缩归档一起（tar + gzip）：

压缩归档示例：

package main

import (
	"archive/tar"
	"compress/gzip"
	"fmt"
	"io"
	"log"
	"os"
	"path/filepath"
)

func main() {
	fw, err := os.Create("test.tar.gz")
	if err != nil {
		log.Fatal(err)
	}
	defer fw.Close()

	var fileList []string
	dirPath := "D:\\testfile"
	dirE, err := os.ReadDir(dirPath)
	if err != nil {
		log.Fatal(err)
	}
	for _, v := range dirE {
		// 不递归目录
		if v.IsDir() {
			continue
		}
		fileList = append(fileList, v.Name())
	}

	gw := gzip.NewWriter(fw)
	defer gw.Close()
	tw := tar.NewWriter(gw)
	defer tw.Close()

	for _, v := range fileList {
		f, err := os.Open(filepath.Join(dirPath, v))
		if err != nil {
			fmt.Printf("open %s err:%s\n", v, err.Error())
			continue
		}

		info, _ := f.Stat()
		// 文件信息写入tar的头
		header, _ := tar.FileInfoHeader(info, "")
		// 定义压缩包中该文件的name, 可以加上多级目录, 如下为test/<file_name>
		header.Name = "test" + "/" + header.Name

		err = tw.WriteHeader(header)
		if err != nil {
			log.Println(err)
			f.Close()
			return
		}
		_, err = io.Copy(tw, f)
		if err != nil {
			log.Printf("copy %s, err:%s\n", v, err.Error())
		}
		f.Close()
	}

	return
}

解压缩示例：

package main

import (
	"archive/tar"
	"compress/gzip"
	"io"
	"log"
	"os"
	"path"
)

func main() {
	var dstDirPath = "/tmp/testDir/"

	f, err := os.Open("test.tar.gz")
	if err != nil {
		log.Fatal(err)
	}
	defer f.Close()

	gr, err := gzip.NewReader(f)
	if err != nil {
		log.Fatal(err)
	}
	defer gr.Close()

	tr := tar.NewReader(gr)
	for {
		hdr, err := tr.Next()
		if err == io.EOF {
			break // End of archive
		}
		if err != nil {
			log.Fatal(err)
		}

		// create directory
		os.MkdirAll(dstDirPath+"/"+path.Dir(hdr.Name), os.ModePerm)

		fw, err := os.OpenFile(dstDirPath+hdr.Name, os.O_CREATE|os.O_WRONLY, 0666 /*os.FileMode(h.Mode)*/)
		if err != nil {
			log.Fatal(err)
		}
		defer fw.Close()
		_, err = io.Copy(fw, tr)
		if err != nil {
			fw.Close()
			log.Fatal(err)
		}
		fw.Close()
	}
}