Go TCP 读取是非阻塞的：解决不完整的数据接收

在 Go 中，TCP 读取是非阻塞的，这意味着它们立即返回任何可用的数据，即使它低于预期。此行为与 C 的阻塞读取不同，C 的阻塞读取会等到收到所需的数据量为止。

非阻塞读取的原因

TCP 作为字节流运行，在传输过程中可能会出现碎片。因此，仅根据接收到的字节数无法确定消息的结束。这就需要自定义分隔符或者其他机制来确定消息边界。

数据不完整的解决方案

要读取特定数量的字节，请使用 io.ReadAtLeast 或 io.ReadFull 。对于任意条件，循环读取调用，直到没有错误或满足指定条件为止。

示例：

package main

import (
    "fmt"
    "net"
    "time"
)

func main() {
    conn, _ := net.Dial("tcp", "127.0.0.1:4243")

    // Set a deadline to prevent hanging reads
    _ = conn.SetReadDeadline(time.Now().Add(10 * time.Second))

    // Loop to read until a newline is encountered
    for {
        buf := make([]byte, 512)
        n, err := conn.Read(buf)
        if err != nil {
            fmt.Println(err)
            break
        }
        if n > 0 {
            str := string(buf[:n])
            if str[len(str)-1] == '\n' {
                fmt.Println(str)
                break
            }
        }
    }
}

其他注意事项

• 处理一次写入多个数据包的可能性。
• 使用分隔符或其他方法来定义消息边界.
• 考虑对基于文本的协议使用 net/textproto。
• 实施超时或截止日期以防止资源匮乏。
• 检查网络操作中的所有错误以避免屏蔽问题。