sockets - golang 套接字在 Write() 方法返回之前不发送所有字节

Question

今天，当我尝试向我的服务器(一个非常简单的 TCP 服务器，也是用 Golang 编写的)发送 100M 数据时，我发现 TCPConn.Write 方法返回 104857600 和 nil 错误，然后我关闭 socket 。但是我的服务器只收到很少的数据。我想是因为 Write 方法工作在异步模式下，所以虽然该方法返回 104857600，但只向服务器发送了很少的数据。所以我想知道是否有一种方法可以将Write设置为同步模式，或者如何检测是否所有数据都从套接字发送到服务器。

代码如下:服务器:

const ListenAddress = "192.168.0.128:8888"
func main() {

    var l net.Listener
    var err error
    l, err = net.Listen("tcp", ListenAddress)
    if err != nil {
        fmt.Println("Error listening:", err)
        os.Exit(1)
    }
    defer l.Close()
    fmt.Println("listen on " + ListenAddress)
    for {
        conn, err := l.Accept()
        if err != nil {
            fmt.Println("Error accepting: ", err)
            os.Exit(1)
        }
        //logs an incoming message
        fmt.Printf("Received message %s -> %s \n", conn.RemoteAddr(), conn.LocalAddr())
        // Handle connections in a new goroutine.
        go handleRequest(conn)
    }
}
func handleRequest(conn net.Conn) {
    defer conn.Close()
    rcvLen := 0
    rcvData := make([]byte,20 * 1024 * 1024) // 20M
    for {
        l , err := conn.Read(rcvData)
        if err != nil {
            fmt.Printf("%v", err)
            return
        }
        rcvLen += l
        fmt.Printf("recv: %d\r\n", rcvLen)

        conn.Write(rcvData[:l])
    }
}

客户:

conn, err := net.Dial("tcp", "192.168.0.128:8888")
if err != nil {
    fmt.Println(err)
    os.Exit(-1)
}
defer conn.Close()
data := make([]byte, 500 * 1024 * 1024)
length, err := conn.Write(data)
fmt.Println("send len: ", length)

客户端的输出:

send len:  524288000

服务器的输出:

listen on 192.168.0.128:8888
Received message 192.168.0.2:50561 -> 192.168.0.128:8888 
recv: 166440
recv: 265720
EOF

我知道如果我可以通过SetLinger 方法让客户端等待一段时间，数据将在套接字关闭之前全部发送到服务器。但我想找到一种方法，让套接字在返回之前发送所有数据，而无需调用 SetLinger()。

请原谅我糟糕的英语。

Answer 1

在尝试写入之前是否轮询了套接字？

套接字后面是操作系统的 tcp 堆栈。在套接字上写入时，您将字节推送到发送缓冲区。然后您的操作系统自行决定何时以及如何发送。如果接收端在其接收缓冲区中没有缓冲区空间可用，则您的发送端知道这一点并且不会在发送缓冲区中放入任何更多信息。

确保您的发送缓冲区有足够的空间容纳您接下来要发送的任何内容。这是通过轮询套接字来完成的。此方法通常称为 Socket.Poll。我建议您检查 golang 文档以了解确切的用法。