当前位置：首页 > news >正文

golang标准库archive/tar实现打包压缩及解压

news 来源：原创 2025/7/21 22:33:03

文章目录

前言
一、单个文件操作
- 1.单个文件打包示例
- 2.单个文件解包示例
二、目录示例
- 1.打包压缩
- 2.解包
补充

前言

这个包就是将文件进行打包和解包，通俗理解就是Linux 下的 tar 命令。主要是通过 tar.Reader 读取 tar 包，通过 tar.Writer 写入 tar包，在写入的过程中再设置一下头，详细的过程以示例的方式进行展示，可以查看代码里面的注释。

标准库 tar 中文文档https://studygolang.com/static/pkgdoc/pkg/archive_tar.html

一、单个文件操作

1.单个文件打包示例

package mainimport ("os""log""archive/tar""fmt""io"
)func main() {// 准备打包的源文件var srcFile = "sshd"// 打包后的文件var desFile = fmt.Sprintf("%s.tar",srcFile)// 需要注意文件的打开即关闭的顺序，因为 defer 是后入先出，所以关闭顺序很重要// 第一次写这个示例的时候就没注意，导致写完的 tar 包不完整// ###### 第 1 步，先准备好一个 tar.Writer 结构，然后再向里面写入内容。 ######// 创建一个文件，用来保存打包后的 passwd.tar 文件fw, err := os.Create(desFile)ErrPrintln(err)defer fw.Close()// 通过 fw 创建一个 tar.Writertw := tar.NewWriter(fw)// 这里不要忘记关闭，如果不能成功关闭会造成 tar 包不完整// 所以这里在关闭的同时进行判断，可以清楚的知道是否成功关闭defer func() {if err := tw.Close(); err != nil {ErrPrintln(err)}}()// ###### 第 2 步，处理文件信息，也就是 tar.Header 相关的 ######// tar 包共有两部分内容：文件信息和文件数据// 通过 Stat 获取 FileInfo，然后通过 FileInfoHeader 得到 hdr tar.*Headerfi, err := os.Stat(srcFile)ErrPrintln(err)hdr, err := tar.FileInfoHeader(fi, "")// 将 tar 的文件信息 hdr 写入到 twerr = tw.WriteHeader(hdr)ErrPrintln(err)// 将文件数据写入// 打开准备写入的文件fr, err := os.Open(srcFile)ErrPrintln(err)defer fr.Close()written, err := io.Copy(tw, fr)ErrPrintln(err)log.Printf("共写入了 %d 个字符的数据\n",written)
}// 定义一个用来打印的函数，少写点代码，因为要处理很多次的 err
// 后面其他示例还会继续使用这个函数，就不单独再写，望看到此函数了解
func ErrPrintln(err error)  {if err != nil {log.Println(err)os.Exit(1)}
}

2.单个文件解包示例

package mainimport ("os""archive/tar""io""log"
)func main() {var srcFile = "passwd.tar"// 将 tar 包打开fr, err := os.Open(srcFile)ErrPrintln(err)defer fr.Close()// 通过 fr 创建一个 tar.*Reader 结构，然后将 tr 遍历，并将数据保存到磁盘中tr := tar.NewReader(fr)for hdr, err := tr.Next(); err != io.EOF; hdr, err = tr.Next(){// 处理 err ！= nil 的情况ErrPrintln(err)// 获取文件信息fi := hdr.FileInfo()// 创建一个空文件，用来写入解包后的数据fw, err := os.Create(fi.Name())ErrPrintln(err)// 将 tr 写入到 fwn, err := io.Copy(fw, tr)ErrPrintln(err)log.Printf("解包： %s 到 %s ，共处理了 %d 个字符的数据。", srcFile,fi.Name(),n)// 设置文件权限，这样可以保证和原始文件权限相同，如果不设置，会根据当前系统的 umask 来设置。os.Chmod(fi.Name(),fi.Mode().Perm())// 注意，因为是在循环中，所以就没有使用 defer 关闭文件// 如果想使用 defer 的话，可以将文件写入的步骤单独封装在一个函数中即可fw.Close()}
}func ErrPrintln(err error){if err != nil {log.Fatalln(err)os.Exit(1)}
}

二、目录示例

打包整个目录，且打包的时候通过 gzip 或者 bzip2 压缩。如果要打包整个目录，可以通过递归的方式来实现。此处只演示 gzip 方式压缩，这个实现非常简单，只需要在 fw 和 tw 之前加上一层压缩即可

1.打包压缩

代码如下（示例）：

package mainimport ("archive/tar""compress/gzip""fmt""io""os""path/filepath"
)
func createTarFile(tarFilePath string, sourceDir string) error {// 创建Tar文件tarFile, err := os.Create(tarFilePath)if err != nil {return err}defer tarFile.Close()// 将 tar 包使用 gzip 压缩，其实添加压缩功能很简单，// 只需要在 fw 和 tw 之前加上一层压缩就行了，和 Linux 的管道的感觉类似gw := gzip.NewWriter(tarFile)defer gw.Close()// 创建Tar写入器tarWriter := tar.NewWriter(tarFile)defer tarWriter.Close()// 遍历源目录并添加文件到Tarerr = filepath.Walk(sourceDir, func(path string, info os.FileInfo, err error) error {if err != nil {return err}// 创建Tar头信息header, err := tar.FileInfoHeader(info, "")if err != nil {return err}// 设置文件路径//这里的思路就是递归处理目录及目录下的所有文件和目录header.Name, err = filepath.Rel(filepath.Dir(sourceDir), path)fmt.Printf("header.Name:%s\n", header.Name)if err != nil {return err}// 写入头信息if err := tarWriter.WriteHeader(header); err != nil {return err}// 如果是文件，写入文件内容if !info.IsDir() {file, err := os.Open(path)if err != nil {return err}defer file.Close()_, err = io.Copy(tarWriter, file)return err}return nil})return err
}
func main(){tarFilePath := "output.tar"sourceDir := "/etc/sshd/sshd_config"err = createTarFile(tarFilePath, sourceDir)if err != nil {panic(err)}
}

2.解包