Go io.SectionReader 结合 Seek 定位分片

我在处理“只读文件中间一段内容”的代码时，最容易犯的错误不是不会调用 Seek，而是把底层文件的绝对偏移和分片自己的局部游标混在一起。io.SectionReader 正好提供了一个受长度限制的读取视图：先用底层 ReaderAt 的绝对起点和长度创建分片，再让 Seek 在这个分片内部移动。这样既能复用同一个底层数据源，也不会让一次读取越过分片末端。

官方文档：https://pkg.go.dev/io#SectionReader

先把底层绝对偏移和分片长度说清楚

SectionReader 实现了 Read、Seek 和 ReadAt，但它并不是复制一份新数据。它保存的是对底层 ReaderAt 的一个有边界视图。假设原始数据从字节 0 开始，分片起点是 6、长度是 12，那么这个分片能看到的底层范围是 [6, 18)，局部位置 0 对应底层位置 6。

因此，第一步要把“从哪里开始”和“最多读多少”交给构造函数，而不是先对底层对象随意 Seek 再靠调用方记住边界：

package main

import (
	"fmt"
	"io"
	"strings"
)

func main() {
	// strings.Reader 同时实现 ReaderAt，适合演示固定字节分片。
	source := strings.NewReader("header|part-A-012345|part-B-abcdef|tail")

	// off 是底层数据的绝对起点，n 是这个读取视图允许访问的字节数。
	section := io.NewSectionReader(source, 7, 10)

	// Size 返回分片长度，不代表底层 ReaderAt 的总长度。
	fmt.Println("section size:", section.Size())
}

这里的 section 只负责 source 的一段数据。若底层对象换成 *os.File，仍然是文件对象负责生命周期，SectionReader 不会替调用方关闭文件。

用 SeekStart 从分片起点定位

对 SectionReader 调用 Seek(offset, io.SeekStart) 时，offset 相对于分片起点解释，而不是相对于底层 ReaderAt 的字节 0。上面的分片从底层 7 开始，调用 Seek(2, io.SeekStart) 就是在底层位置 9 开始下一次读取。

func readFromSectionStart(r *io.SectionReader, localOffset int64, size int) ([]byte, error) {
	// SeekStart 的偏移是 SectionReader 内的局部偏移。
	if _, err := r.Seek(localOffset, io.SeekStart); err != nil {
		return nil, fmt.Errorf("定位分片失败: %w", err)
	}

	buf := make([]byte, size)
	n, err := io.ReadFull(r, buf)
	if err != nil {
		// 读不到完整 size 时，返回已读数据和原始错误，方便调用方判断边界。
		return buf[:n], fmt.Errorf("读取分片失败: %w", err)
	}
	return buf, nil
}

这个函数的关键在于 localOffset 不应直接填入底层文件的绝对位置。如果业务拿到的是绝对位置，就先减去分片起点，再传给 SeekStart。同时，io.ReadFull 会要求填满缓冲区；当请求长度超过分片剩余长度时，调用方会得到部分数据和错误，而不是误以为已经读完。

SeekCurrent 和 SeekEnd 只在分片范围内移动

io.SeekCurrent 适合从当前局部游标继续移动，io.SeekEnd 则把分片末端作为参考点。例如要读取分片最后 4 个字节，可以使用 Seek(-4, io.SeekEnd)。这些常量的语义来自 io.Seeker，但最终位置仍受 SectionReader 的边界约束。

func readTail(r *io.SectionReader) ([]byte, error) {
	// 从分片末尾向前定位，-4 表示分片最后四个字节的起点。
	if _, err := r.Seek(-4, io.SeekEnd); err != nil {
		return nil, fmt.Errorf("定位分片尾部失败: %w", err)
	}

	tail := make([]byte, 4)
	if _, err := io.ReadFull(r, tail); err != nil {
		return nil, fmt.Errorf("读取分片尾部失败: %w", err)
	}
	return tail, nil
}

func moveForward(r *io.SectionReader, delta int64) error {
	// SeekCurrent 从当前局部游标开始移动，适合顺序解析多个字段。
	if _, err := r.Seek(delta, io.SeekCurrent); err != nil {
		return fmt.Errorf("移动局部游标失败: %w", err)
	}
	return nil
}

实践中我会把“移动后要读多少”一起写出来，因为只看 Seek 的返回位置很容易漏掉剩余长度。分片末尾之后的位置不应被当成可用数据；一旦读取返回 io.EOF 或 io.ErrUnexpectedEOF，就应把它当作边界信号处理。

顺序定位用 Seek，独立偏移用 ReadAt

Seek 改变的是下一次 Read 使用的局部游标，适合一个解析器按顺序消费分片。如果多个调用方只想读取不同局部偏移，使用 ReadAt 更直接：偏移作为参数传入，每次调用都独立计算，不需要先移动共享游标。官方文档也明确说明，ReadAt 不应影响底层的 seek offset。

func readHeaderAt(r *io.SectionReader, localOffset int64, size int) ([]byte, error) {
	// ReadAt 的 offset 仍是 SectionReader 内的局部偏移。
	if localOffset 

最后检查分片边界和资源所有权

可以把一次读取前的判断收束成四个问题：分片起点是不是底层绝对偏移；传给 Seek 或 ReadAt 的位置是不是分片内偏移；请求长度是否超过当前位置到分片末尾的剩余空间；底层资源是否由创建它的调用方负责关闭。

• 需要从底层绝对位置 absolute 定位时，先计算 absolute - sectionStart。
• 需要继续上一次读取时，用 SeekCurrent；需要从固定末尾倒推时，用 SeekEnd。
• 需要并发或不希望调用方共享游标时，用 ReadAt 或为每个任务创建独立的读取视图。
• 底层是文件、网络对象或其他资源时，仍由资源拥有者负责关闭、取消或回收。

常见问题

SeekStart 的 0 是底层文件的 0 吗？

不是。对 SectionReader 来说，Seek(0, io.SeekStart) 回到当前分片起点；分片起点在底层数据中的绝对位置由 NewSectionReader 的第二个参数决定。

Seek 到分片末尾后还能读到数据吗？

末尾位置表示没有剩余字节。继续读取通常得到 io.EOF；如果一次读取请求跨过末尾，也可能得到部分数据和 io.ErrUnexpectedEOF，应按实际 n 和错误共同判断。

ReadAt 能不能替代所有 Seek？

不能。顺序解析需要一个自然递进的游标时，Seek 加 Read 更符合代码表达；固定偏移、随机读取和并发读取才更适合 ReadAt。

SectionReader 会自动关闭底层文件吗？

不会。它只是一个受边界限制的读取视图，底层文件的打开和关闭仍由调用方管理。