| | |
| | | import ( |
| | | "archive/tar" |
| | | "archive/zip" |
| | | "basic.com/valib/logger.git" |
| | | "bytes" |
| | | "compress/gzip" |
| | | "crypto/md5" |
| | |
| | | return nil |
| | | } |
| | | |
| | | func TarGzCheck(filePath string) bool { |
| | | f, err := os.Open(filePath) |
| | | //压缩 使用gzip压缩成tar.gz |
| | | func TarCompress(files []*os.File, dest string) error { |
| | | d, _ := os.Create(dest) |
| | | defer d.Close() |
| | | gw := gzip.NewWriter(d) |
| | | defer gw.Close() |
| | | tw := tar.NewWriter(gw) |
| | | defer tw.Close() |
| | | for _, file := range files { |
| | | err := gzipCompress(file, "", tw) |
| | | if err != nil { |
| | | return err |
| | | } |
| | | } |
| | | return nil |
| | | } |
| | | |
| | | func gzipCompress(file *os.File, prefix string, tw *tar.Writer) error { |
| | | info, err := file.Stat() |
| | | if err != nil { |
| | | return false |
| | | return err |
| | | } |
| | | defer f.Close() |
| | | buf := make([]byte, 4) |
| | | if n, err := f.Read(buf); err != nil || n < 4 { |
| | | return false |
| | | if info.IsDir() { |
| | | prefix = prefix + "/" + info.Name() |
| | | fileInfos, err := file.Readdir(-1) |
| | | if err != nil { |
| | | return err |
| | | } |
| | | for _, fi := range fileInfos { |
| | | f, err := os.Open(file.Name() + "/" + fi.Name()) |
| | | if err != nil { |
| | | return err |
| | | } |
| | | err = gzipCompress(f, prefix, tw) |
| | | if err != nil { |
| | | return err |
| | | } |
| | | } |
| | | } else { |
| | | header, err := tar.FileInfoHeader(info, "") |
| | | header.Name = prefix + "/" + header.Name |
| | | if err != nil { |
| | | return err |
| | | } |
| | | err = tw.WriteHeader(header) |
| | | if err != nil { |
| | | return err |
| | | } |
| | | _, err = io.Copy(tw, file) |
| | | file.Close() |
| | | if err != nil { |
| | | return err |
| | | } |
| | | } |
| | | fType := hex.EncodeToString(buf) |
| | | logger.Debug("file type :",fType) |
| | | return fType == "8b1f" |
| | | return nil |
| | | } |
| | | |
| | | //解压 tar.gz |