Golang 小技巧:使用 io.ReadFull 确保读取指定长度的数据
【摘要】 在 Go 中调用:n, err := reader.Read(buf)并不能保证一次就把 buf 填满。即使还有数据可读,Read 也可能只返回部分内容:buf := make([]byte, 8)n, err := reader.Read(buf)fmt.Println(n)如果业务要求必须读取完整的 8 个字节,可以使用:buf := make([]byte, 8)_, err := i...
在 Go 中调用:
n, err := reader.Read(buf)
并不能保证一次就把 buf 填满。
即使还有数据可读,Read 也可能只返回部分内容:
buf := make([]byte, 8)
n, err := reader.Read(buf)
fmt.Println(n)
如果业务要求必须读取完整的 8 个字节,可以使用:
buf := make([]byte, 8)
_, err := io.ReadFull(reader, buf)
if err != nil {
return err
}
io.ReadFull 会持续读取,直到:
buf被填满;- 或者读取过程中遇到错误。
例如读取固定长度的协议头:
header := make([]byte, 4)
if _, err := io.ReadFull(conn, header); err != nil {
return err
}
这样可以避免把一次 Read 错误地当成“肯定能读满”。
小结
当你需要准确读取固定长度的数据时,可以使用:
_, err := io.ReadFull(reader, buf)
这个技巧在 TCP 协议解析、二进制文件读取、自定义数据包处理等场景中非常实用。
【声明】本内容来自华为云开发者社区博主,不代表华为云及华为云开发者社区的观点和立场。转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息,否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱:
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论(0)