Go语言使用compress/gzip包进行数据压缩与解压缩教程

package main

import (
    "bytes"
    "compress/gzip"
    "fmt"
    "log"
)

func main() {
    // 原始数据
    originalData := "hello, world\nThis is a test string that will be gzipped."

    // 1. 创建一个bytes.Buffer作为压缩数据的目标
    var compressedBuffer bytes.Buffer

    // 2. 创建gzip.Writer，将压缩数据写入compressedBuffer
    // gzip.BestCompression 或 gzip.DefaultCompression 等可用于设置压缩级别
    gzipWriter := gzip.NewWriter(&compressedBuffer)

    // 3. 写入原始数据
    _, err := gzipWriter.Write([]byte(originalData))
    if err != nil {
        log.Fatalf("写入数据失败: %v", err)
    }

    // 4. 关闭gzipWriter，确保所有数据被刷新并写入GZIP尾部
    err = gzipWriter.Close()
    if err != nil {
        log.Fatalf("关闭gzipWriter失败: %v", err)
    }

    fmt.Printf("原始数据大小: %d 字节\n", len(originalData))
    fmt.Printf("压缩后数据大小: %d 字节\n", compressedBuffer.Len())
    fmt.Printf("压缩数据 (部分): %x...\n", compressedBuffer.Bytes()[:20]) // 打印前20字节的十六进制表示
}

package main

import (
    "bytes"
    "compress/gzip"
    "fmt"
    "io"
    "log"
)

func main() {
    // 假设这是之前压缩后的数据（这里直接构建一个模拟的压缩数据）
    // 实际应用中，这通常是从文件或网络读取的
    originalData := "hello, world\nThis is a test string that will be gzipped."
    var compressedBuffer bytes.Buffer
    gzipWriter := gzip.NewWriter(&compressedBuffer)
    _, _ = gzipWriter.Write([]byte(originalData))
    _ = gzipWriter.Close()
    // 至此，compressedBuffer包含了压缩后的数据

    // 1. 创建一个bytes.Buffer作为解压缩数据的源
    // 注意：这里我们使用bytes.NewReader来包装compressedBuffer，
    // 因为gzip.NewReader需要一个io.Reader接口
    compressedDataReader := bytes.NewReader(compressedBuffer.Bytes())

    // 2. 创建gzip.Reader，从compressedDataReader读取压缩数据
    gzipReader, err := gzip.NewReader(compressedDataReader)
    if err != nil {
        log.Fatalf("创建gzipReader失败: %v", err)
    }
    defer func() {
        if closeErr := gzipReader.Close(); closeErr != nil {
            log.Printf("关闭gzipReader失败: %v", closeErr)
        }
    }() // 使用defer确保Reader被关闭

    // 3. 读取解压后的数据
    var decompressedBuffer bytes.Buffer
    _, err = io.Copy(&decompressedBuffer, gzipReader)
    if err != nil {
        log.Fatalf("解压数据失败: %v", err)
    }

    // 4. 打印解压后的数据
    fmt.Printf("解压后的数据:\n%s\n", decompressedBuffer.String())

    // 验证数据是否一致
    if decompressedBuffer.String() != originalData {
        fmt.Println("警告: 解压后的数据与原始数据不一致！")
    } else {
        fmt.Println("解压成功，数据一致。")
    }
}

package main

import (
    "bytes"
    "compress/gzip"
    "fmt"
    "io"
    "log"
    "os"
)

const (
    fileName = "example.txt.gz"
    dataToCompress = "This is a sample text that will be compressed into a gzip file. " +
        "It contains multiple lines and some repetitive content to show compression effectiveness.\n" +
        "Line 2 of the sample text.\n" +
        "Line 3 of the sample text."
)

func main() {
    // --- 1. 压缩数据到文件 ---
    fmt.Println("--- 开始压缩数据到文件 ---")
    err := compressToFile(fileName, dataToCompress)
    if err != nil {
        log.Fatalf("压缩文件失败: %v", err)
    }
    fmt.Printf("数据已成功压缩到 %s\n", fileName)
    fmt.Printf("原始数据大小: %d 字节\n", len(dataToCompress))

    fileInfo, err := os.Stat(fileName)
    if err == nil {
        fmt.Printf("压缩文件大小: %d 字节\n", fileInfo.Size())
    }


    // --- 2. 从文件解压缩数据 ---
    fmt.Println("\n--- 开始从文件解压缩数据 ---")
    decompressedData, err := decompressFromFile(fileName)
    if err != nil {
        log.Fatalf("解压缩文件失败: %v", err)
    }
    fmt.Printf("从 %s 解压后的数据:\n%s\n", fileName, decompressedData)

    // 验证数据一致性
    if decompressedData == dataToCompress {
        fmt.Println("解压成功，数据与原始数据一致。")
    } else {
        fmt.Println("警告: 解压后的数据与原始数据不一致！")
    }

    // 清理生成的压缩文件
    err = os.Remove(fileName)
    if err != nil {
        log.Printf("清理文件 %s 失败: %v", fileName, err)
    } else {
        fmt.Printf("已删除临时文件: %s\n", fileName)
    }
}

// compressToFile 将字符串数据压缩并写入指定的GZIP文件
func compressToFile(filePath, data string) error {
    file, err := os.Create(filePath)
    if err != nil {
        return fmt.Errorf("创建文件失败: %w", err)
    }
    defer func() {
        if closeErr := file.Close(); closeErr != nil {
            log.Printf("关闭文件 %s 失败: %v", filePath, closeErr)
        }
    }()

    gzipWriter := gzip.NewWriter(file)
    defer func() {
        if closeErr := gzipWriter.Close(); closeErr != nil {
            log.Printf("关闭gzipWriter失败: %v", closeErr)
        }
    }()

    _, err = gzipWriter.Write([]byte(data))
    if err != nil {
        return fmt.Errorf("写入GZIP数据失败: %w", err)
    }
    return nil
}

// decompressFromFile 从指定的GZIP文件读取并解压缩数据
func decompressFromFile(filePath string) (string, error) {
    file, err := os.Open(filePath)
    if err != nil {
        return "", fmt.Errorf("打开文件失败: %w", err)
    }
    defer func() {
        if closeErr := file.Close(); closeErr != nil {
            log.Printf("关闭文件 %s 失败: %v", filePath, closeErr)
        }
    }()

    gzipReader, err := gzip.NewReader(file)
    if err != nil {
        return "", fmt.Errorf("创建gzipReader失败: %w", err)
    }
    defer func() {
        if closeErr := gzipReader.Close(); closeErr != nil {
            log.Printf("关闭gzipReader失败: %v", closeErr)
        }
    }()

    var decompressedBuffer bytes.Buffer
    _, err = io.Copy(&decompressedBuffer, gzipReader)
    if err != nil {
        return "", fmt.Errorf("从GZIP读取解压数据失败: %w", err)
    }

    return decompressedBuffer.String(), nil
}