如何在 Go 中使用 net.Conn.Read 可靠地处理持久 TCP 套接字的消息帧？

正确使用 net.Conn.Read 来实现持久 TCP 套接字

在 Go 中，使用持久 TCP 套接字涉及建立连接并不断地进行操作读取传入的数据。 net.Conn.Read 函数负责从套接字检索数据，但其行为可能不会立即清晰。

理解 TCP 中的消息帧

与某些与其他协议相比，TCP 本身并不提供消息成帧。这意味着由应用程序定义一种在连续数据流中分隔消息的方法。

带标头的传统消息框架

在您之前的 C# 经验中，消息前面带有包含消息大小的标头。这种方法允许接收应用程序知道后续消息的确切长度。

Go 如何处理消息处理

Go 采用了不同的方法。默认情况下，net.Conn.Read() 没有确定消息结束的固有机制。您在问题中提供的代码片段只是在循环中调用 conn.Read() ，而不考虑消息边界。这可能适用于某些场景，但它不是一个可靠或可扩展的解决方案。

解决方案：自定义消息帧

要正确处理持久 TCP 套接字，您需要实现自定义消息框架。这涉及缓冲传入数据并根据您自己定义的协议对其进行解析。

使用 bufio.Reader 的示例

一种推荐的方法是使用内置 bufio .Reader 包裹您的网络。Conn。此包装器提供了附加功能，并使读取数据更加高效。

import (
    "bufio"
    "fmt"
    "io"
    "net"
)

func main() {
    listener, err := net.Listen("tcp", ":8080")
    if err != nil {
        panic(err)
    }

    for {
        conn, err := listener.Accept()
        if err != nil {
            continue
        }

        go handleConnection(conn)
    }
}

func handleConnection(conn net.Conn) {
    defer conn.Close()
    r := bufio.NewReader(conn)

    for {
        size, err := r.ReadByte()
        if err != nil {
            return
        }

        buff := make([]byte, size)
        if _, err := io.ReadFull(r, buff); err != nil {
            return
        }

        fmt.Println("Received:", buff)
    }
}

在此示例中，handleConnection 函数首先从连接中读取单个字节，表示后续消息的长度。然后，它创建一个缓冲区并使用 io.ReadFull 读取指定大小的完整消息。这允许应用程序无缝处理不同长度的消息。

以上是如何在 Go 中使用 net.Conn.Read 可靠地处理持久 TCP 套接字的消息帧？的详细内容。更多信息请关注PHP中文网其他相关文章！