=Start=
缘由：
在学习Go 语言的过程中整理总结一下常见功能的实现代码，方便以后需要的时候参考。
正文：
参考解答：
话不多说，先上代码。先可以运行，再逐步按需理解和改进。
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 |
package main import ( "archive/tar" "compress/gzip" "flag" "fmt" "io" "log" "os" "path" ) // main functions shows how to TarGz a directory and // UnTarGz a file func main() { var srcDirPath = *flag.String("srcDir", "testdata", "The source directory name.") var targetFilePath = *flag.String("targetName", "testdata.tar.gz", "The target file name.") flag.Parse() fmt.Printf("srcDirPath: %s, type: %T\n", srcDirPath, srcDirPath) fmt.Printf("targetFilePath: %s, type: %T\n", targetFilePath, targetFilePath) TarGz(srcDirPath, targetFilePath) // 压缩 UnTarGz(targetFilePath, srcDirPath+"_temp") // 解压 } // Gzip and tar from source directory or file to destination file // you need check file exist before you call this function func TarGz(srcDirPath string, destFilePath string) { fw, err := os.Create(destFilePath) handleError(err) defer fw.Close() // Gzip writer gw := gzip.NewWriter(fw) defer gw.Close() // Tar writer tw := tar.NewWriter(gw) defer tw.Close() // Check if it's a file or a directory f, err := os.Open(srcDirPath) handleError(err) fi, err := f.Stat() handleError(err) if fi.IsDir() { // handle source directory fmt.Println("Cerating tar.gz from directory...") tarGzDir(srcDirPath, path.Base(srcDirPath), tw) } else { // handle file directly fmt.Println("Cerating tar.gz from " + fi.Name() + "...") tarGzFile(srcDirPath, fi.Name(), tw, fi) } fmt.Println("Well done!") } // Deal with directories // if find files, handle them with tarGzFile // Every recurrence append the base path to the recPath // recPath is the path inside of tar.gz func tarGzDir(srcDirPath string, recPath string, tw *tar.Writer) { // Open source diretory dir, err := os.Open(srcDirPath) handleError(err) defer dir.Close() // Get file info slice fis, err := dir.Readdir(0) handleError(err) for _, fi := range fis { // Append path curPath := srcDirPath + "/" + fi.Name() // Check it is directory or file if fi.IsDir() { // Directory // (Directory won't add unitl all subfiles are added) fmt.Printf("Adding path...%s\\n", curPath) tarGzDir(curPath, recPath+"/"+fi.Name(), tw) } else { // File fmt.Printf("Adding file...%s\\n", curPath) } tarGzFile(curPath, recPath+"/"+fi.Name(), tw, fi) } } // Deal with files func tarGzFile(srcFile string, recPath string, tw *tar.Writer, fi os.FileInfo) { if fi.IsDir() { // Create tar header hdr := new(tar.Header) // if last character of header name is '/' it also can be directory // but if you don't set Typeflag, error will occur when you untargz hdr.Name = recPath + "/" hdr.Typeflag = tar.TypeDir hdr.Size = 0 //hdr.Mode = 0755 | c_ISDIR hdr.Mode = int64(fi.Mode()) hdr.ModTime = fi.ModTime() // Write hander err := tw.WriteHeader(hdr) handleError(err) } else { // File reader fr, err := os.Open(srcFile) handleError(err) defer fr.Close() // Create tar header hdr := new(tar.Header) hdr.Name = recPath hdr.Size = fi.Size() hdr.Mode = int64(fi.Mode()) hdr.ModTime = fi.ModTime() // Write hander err = tw.WriteHeader(hdr) handleError(err) // Write file data _, err = io.Copy(tw, fr) handleError(err) } } // Ungzip and untar from source file to destination directory // you need check file exist before you call this function func UnTarGz(srcFilePath string, destDirPath string) { fmt.Println("UnTarGzing " + srcFilePath + "...") // Create destination directory os.Mkdir(destDirPath, os.ModePerm) fr, err := os.Open(srcFilePath) handleError(err) defer fr.Close() // Gzip reader gr, err := gzip.NewReader(fr) // Tar reader tr := tar.NewReader(gr) for { hdr, err := tr.Next() if err == io.EOF { // End of tar archive break } //handleError(err) fmt.Println("UnTarGzing file..." + hdr.Name) // Check if it is diretory or file if hdr.Typeflag != tar.TypeDir { // Get files from archive // Create diretory before create file os.MkdirAll(destDirPath+"/"+path.Dir(hdr.Name), os.ModePerm) // Write data to file fw, _ := os.Create(destDirPath + "/" + hdr.Name) handleError(err) _, err = io.Copy(fw, tr) handleError(err) } } fmt.Println("Well done!") } func handleError(err error) { log.Println(err) } |
&
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
package main import ( "archive/tar" "compress/gzip" "flag" "fmt" "io" "os" "path/filepath" "strings" ) func main() { var srcDirPath = *flag.String("srcDir", "testdata", "The source directory name.") var targetFilePath = *flag.String("targetName", "testdata.tar.gz", "The target file name.") flag.Parse() fmt.Printf("srcDirPath: %s, type: %T\n", srcDirPath, srcDirPath) fmt.Printf("targetFilePath: %s, type: %T\n", targetFilePath, targetFilePath) TarFilesDirs(srcDirPath, targetFilePath) // 仅压缩，解压可以使用tar命令进行 } func TarFilesDirs(path string, tarFilePath string) error { file, err := os.Create(tarFilePath) if err != nil { return err } defer file.Close() gz := gzip.NewWriter(file) defer gz.Close() tw := tar.NewWriter(gz) defer tw.Close() if err := tarit(path, tw); err != nil { return err } return nil } func tarit(source string, tw *tar.Writer) error { info, err := os.Stat(source) if err != nil { return nil } var baseDir string if info.IsDir() { baseDir = filepath.Base(source) } return filepath.Walk(source, func(path string, info os.FileInfo, err error) error { if err != nil { return err } var link string if info.Mode()&os.ModeSymlink != 0 { if link, err = os.Readlink(path); err != nil { return err } } header, err := tar.FileInfoHeader(info, link) if err != nil { return err } if baseDir != "" { header.Name = filepath.Join(baseDir, strings.TrimPrefix(path, source)) } if !info.Mode().IsRegular() { //nothing more to do for non-regular return nil } if err := tw.WriteHeader(header); err != nil { return err } if info.IsDir() { return nil } file, err := os.Open(path) if err != nil { return err } defer file.Close() buf := make([]byte, 16) if _, err = io.CopyBuffer(tw, file, buf); err != nil { return err } return nil }) } |
参考链接：
- Quick golang script for creating a gzipped tarball#测试OK，但比较简单
- Go语言实现tar.gz打包解包#测试OK，简单加工之后可使用
- Golang 实现tar.gz打包#测试OK，且比较简洁
- 代码片段 – Golang 创建 .tar.gz 压缩包
- Golang-压缩-解压-Tar.Gz
- golang压缩,解压tar.gz文件
- golang tar gzip 压缩，解压（含目录文件）
- https://golang.org/pkg/archive/tar/
- Golang: Working with Gzip and Tar
https://gist.github.com/svett/76799ba2edca89961be6 - Golang 学习笔记（四）- archive/tar 实现打包压缩及解压
https://broqiang.com/posts/45 - 5.压缩文件读写
=END=