Go-归档文件-tar

package main

import (
	"log"
	"fmt"
	"io/ioutil"
	"os"
	"archive/tar"
	"io"
)

func main() {
	// 目标路径
	targetPath  := "C:\\workspace"
	//归档文件名
	appleFrameName := "appleFrame.tar"
	// 文件归档
	err := TarFile(targetPath, appleFrameName)
	if err != nil {
		log.Fatalln(err)
		return
	}
	err = TarRead(targetPath, appleFrameName)
	if err != nil {
		log.Fatalln(err)
		return
	}
}

// 归档目标下文件
func TarFile(targetPath, appleFrameName string) (err error) {
	// 获取目标目录下所有文件
	files := make([]string, 0)
	dirs, err := ioutil.ReadDir(targetPath)
	if err != nil {
		return
	}
	fmt.Println(dirs)
	for _, file := range dirs {
		if file.IsDir() {
			continue
		}
		files = append(files, file.Name())
	}

	fmt.Println(files)
	// 创建苹果框
	appleFrame, err := os.OpenFile("C:\\workspace\\" + appleFrameName,os.O_CREATE|os.O_WRONLY, 644)
	if err != nil {
		log.Fatalln(err)
		return
	}
	defer appleFrame.Close()
	appleTar := tar.NewWriter(appleFrame)
	defer appleTar.Close()

	// 放入苹果, 同名则直接覆盖掉
	for _, file := range files {
		// 获取文件信息
		fileInfo, err := os.Stat(targetPath + "\\" + file)
		if err != nil {
			log.Fatalln(err)
			return err
		}
		// 自动填充头部信息
		hdr, err := tar.FileInfoHeader(fileInfo, "")
		if err != nil {
			log.Fatalln(err)
			return err
		}
		// 写入头部信息
		err = appleTar.WriteHeader(hdr)
		if err != nil {
			log.Fatalln(err)
			return err
		}
		f, err := os.Open(targetPath + "\\" + file)
		if err != nil {
			fmt.Println(err)
			return err
		}
		// 读取文件所有内容
		content, err := ioutil.ReadAll(f)
		if err != nil {
			log.Fatalln(err)
			return err
		}
		// 写入到苹果框中
		appleTar.Write(content)
		// 关闭文件
		f.Close()
	}
	return
}

//读取归档文件
func TarRead(targetPath, appleFrameName string) (err error) {
	f, err := os.Open(targetPath + "\\" + appleFrameName)
	if err != nil {
		return
	}
	defer f.Close()
	tr := tar.NewReader(f)
	for hdr, err := tr.Next(); err != io.EOF; hdr, err = tr.Next(){
		if err != nil {
			return err
		}
		fmt.Println(hdr.Name)
	}
	return
}