Go flate Writer 怎样复用压缩器

连续压缩许多独立响应时，不必每次都调用 flate.NewWriter 创建新的压缩器。更合适的做法是保留一个 *flate.Writer，每段数据结束后检查 Close，下一段开始前用 Reset 换到新的输出目标。Flush 只解决“流还没结束但接收端要读到当前数据”的问题，不能替代每段压缩数据的完整收尾。

先用 Reset 复用 Writer 的内部状态

flate.Writer 的 Reset 会把 Writer 恢复到类似重新调用 NewWriter 的状态，同时沿用原来的压缩级别和字典。它的价值在于复用已分配的工作内存，减少批量小对象压缩时反复创建和回收压缩器的成本。

下面的例子把每条消息压缩到各自的 bytes.Buffer。关键点是：上一段先 Close，下一段再 Reset；不能在一个尚未收尾的压缩段中途切换目标。

package main

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

func main() {
    messages := []string{"订单创建", "库存已锁定", "支付待确认"}
    writer, err := flate.NewWriter(nil, flate.DefaultCompression)
    if err != nil {
        panic(err) // 初始化失败时没有可复用的 Writer
    }

    for _, message := range messages {
        var compressed bytes.Buffer
        writer.Reset(&compressed) // 切换输出目标并复用内部缓冲

        if _, err := writer.Write([]byte(message)); err != nil {
            panic(err) // 底层写入错误不能留到下一轮处理
        }
        if err := writer.Close(); err != nil {
            panic(err) // Close 负责当前压缩段的最终收尾
        }
        fmt.Printf("%s: %d bytes\\n", message, compressed.Len())
    }
}

每段数据结束后用 Close 完成收尾

Write 只表示数据交给压缩器，压缩后的字节可能还在 Writer 的内部缓冲中。对“每条消息一段独立压缩结果”的场景，正确生命周期是 Reset → Write → Close。Close 会先 flush 待写数据，再完成当前流的结束标记；返回值必须检查，因为底层 io.Writer 可能在最后一次写入时失败。

复用并不等于并发共享。一个 Writer 在某段数据完成前不能被另一个 goroutine 调用 Reset；如果服务需要并行压缩，应按 worker 独占实例，或在对象池外层安排串行归还。

网络分帧场景单独使用 Flush

长连接协议有另一种需求：压缩流还要继续写，但接收端现在就要读到已经发送的内容。这时调用 Flush。官方文档将它描述为把待写数据刷到底层 Writer，并写入同步标记；即使没有待写数据，也可能产生至少 4 字节的同步标记，因此不应在每个很小的片段上无节制调用。

func sendFrame(writer *flate.Writer, frame []byte) error {
    if _, err := writer.Write(frame); err != nil {
        return err // 先返回当前帧的压缩或底层写入错误
    }
    if err := writer.Flush(); err != nil {
        return err // Flush 后接收端才有机会读到当前帧
    }
    return nil // 流仍然开放，后面还可以继续写帧
}

func finishStream(writer *flate.Writer) error {
    return writer.Close() // 关闭时做最终 flush，并结束 DEFLATE 流
}

因此，Flush 和 Close 不是互相替代的两个版本：前者让流在未结束时可消费，后者结束当前流。若把多条独立消息分别存储，优先每段 Close 后再 Reset；若是一个持续连接，则保持一个流并按协议节奏选择性 Flush，最后只调用一次 Close。

按输出目标、并发和错误检查做复用决策

最后做三项核对：第一，输出目标在 Reset 前是否仍被上一段使用；第二，是否把 Flush 误当成了最终结束；第三，Write、Flush 和 Close 的错误是否都能回到调用方。满足这三点，复用 Writer 才是可控的工程优化。

常见问题

Reset 会改变原来的压缩级别吗？

不会。它保留 Writer 的压缩级别和字典，只切换到新的输出目标并重置当前状态。

调用 Flush 后还需要 Close 吗？

需要。Flush 只让当前数据可被继续读取，Close 才负责最终收尾并结束压缩流。

可以把一个 flate.Writer 放进多个 goroutine 共用吗？

不建议。Writer 的生命周期操作应由一个串行拥有者管理；并发场景请按 worker 分配实例。