Go bufio.Reader 怎么用 Discard 跳过固定长度头部

固定协议头部已经确定长度时，Go 中可以直接调用 bufio.Reader.Discard(n) 跳过前 n 个字节。关键不是只调用这一行，而是检查返回的 discarded 是否等于 n：相等才表示头部完整消费；小于 n 时，输入可能提前结束，后续载荷不能继续按完整协议解析。

官方文档：https://pkg.go.dev/bufio

Discard 的调用关系：固定长度按字节前进

Reader 只是给底层 io.Reader 增加缓冲。Discard(n) 会先消费当前缓冲中的字节，不够时再请求底层输入；调用者拿到的是实际跳过数量和错误，而不是头部内容。它适合“这段数据不需要解析，但长度已知”的协议入口。

固定头部的最小写法要检查实际跳过数量

下面假设协议头部固定为 8 字节，头部之后才是 JSON 载荷。示例把检查放在读取载荷之前，避免短输入被误当成合法消息。

package main

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

func readPayload(input string, headerLen int) ([]byte, error) {
	// 用字节数表达协议头长度，不能用中文字符数替代。
	r := bufio.NewReader(strings.NewReader(input))
	discarded, err := r.Discard(headerLen)
	// 返回数量小于目标值时，头部并未完整到达。
	if err != nil {
		return nil, fmt.Errorf("skip header: discarded=%d want=%d: %w", discarded, headerLen, err)
	}
	if discarded != headerLen {
		return nil, fmt.Errorf("short header: discarded=%d want=%d", discarded, headerLen)
	}
	// 头部已消费，剩余字节才交给载荷解析器。
	payload, err := io.ReadAll(r)
	if err != nil {
		return nil, fmt.Errorf("read payload: %w", err)
	}
	return payload, nil
}

func main() {
	// 前八个字节是固定头部，后面是待解析的正文。
	payload, err := readPayload("HDR00001{\"ok\":true}", 8)
	if err != nil {
		fmt.Println("读取失败:", err)
		return
	}
	fmt.Printf("载荷: %s\n", payload)
}

这里的 HDR00001 恰好是 8 个字节。若协议头包含多字节 UTF-8 文本，长度仍应按编码后的字节数计算；更稳妥的二进制协议通常使用明确的字节字段，而不是把可变中文文本塞进固定头部。

短输入、负数和缓冲边界要分开处理

官方定义有两个容易被忽略的边界：当 0 时，跳过动作保证不会读取底层 reader；如果剩余输入少于 n，Discard 会返回少于 n 的数量并带上错误。负数则不是“跳过 0 字节”，应在进入 API 前拒绝。

动态长度协议先校验，再决定是否跳过

如果头部长度来自网络字段，不要把它直接传入 Discard。先检查下限、上限和协议允许的总长度，再执行跳过；在网络连接上还应配合读取超时，避免等待一个永远凑不齐的头部。对于需要保留头部进行校验的场景，先用 Peek(n) 查看但不前进；对于必须读满 n 字节的字段，使用 io.ReadFull(r, buf) 更直观。只有确认数据无需保存、长度也已可信时，Discard 才是最简洁的选择。

func skipHeader(r *bufio.Reader, headerLen, maxHeader int) error {
	// 外部长度先做边界检查，避免异常值消耗过多输入。
	if headerLen  maxHeader {
		return fmt.Errorf("invalid header length: %d", headerLen)
	}
	discarded, err := r.Discard(headerLen)
	// 同时检查错误和数量，防止短输入进入下一层解析。
	if err != nil || discarded != headerLen {
		return fmt.Errorf("incomplete header: discarded=%d want=%d err=%v", discarded, headerLen, err)
	}
	return nil
}

常见问题

Discard 会把跳过的数据返回出来吗？

不会。它只返回实际跳过的字节数和错误；如果还要检查头部内容，应先用 Peek 或读入独立缓冲区。

Discard 返回 nil 就一定跳过了完整头部吗？

正常实现中短跳过会带错误，但健壮的协议代码仍应同时检查 discarded == headerLen，让完整性条件明确写在代码里。

固定长度是字符数还是字节数？

Discard 的参数是字节数。UTF-8 文本的字符数和字节数可能不同，协议应明确长度单位。

把 Discard 当成“推进读取游标”的操作即可：长度可信时直接跳过，长度不可信时先校验，返回数量不足时立即终止当前消息解析。