zipfiles.go 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374
  1. package utils
  2. import (
  3. "archive/zip"
  4. "io"
  5. "os"
  6. "strings"
  7. )
  8. //@author: [piexlmax](https://github.com/piexlmax)
  9. //@function: ZipFiles
  10. //@description: 压缩文件
  11. //@param: filename string, files []string, oldform, newform string
  12. //@return: error
  13. func ZipFiles(filename string, files []string, oldForm, newForm string) error {
  14. newZipFile, err := os.Create(filename)
  15. if err != nil {
  16. return err
  17. }
  18. defer func() {
  19. _ = newZipFile.Close()
  20. }()
  21. zipWriter := zip.NewWriter(newZipFile)
  22. defer func() {
  23. _ = zipWriter.Close()
  24. }()
  25. // 把files添加到zip中
  26. for _, file := range files {
  27. zipFile, err := os.Open(file)
  28. if err != nil {
  29. return err
  30. }
  31. err = func() error {
  32. defer func() {
  33. _ = zipFile.Close()
  34. }()
  35. // 获取file的基础信息
  36. info, err := zipFile.Stat()
  37. if err != nil {
  38. return err
  39. }
  40. header, err := zip.FileInfoHeader(info)
  41. if err != nil {
  42. return err
  43. }
  44. // 使用上面的FileInfoHeader() 就可以把文件保存的路径替换成我们自己想要的了,如下面
  45. header.Name = strings.Replace(file, oldForm, newForm, -1)
  46. // 优化压缩
  47. // 更多参考see http://golang.org/pkg/archive/zip/#pkg-constants
  48. header.Method = zip.Deflate
  49. writer, err := zipWriter.CreateHeader(header)
  50. if err != nil {
  51. return err
  52. }
  53. if _, err = io.Copy(writer, zipFile); err != nil {
  54. return err
  55. }
  56. return nil
  57. }()
  58. if err != nil {
  59. return err
  60. }
  61. }
  62. return nil
  63. }