go语言制作的zip压缩程序

这篇文章主要介绍了go语言制作的zip压缩程序，其主体思路是首先创建一个读写缓冲，然后用压缩器包装该缓冲，用Walk方法来将所有目录下的文件写入zip，有需要的小伙伴参考下。

可以压缩文件和目录。

1. package main 
2. import ( 
3. "archive/zip" 
4. "bytes" 
5. "fmt" 
6. "io/ioutil" 
7. "os" 
8. "path/filepath" 
9. ) 
10. func main() { 
11. if err := compress(`gopkg`, `gopkg.zip`); err != nil { 
12. fmt.Println(err) 
13. } 
14. } 
15. // 参数frm可以是文件或目录，不会给dst添加.zip扩展名 
16. func compress(frm, dst string) error { 
17. buf := bytes.NewBuffer(make([]byte, 0, 10*1024*1024)) // 创建一个读写缓冲 
18. myzip := zip.NewWriter(buf) // 用压缩器包装该缓冲 
19. // 用Walk方法来将所有目录下的文件写入zip 
20. err := filepath.Walk(frm, func(path string, info os.FileInfo, err error) error { 
21. var file []byte 
22. if err != nil { 
23. return filepath.SkipDir 
24. } 
25. header, err := zip.FileInfoHeader(info) // 转换为zip格式的文件信息 
26. if err != nil { 
27. return filepath.SkipDir 
28. } 
29. header.Name, _ = filepath.Rel(filepath.Dir(frm), path) 
30. if !info.IsDir() { 
31. // 确定采用的压缩算法（这个是内建注册的deflate） 
32. header.Method = 8 
33. file, err = ioutil.ReadFile(path) // 获取文件内容 
34. if err != nil { 
35. return filepath.SkipDir 
36. } 
37. } else { 
38. file = nil 
39. } 
40. // 上面的部分如果出错都返回filepath.SkipDir 
41. // 下面的部分如果出错都直接返回该错误 
42. // 目的是尽可能的压缩目录下的文件，同时保证zip文件格式正确 
43. w, err := myzip.CreateHeader(header) // 创建一条记录并写入文件信息 
44. if err != nil { 
45. return err 
46. } 
47. _, err = w.Write(file) // 非目录文件会写入数据，目录不会写入数据 
48. if err != nil { // 因为目录的内容可能会修改 
49. return err // 最关键的是我不知道咋获得目录文件的内容 
50. } 
51. return nil 
52. }) 
53. if err != nil { 
54. return err 
55. } 
56. myzip.Close() // 关闭压缩器，让压缩器缓冲中的数据写入buf 
57. file, err := os.Create(dst) // 建立zip文件 
58. if err != nil { 
59. return err 
60. } 
61. defer file.Close() 
62. _, err = buf.WriteTo(file) // 将buf中的数据写入文件 
63. if err != nil { 
64. return err 
65. } 
66. return nil 
67. } 

以上所述就是本文的全部内容了，希望大家能够喜欢。