在 Golang 中压缩文件

在 Golang 中，你可以通过使用标准库中的 archive/zip 包来压缩文件。

压缩单个文件

下面是一个简单的程序，用于压缩单个文件：

package main

import (
	"archive/zip"
	"fmt"
	"io"
	"os"
)

func main() {
	inputFile := "file.txt"
	outputFile := "file.zip"

	// 创建一个新的 zip 文件
	zipFile, err := os.Create(outputFile)
	if err != nil {
		fmt.Println(err)
		return
	}
	defer zipFile.Close()

	// 创建一个新的 zip 结构体
	zipWriter := zip.NewWriter(zipFile)
	defer zipWriter.Close()

	// 打开要压缩的文件
	fileToZip, err := os.Open(inputFile)
	if err != nil {
		fmt.Println(err)
		return
	}
	defer fileToZip.Close()

	// 获取要压缩的文件的信息
	info, err := fileToZip.Stat()
	if err != nil {
		fmt.Println(err)
		return
	}

	// 创建一个新的文件头
	header, err := zip.FileInfoHeader(info)
	if err != nil {
		fmt.Println(err)
		return
	}
	header.Method = zip.Deflate // 设置压缩算法

	// 将文件头写入 zip 文件中
	writer, err := zipWriter.CreateHeader(header)
	if err != nil {
		fmt.Println(err)
		return
	}

	// 将要压缩的文件写入 zip 文件中
	_, err = io.Copy(writer, fileToZip)
	if err != nil {
		fmt.Println(err)
		return
	}

	fmt.Println("压缩完成")
}

压缩多个文件

下面是一个用于压缩多个文件的程序：

package main

import (
	"archive/zip"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strings"
)

func main() {
	inputDir := "./input"
	outputFile := "files.zip"

	// 创建一个新的 zip 文件
	zipFile, err := os.Create(outputFile)
	if err != nil {
		fmt.Println(err)
		return
	}
	defer zipFile.Close()

	// 创建一个新的 zip 结构体
	zipWriter := zip.NewWriter(zipFile)
	defer zipWriter.Close()

	// 遍历要压缩的文件夹
	filepath.Walk(inputDir, func(path string, info os.FileInfo, err error) error {
		// 忽略文件夹
		if info.IsDir() {
			return nil
		}

		// 打开要压缩的文件
		fileToZip, err := os.Open(path)
		if err != nil {
			fmt.Println(err)
			return err
		}
		defer fileToZip.Close()

		// 创建一个新的文件头
		header, err := zip.FileInfoHeader(info)
		if err != nil {
			fmt.Println(err)
			return err
		}

		// 设置压缩算法
		header.Method = zip.Deflate

		// 路径处理
		rel, err := filepath.Rel(inputDir, path)
		if err != nil {
			fmt.Println(err)
			return err
		}
		header.Name = strings.Replace(rel, "\\", "/", -1)

		// 将文件头写入 zip 文件中
		writer, err := zipWriter.CreateHeader(header)
		if err != nil {
			fmt.Println(err)
			return err
		}

		// 将要压缩的文件写入 zip 文件中
		_, err = io.Copy(writer, fileToZip)
		if err != nil {
			fmt.Println(err)
			return err
		}

		return nil
	})

	fmt.Println("压缩完成")
}

结论

在 Golang 中，使用标准库中的 archive/zip 包来压缩文件非常简单，只需几行代码就能实现。本文提供了一个简单的压缩单个文件的示例程序和一个用于压缩多个文件的示例程序。