如何在libp2p-go的直连P2P连接中手动刷新Write操作？

问题原因 #

go-libp2p 的 network.Stream 本身没有公开的显式Flush API，写入后对端收不到、只有关流才收到的问题通常来自两个原因：

1. 对端读逻辑使用了 io.ReadAll 这类需要读到流EOF才返回的方法，只要不关闭流的写方向，对端的读操作会一直阻塞，看起来就像消息没发过去。
2. 流多路复用层（默认是yamux/mplex）启用了写入合并优化，会攒够一定大小的数据再批量发送以降低网络开销，小数据写入会暂时存在协议层缓冲里，不会立即发送。

使用bufio没有效果是因为bufio.Flush()只能刷新bufio自身的用户态缓冲，刷完的数据还是会写入libp2p的流对象，无法触达多路复用层的协议缓冲。

解决方案 #

根据流使用场景选对应方案即可：

场景1：单次请求响应的短流 #

写完待发送的数据后，调用流的CloseWrite()方法，这个方法会强制把所有缓冲数据全部发送给对端，同时通知对端不会再写入新数据，且只会关闭写方向，还可以正常读取对端返回的响应。
可以给封装类加一个刷新方法：

// FlushWrite 强制刷新所有写入缓冲到对端，调用后无法再写入新数据
func (rwc *NetReaderWriter) FlushWrite() error {
    return rwc.stream.CloseWrite()
}

写完数据后手动调用一次FlushWrite()，对端就能立即收到数据。

场景2：双向持续通信的长流 #

如果需要反复写入数据，不能每次写完都关写端，可以用两种方案：

1. 禁用多路复用层的写入合并
   初始化libp2p节点时，修改yamux的默认配置，关闭写入合并优化，每次写入都会立即发送：

import (
    "github.com/libp2p/go-libp2p"
    yamux "github.com/libp2p/go-libp2p-yamux"
)

func initLibp2pHost() (host.Host, error) {
    yamuxCfg := yamux.DefaultConfig()
    // 关闭写入合并，禁用缓冲攒包
    yamuxCfg.WriteCoalesce = false
    return libp2p.New(
        libp2p.Muxer("/yamux/1.0.0", yamux.Transport(yamuxCfg)),
        // 其余节点配置项
    )
}

2. 应用层实现消息分帧
   不管底层缓冲逻辑，在应用层给每个消息加上明确的边界，比如每个消息前加4字节的固定长度标记消息大小，对端读的时候先读4字节拿到长度，再读取对应长度的字节即可拿到完整消息，不需要依赖流关闭或者底层刷新。

内容的提问来源于stack exchange，提问作者Eric '3ToedSloth'