模拟TCP发送中的背压

Question

我正在编写一些java TCP/IP网络代码(客户端 - 服务器),我必须在其中处理发送比接收快得多的情况,从而阻止一端的发送操作.(因为send和recv缓冲区已填满).为了设计我的代码,我想首先围绕这些情况,看看客户端和服务器在不同负载下的行为.但是我无法适当地设置参数来实现这种背压.我尝试设置Socket.setSendBufferSize(int size)和Socket.setReceiveBufferSize(int size)小值 - 希望很快就会填满,但我可以看到发送操作完成而不等待客户端消耗已经写入的足够数据.(这意味着小的send和recv缓冲区大小没有影响)

我采用的另一种方法是使用Netty,然后设置ServerBootstrap.setOption("child.sendBufferSize", 256);,但即使这样也没用多少.任何人都可以帮我理解我做错了什么/

Answer 1

缓冲区的最小大小取决于操作系统，通常约为 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