io.Copy() 转发不正确:如何确保首次发送的消息被正确转发?

io.copy() 转发不正确:如何确保首次发送的消息被正确转发?

解决 io.copy() 转发不正确的问题

在示例代码中,发现客户端首次发送的消息无法被转发。问题出在 handleconn 函数中,首次读取数据时并未进行转发:

data := make([]byte, 128)
for {
    _, err := conn.read(data)
    if err != nil || err == io.eof {
        log.println("连接已断开!")
        break
    }

    // 转发
    go io.copy(conn, dstconn)
    io.copy(dstconn, conn)    
}

正确的做法是,将首次读取的数据也进行转发:

data := make([]byte, 128)
for {
    n, err := conn.Read(data)
    if err != nil || err == io.EOF || n == 0 {
        log.Println("连接已断开!")
        break
    }

    // 转发
    go io.Copy(conn, dstConn)
    io.Copy(dstConn, conn)    
}

此外,在首次读取数据时,可以省略 go 协程,直接同步执行转发操作,以提高首次转发消息的效率。

以上就是io.Copy() 转发不正确:如何确保首次发送的消息被正确转发?的详细内容,更多请关注其它相关文章!