這篇文章主要介紹了go語言制作的zip壓縮程序，其主體思路是首先創(chuàng)建一個讀寫緩沖，然后用壓縮器包裝該緩沖，用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. // 參數(shù)frm可以是文件或目錄，不會給dst添加.zip擴展名 
16. func compress(frm, dst string) error { 
17. buf := bytes.NewBuffer(make([]byte, 0, 10*1024*1024)) // 創(chuàng)建一個讀寫緩沖 
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) // 轉(zhuǎn)換為zip格式的文件信息 
26. if err != nil { 
27. return filepath.SkipDir 
28. } 
29. header.Name, _ = filepath.Rel(filepath.Dir(frm), path) 
30. if !info.IsDir() { 
31. // 確定采用的壓縮算法（這個是內(nèi)建注冊的deflate） 
32. header.Method = 8 
33. file, err = ioutil.ReadFile(path) // 獲取文件內(nèi)容 
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) // 創(chuàng)建一條記錄并寫入文件信息 
44. if err != nil { 
45. return err 
46. } 
47. _, err = w.Write(file) // 非目錄文件會寫入數(shù)據(jù)，目錄不會寫入數(shù)據(jù) 
48. if err != nil { // 因為目錄的內(nèi)容可能會修改 
49. return err // 最關(guān)鍵的是我不知道咋獲得目錄文件的內(nèi)容 
50. } 
51. return nil 
52. }) 
53. if err != nil { 
54. return err 
55. } 
56. myzip.Close() // 關(guān)閉壓縮器，讓壓縮器緩沖中的數(shù)據(jù)寫入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中的數(shù)據(jù)寫入文件 
63. if err != nil { 
64. return err 
65. } 
66. return nil 
67. } 

以上所述就是本文的全部內(nèi)容了，希望大家能夠喜歡。