java - 模拟TCP发送中的背压

我正在编写一些Java TCP/IP网络代码(客户端-服务器)，在该代码中，我必须处理发送比接收快得多的情况，从而阻止了发送操作的一端。 (因为send和recv缓冲区已满)。为了设计我的代码，我想先解决这些情况，然后看看客户端和服务器在变化的负载下如何工作。但是我无法适当设置参数来实现这种背压。我尝试将Socket.setSendBufferSize(int size)和Socket.setReceiveBufferSize(int size)设置为较小的值-希望很快就会填满，但是我可以看到发送操作完成了，而无需等待客户端消耗已经写入的足够数据。 (这意味着较小的send和recv缓冲区大小无效)

我采用的另一种方法是使用Netty，并设置ServerBootstrap.setOption("child.sendBufferSize", 256);，但是即使这样也没有太大用处。谁能帮助我了解我在做什么错/

最佳答案

缓冲区的大小取决于操作系统，通常为8 KB左右。

public static void main(String... args) throws IOException, InterruptedException {
    ServerSocketChannel ssc = ServerSocketChannel.open();
    ssc.bind(new InetSocketAddress(0)); // open on a random port
    InetSocketAddress remote = new InetSocketAddress("localhost", ssc.socket().getLocalPort());
    SocketChannel sc = SocketChannel.open(remote);
    configure(sc);
    SocketChannel accept = ssc.accept();
    configure(accept);

    ByteBuffer bb = ByteBuffer.allocateDirect(16 * 1024 * 1024);
    // write as much as you can
    while (sc.write(bb) > 0)
        Thread.sleep(1);
    System.out.println("The socket write wrote " + bb.position() + " bytes.");
}

private static void configure(SocketChannel socketChannel) throws IOException {
    socketChannel.configureBlocking(false);
    socketChannel.socket().setSendBufferSize(8);
    socketChannel.socket().setReceiveBufferSize(8);
}

在我的机器上打印

The socket write wrote 32768 bytes.

这是发送和接收缓冲区的总和，但是我怀疑它们都是16 KB