Go io.NewSectionReader 怎样限制读取范围

处理大文件、归档分片或对象存储中的局部数据时，最怕把“这一段”误当成“整个输入”。io.NewSectionReader 的作用就是把一个实现了 io.ReaderAt 的数据源包装成固定区间：从偏移 off 开始，最多读取 n 个字节，区间结束后返回 io.EOF。

官方地址：https://pkg.go.dev/io#NewSectionReader

如果任务是读取一个文件或字节源的固定片段，先把底层数据视为 ReaderAt，再用 NewSectionReader(r, off, n) 建立边界。之后的 Read、ReadAt 和 Seek 都以这个片段为视野，而不是重新暴露整个数据源。

先把底层数据变成 ReaderAt

NewSectionReader 的第一个参数不是普通的 io.Reader，而是 io.ReaderAt。它需要能够从指定的底层绝对偏移读取数据，因此文件对象、字节读取器等实现了 ReadAt 的类型都可以作为输入。

package main

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

func main() {
    // strings.Reader 同时提供 ReaderAt，适合演示固定区间读取。
    source := strings.NewReader("header|payload|trailer")

    // off 是底层数据的起点，n 是这个视图允许看到的最大长度。
    section := io.NewSectionReader(source, 7, 7)
    fmt.Println(section.Size())
}

这里的 off=7 和 n=7 描述的是一个独立视图。创建对象时不会把整个输入复制到新切片里，关键是保存底层 ReaderAt 与区间元数据。

用 Read 读取受限区间

顺序读取时，SectionReader.Read 会把区间内的数据交给调用方。即使传入的缓冲区很大，读取也不会越过创建时的长度；当区间剩余数据读完，后续读取会看到 io.EOF。

buf := make([]byte, 32)

// 缓冲区可以大于区间，SectionReader 仍只返回区间内的字节。
n, err := section.Read(buf)
if err != nil && err != io.EOF {
    // 非 EOF 错误仍然要交给上层处理。
    panic(err)
}
fmt.Printf("%q, %d\n", buf[:n], n)

// 区间耗尽后再读，按固定视图语义结束。
n, err = section.Read(buf)
fmt.Println(n, err == io.EOF)

一个容易误判的地方是：Read 可能在返回最后一段数据的同时带回 io.EOF，也可能在下一次读取时才返回 EOF。调用方应优先处理 n，再按是否为 EOF 判断是否继续循环。

用 ReadAt 与 Seek 访问区间内部

SectionReader.ReadAt 的偏移是相对于区间起点的，不是底层数据源的绝对偏移。例如区间从底层偏移 7 开始，调用 ReadAt(buf, 2) 访问的是底层偏移 9 附近的数据。这个相对坐标很适合在分片任务中复用同一套解析逻辑。

chunk := make([]byte, 4)

// off=2 表示区间内部的相对位置，不是 source 的绝对位置。
n, err := section.ReadAt(chunk, 2)
if err != nil && err != io.EOF {
    // 读取失败时保留已经读到的字节数，便于上层决定是否重试。
    panic(err)
}
fmt.Printf("%q, %d\n", chunk[:n], n)

// Seek 的位置同样受 SectionReader 的区间长度约束。
pos, err := section.Seek(1, io.SeekStart)
if err != nil {
    panic(err)
}
fmt.Println(pos)

ReadAt 不改变顺序读取游标，适合并行或随机访问区间中的记录；Seek 则调整这个 SectionReader 自己的顺序读取位置。二者都不应该被理解为把底层文件永久移动到某个位置。

用 Size 和 Outer 复查边界

当区间对象要传给多个处理函数时，建议把 Size() 当作协议边界保存下来。它返回创建时的 n，可以用于计算分片数量、限制解析器输入或记录元数据。

在需要追踪来源时，Go 还提供了 Outer()：它返回底层 ReaderAt、起始偏移和区间长度。这样可以在日志或任务元数据里区分“区间内偏移”和“原始数据偏移”，避免重复套用偏移量。

常见问题

区间读完为什么不是底层文件的 EOF？

因为 SectionReader 把人为设置的长度当作结束边界。它返回 EOF 只能说明当前区间没有更多数据，不代表底层输入已经结束。

ReadAt 的偏移为什么和文件偏移对不上？

ReadAt 使用 SectionReader 的相对坐标。需要换算回底层位置时，应使用创建区间时的起点加上区间内偏移，并避免重复叠加。

总结：用 NewSectionReader 的关键不是“截取一份新数据”，而是建立一个带长度边界的读取视图。先确认底层实现了 ReaderAt，再分别理解 Read、ReadAt、Seek 的坐标语义，最后用 Size 和 Outer 保存边界信息，固定区间读取就不容易越界或错位。