Go flate.Resetter 怎么复用解压器处理多段数据

处理很多彼此独立的原始 DEFLATE 数据段时，可以只调用一次 flate.NewReader，把返回值断言为 flate.Resetter，然后在每段开始前执行 Reset(newReader, dict)。Reset 会丢弃上一段的缓冲状态，并像使用新输入重新初始化，但能利用解压器已经分配的内存。

先划清目标：复用解压器，不是拼接数据流

flate.Resetter 解决的是“多个独立压缩段反复创建解压器”的分配成本，不会把几段数据自动拼成一个连续 DEFLATE 流。最稳妥的输入形态是 [][]byte、独立消息体或已知长度的分帧数据，每段都能单独读到该 DEFLATE 流的结束标记。

如果数据实际带 gzip 文件头、校验和或 zlib 包装，直接交给 compress/flate 会丢失外层格式语义，应改用 compress/gzip 或 compress/zlib。只有明确为 RFC 1951 原始 DEFLATE 时才使用这里的流程。

创建一次 Reader，在每段前切换输入

flate.NewReader 返回 io.ReadCloser，官方保证该返回值也实现 flate.Resetter。可以用空输入创建一次，再把每个 bytes.Reader 交给 Reset。

package decode

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

func DecodeSegments(segments [][]byte, dict []byte) ([][]byte, error) {
	// 只创建一次解压器；后续由 Reset 切换底层输入。
	zr := flate.NewReader(nil)
	resetter, ok := zr.(flate.Resetter)
	if !ok {
		_ = zr.Close()
		return nil, fmt.Errorf("flate reader 不支持 Reset")
	}

	results := make([][]byte, 0, len(segments))
	for i, compressed := range segments {
		// bytes.Reader 提供清晰的单段边界，并实现 io.ByteReader。
		if err := resetter.Reset(bytes.NewReader(compressed), dict); err != nil {
			_ = zr.Close()
			return nil, fmt.Errorf("第 %d 段 Reset 失败: %w", i, err)
		}

		var out bytes.Buffer
		// 必须读到 EOF，才能确认这一段的 DEFLATE 流完整结束。
		if _, err := io.Copy(&out, zr); err != nil {
			_ = zr.Close()
			return nil, fmt.Errorf("第 %d 段解压失败: %w", i, err)
		}

		// 每段读取完成后关闭当前状态；下一轮仍可再次 Reset。
		if err := zr.Close(); err != nil {
			return nil, fmt.Errorf("第 %d 段关闭失败: %w", i, err)
		}
		results = append(results, bytes.Clone(out.Bytes()))
	}

	return results, nil
}

把段号放进错误信息很重要：Reset 失败通常表示初始化输入或字典有问题，io.Copy 失败则可能是压缩数据损坏或底层读取错误。发生错误后不要继续复用当前状态；先关闭并把错误返回给上层。

按同一生命周期处理每一段

当底层输入只实现 io.Reader、没有实现 io.ByteReader 时，flate 解压器可能读取超过当前流结束位置的数据。若多段数据紧挨在同一长流里，应该先由外层协议按长度切出每段，或提供能保持字节边界的 reader；不要假设解压器会替业务协议管理下一帧。

字典与并发是最容易遗漏的两条线

如果压缩端使用 flate.NewWriterDict，解压端每次 Reset 都必须传入约定好的相同字典。普通无字典数据则传 nil。不要在不同段之间随意混用字典，否则输入可能表现为损坏。

复用降低的是反复分配解压器内部结构的成本，不代表对象可以并发共享。同一个 Reader 的 Reset、Read 和 Close 会修改同一份状态。并发任务应各自持有实例；若确实需要控制分配，可以在任务之间使用对象池，但归还前必须完成读取与关闭，取出后先 Reset 到新输入。

常见问题

Reset 前必须先 Close 上一段吗？

推荐完整读取到 EOF 后检查 Close，再 Reset 下一段，这与标准库示例的生命周期一致。不要在上一段仍读取时切换输入。

为什么复用后仍然出现额外分配？

Reset 主要复用解压器内部资源，输出缓冲、结果切片、错误包装和上层 reader 仍可能分配。要评估收益应针对整个处理函数做基准测试，而不是假设分配会归零。