zipfiles.go 1.2 KB

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