首页 >后端开发 >Golang >如何在 Go 中从连接读取任意字节时保留缓冲区大小?

如何在 Go 中从连接读取任意字节时保留缓冲区大小?

Linda Hamilton
Linda Hamilton原创
2024-10-26 07:47:301029浏览

How to Preserve Buffer Size While Reading Arbitrary Bytes from a Connection in Go?

在 Go 中从连接读取任意字节时保留缓冲区

在必须从连接读取数据流并进一步读取的情况下以块的形式处理时,出现了一个常见的挑战:处理不同的流长度并适应任意字节数。

考虑以下代码片段,其中使用 256 字节缓冲区来读取和处理来自连接的数据:

buf := make([]byte, 256)
for {
    n, err := conn.Read(buf)
    fmt.Println(string(buf))
    if err != nil || n== 0 {
        return
    }
    Handle(buf[:n])
}

虽然这种方法在有足够字节可用时效果很好,但在流结束时会遇到问题,导致可读字节少于 256 个。为了妥善处理这种情况,需要替代解决方案。

一种方法是利用 bytes.Buffer,它提供了一种收集从连接接收的数据的便捷方法。通过利用 bytes.Buffer,可以在单个操作中累积完整的数据流并将其传递到所需的处理程序:

var b bytes.Buffer
if _, err := io.Copy(&b, conn); err != nil {
   return err
}
Handle(b.Bytes())

通过此实现,Handle 函数将整个数据流作为单个数据流接收字节切片,确保无缝处理,无论其长度如何。

通过采用这种方法,开发人员可以有效地处理任意长度的流,保留所需的缓冲区大小,同时保持从连接接收的数据的完整性。

以上是如何在 Go 中从连接读取任意字节时保留缓冲区大小?的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn