本文介绍如何将未压缩的 http.Response.Body 流式压缩为 Gzip 数据，利用 io.Copy 与 gzip.Writer 实现内存高效、无中间缓冲的压缩流程，并提供可直接复用的封装函数及关键注意事项。
本文介绍如何将未压缩的 `http.response.body` 流式压缩为 gzip 数据，利用 `io.copy` 与 `gzip.writer` 实现内存高效、无中间缓冲的压缩流程，并提供可直接复用的封装函数及关键注意事项。
在 Go 的 HTTP 客户端开发中，常需对响应体进行缓存（如存入 Redis）。若原始响应未启用 Gzip 压缩（即 Content-Encoding 为空或非 gzip），而你希望统一以压缩格式持久化以节省存储空间和带宽，就需要在内存中完成流式 Gzip 压缩。核心挑战在于：http.Response.Body 是 io.ReadCloser，而 gzip.Writer 需要 io.Writer —— 二者接口方向相反，不能直接对接，但可通过 io.Copy 桥接实现零拷贝式流式处理。
以下是一个简洁、健壮的实现方案：
import (
"bytes"
"compress/gzip"
"io"
"net/http"
)
// getCompressedBody 将未压缩的响应体压缩为 []byte
// 注意：调用方需确保 r.Body 未被读取过，且不为 nil
func getCompressedBody(r *http.Response) ([]byte, error) {
var buf bytes.Buffer
gz := gzip.NewWriter(&buf)
defer gz.Close() // 确保 Close 被调用，触发压缩数据刷新
_, err := io.Copy(gz, r.Body)
if err != nil {
return nil, err
}
// 必须显式 Close()，否则部分压缩数据可能滞留在内部 buffer 中未写出
if err = gz.Close(); err != nil {
return nil, err
}
return buf.Bytes(), nil
}✅ 关键点说明：
- io.Copy(gz, r.Body) 是流式核心：它持续从 r.Body 读取数据块，并写入 gzip.Writer，全程无需加载全部响应体到内存；
- gz.Close() 不可省略：Gzip 是块编码格式，Close() 会写入尾部校验和（CRC32）和长度信息，缺失将导致解压失败；
- 使用 bytes.Buffer 作为底层 io.Writer 是安全的，因其支持多次写入且 Bytes() 返回只读切片，适合构造最终字节数据；
- 若后续需直接写入 Redis（如 HSET 字段值），返回的 []byte 可直接使用；若目标是文件或网络流，建议改用 io.Writer 参数化设计，避免内存缓冲（见下文优化建议）。
⚠️ 注意事项：
- 此函数不处理 r.Body 的关闭逻辑——调用后仍需手动调用 r.Body.Close()（除非你已将其所有权转移）；
- 若响应体极大（如百 MB 级），bytes.Buffer 仍会占用等量内存，此时应优先考虑「流式直写」方案（例如：gzip.NewWriter(redisWriter)），跳过 []byte 中间态；
- 该函数假设输入 r.Body 确实未压缩。生产环境建议先检查 r.Header.Get("Content-Encoding")，避免重复压缩或误压已压缩内容；
- 如需复用性更强，推荐重构为接受 io.Reader 的通用函数：
func compressReader(r io.Reader) ([]byte, error) { /* 同上 */ }这样既可传入 r.Body，也可传入任意 io.Reader（如文件、字符串 reader），提升正交性与测试友好性。
综上，Go 的 io 接口设计天然支持流式转换：只需找准 Reader → Writer 的管道入口，配合 io.Copy 与适当的 Close() 时机，即可实现高效、低内存开销的 Gzip 压缩逻辑。
