本教程详细介绍了如何在go语言中使用 `compress/gzip` 包对数据进行gzip压缩和解压。通过实际的代码示例，您将学习如何创建gzip写入器和读取器，将数据写入内存或文件进行压缩，以及如何从压缩数据中读取原始内容，确保数据完整性和资源管理。
介绍 compress/gzip 包
Go标准库中的 compress/gzip 包提供了对Gzip数据格式的读写支持，广泛用于数据压缩和传输。它遵循 io.Reader 和 io.Writer 接口，使得与Go的I/O系统无缝集成。使用此包，您可以轻松地将数据压缩为Gzip格式，或将Gzip格式的数据解压回原始形式。
数据压缩（Gzip Writer）
要压缩数据，我们需要使用 gzip.NewWriter 函数创建一个 *gzip.Writer。这个写入器会将其接收到的数据进行Gzip压缩，并写入到底层的 io.Writer。
关键步骤：
- 创建一个目标 io.Writer，例如 bytes.Buffer 用于内存操作，或 *os.File 用于文件操作。
- 调用 gzip.NewWriter(targetWriter) 创建一个 *gzip.Writer。
- 使用 gzipWriter.Write([]byte) 方法写入要压缩的数据。
- 非常重要： 调用 gzipWriter.Close() 来刷新所有待处理的压缩数据并写入Gzip文件尾。如果不调用 Close()，压缩文件可能不完整或无法解压。
示例：内存数据压缩
立即学习“go语言免费学习笔记（深入）”；
package main
import (
"bytes"
"compress/gzip"
"fmt"
"log"
)
func main() {
// 原始数据
originalData := []byte("hello, world\nThis is a test string for gzip compression.")
// 创建一个 bytes.Buffer 作为压缩数据的目标
var compressedBuffer bytes.Buffer
// 创建一个 gzip.Writer，将压缩数据写入 compressedBuffer
gzipWriter := gzip.NewWriter(&compressedBuffer)
// 写入原始数据进行压缩
_, err := gzipWriter.Write(originalData)
if err != nil {
log.Fatalf("Failed to write data to gzip writer: %v", err)
}
// 关闭 gzip.Writer，确保所有数据被刷新并写入文件尾
err = gzipWriter.Close()
if err != nil {
log.Fatalf("Failed to close gzip writer: %v", err)
}
fmt.Printf("Original data length: %d bytes\n", len(originalData))
fmt.Printf("Compressed data length: %d bytes\n", compressedBuffer.Len())
// fmt.Printf("Compressed data (hex): %x\n", compressedBuffer.Bytes()) // 可选：查看压缩后的字节
}数据解压（Gzip Reader）
要解压数据，我们需要使用 gzip.NewReader 函数创建一个 *gzip.Reader。这个读取器会从底层的 io.Reader 中读取Gzip压缩数据，并提供原始的解压内容。
关键步骤：
- 创建一个源 io.Reader，例如 bytes.Buffer 包含压缩数据，或 *os.File 读取压缩文件。
- 调用 gzip.NewReader(sourceReader) 创建一个 *gzip.Reader。
- 使用 io.Copy(destinationWriter, gzipReader) 或 gzipReader.Read([]byte) 方法读取解压后的数据。
- 非常重要： 调用 gzipReader.Close() 来释放资源。
示例：内存数据解压
package main
import (
"bytes"
"compress/gzip"
"fmt"
"io"
"log"
)
func main() {
// 假设 compressedBuffer 已经包含了之前压缩的数据
originalData := []byte("hello, world\nThis is a test string for gzip compression.")
var compressedBuffer bytes.Buffer
gzipWriter := gzip.NewWriter(&compressedBuffer)
_, err := gzipWriter.Write(originalData)
if err != nil {
log.Fatalf("Failed to write data to gzip writer: %v", err)
}
err = gzipWriter.Close()
if err != nil {
log.Fatalf("Failed to close gzip writer: %v", err)
}
// --- 压缩部分结束 ---
// 创建一个 gzip.Reader 从 compressedBuffer 中读取压缩数据
gzipReader, err := gzip.NewReader(&compressedBuffer)
if err != nil {
log.Fatalf("Failed to create gzip reader: %v", err)
}
defer gzipReader.Close() // 确保 reader 被关闭，释放资源
// 创建一个 bytes.Buffer 来存储解压后的数据
var decompressedBuffer bytes.Buffer
// 将解压后的数据复制到 decompressedBuffer
_, err = io.Copy(&decompressedBuffer, gzipReader)
if err != nil {
log.Fatalf("Failed to decompress data: %v", err)
}
fmt.Printf("Decompressed data length: %d bytes\n", decompressedBuffer.Len())
fmt.Printf("Decompressed data:\n%s", decompressedBuffer.String())
// 验证解压后的数据是否与原始数据一致
if bytes.Equal(originalData, decompressedBuffer.Bytes()) {
fmt.Println("Decompression successful: Data matches original.")
} else {
fmt.Println("Decompression failed: Data does not match original.")
}
}完整示例：文件压缩与解压
在实际应用中，我们通常需要对文件进行Gzip压缩和解压。以下是一个完整的示例，演示如何将一个文件压缩成 .gz 文件，然后再将其解压回原始文件。
package main
import (
"compress/gzip"
"fmt"
"io"
"log"
"os"
"path/filepath"
"strings"
)
// compressFile 将源文件压缩到目标文件
func compressFile(srcPath, destPath string) error {
srcFile, err := os.Open(srcPath)
if err != nil {
return fmt.Errorf("failed to open source file %s: %w", srcPath, err)
}
defer srcFile.Close() // 确保源文件被关闭
destFile, err := os.Create(destPath)
if err != nil {
return fmt.Errorf("failed to create destination file %s: %w", destPath, err)
}
defer destFile.Close() // 确保目标文件被关闭
gzipWriter := gzip.NewWriter(destFile)
defer gzipWriter.Close() // 确保 gzipWriter 被关闭，完成压缩
_, err = io.Copy(gzipWriter, srcFile) // 将源文件内容复制到 gzipWriter 进行压缩
if err != nil {
return fmt.Errorf("failed to compress data: %w", err)
}
return nil
}
// decompressFile 将源Gzip文件解压到目标文件
func decompressFile(srcPath, destPath string) error {
srcFile, err := os.Open(srcPath)
if err != nil {
return fmt.Errorf("failed to open source gzip file %s: %w", srcPath, err)
}
defer srcFile.Close() // 确保源Gzip文件被关闭
gzipReader, err := gzip.NewReader(srcFile)
if err != nil {
return fmt.Errorf("failed to create gzip reader for %s: %w", srcPath, err)
}
defer gzipReader.Close() // 确保 gzipReader 被关闭，释放资源
destFile, err := os.Create(destPath)
if err != nil {
return fmt.Errorf("failed to create destination file %s: %w", destPath, err)
}
defer destFile.Close() // 确保目标文件被关闭
_, err = io.Copy(destFile, gzipReader) // 将解压后的数据复制到目标文件
if err != nil {
return fmt.Errorf("failed to decompress data: %w", err)
}
return nil
}
func main() {
// 1. 创建一个测试文件
originalFileName := "test_data.txt"
originalContent := "This is some test content for the original file.\n" +
"It spans multiple lines to demonstrate file operations.\n" +
"Go's compress/gzip package is very efficient!"
err := os.WriteFile(originalFileName, []byte(originalContent), 0644)
if err != nil {
log.Fatalf("Failed to create original file: %v", err)
}
fmt.Printf("Created original file: %s\n", originalFileName)
// 2. 压缩文件
compressedFileName := originalFileName + ".gz"
err = compressFile(originalFileName, compressedFileName)
if err != nil {
log.Fatalf("Failed to compress file %s: %v", originalFileName, err)
}
fmt.Printf("Compressed %s to %s\n", 
