Golangcompress/gzip数据压缩与解压方法

package main

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

func compressData(data []byte) ([]byte, error) {
    var buf bytes.Buffer
    writer := gzip.NewWriter(&buf)
    
    _, err := writer.Write(data)
    if err != nil {
        return nil, err
    }

    err = writer.Close() // 必须调用 Close() 以写入尾部并刷新数据
    if err != nil {
        return nil, err
    }

    return buf.Bytes(), nil
}

func main() {
    original := []byte("Hello, this is a test string for gzip compression.")
    compressed, err := compressData(original)
    if err != nil {
        panic(err)
    }
    fmt.Printf("原始大小: %d\n", len(original))
    fmt.Printf("压缩后大小: %d\n", len(compressed))
}

func decompressData(compressed []byte) ([]byte, error) {
    buf := bytes.NewReader(compressed)
    reader, err := gzip.NewReader(buf)
    if err != nil {
        return nil, err
    }
    defer reader.Close()

    var result bytes.Buffer
    _, err = result.ReadFrom(reader)
    if err != nil {
        return nil, err
    }

    return result.Bytes(), nil
}

decompressed, err := decompressData(compressed)
if err != nil {
    panic(err)
}
fmt.Printf("解压后数据: %s\n", decompressed)

file, _ := os.Create("data.gz")
defer file.Close()
writer := gzip.NewWriter(file)
writer.Write([]byte("your data here"))
writer.Close()

file, _ := os.Open("data.gz")
defer file.Close()
reader, _ := gzip.NewReader(file)
io.Copy(os.Stdout, reader)
reader.Close()